From 4c9e55d662443921865476dde3195d6f8f504d0f Mon Sep 17 00:00:00 2001
From: Devin Villarosa <devinvillarosa92.business@gmail.com>
Date: Wed, 11 Dec 2024 14:11:40 -0800
Subject: [PATCH] changes to toHaveBeenCalledWith()

---
 .../data-table/data-table.test.tsx            | 74 ++++++++++++++-----
 .../data-table/data-table.test.tsx            | 28 ++++---
 2 files changed, 70 insertions(+), 32 deletions(-)

diff --git a/ui-v2/src/components/concurrency/global-concurrency-view/data-table/data-table.test.tsx b/ui-v2/src/components/concurrency/global-concurrency-view/data-table/data-table.test.tsx
index 162194c5628e..16e35f843441 100644
--- a/ui-v2/src/components/concurrency/global-concurrency-view/data-table/data-table.test.tsx
+++ b/ui-v2/src/components/concurrency/global-concurrency-view/data-table/data-table.test.tsx
@@ -1,27 +1,26 @@
+import { Toaster } from "@/components/ui/toaster";
 import { render, screen } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
 import { createWrapper } from "@tests/utils";
 import { describe, expect, it, vi } from "vitest";
 import { Table } from "./data-table";
 
-const MOCK_DATA = [
-	{
-		id: "0",
-		created: "2021-01-01T00:00:00Z",
-		updated: "2021-01-01T00:00:00Z",
-		active: true,
-		name: "global concurrency limit 0",
-		limit: 0,
-		active_slots: 0,
-		slot_decay_per_second: 0,
-	},
-];
+const MOCK_ROW = {
+	id: "0",
+	created: "2021-01-01T00:00:00Z",
+	updated: "2021-01-01T00:00:00Z",
+	active: true,
+	name: "global concurrency limit 0",
+	limit: 0,
+	active_slots: 0,
+	slot_decay_per_second: 0,
+};
 
 describe("GlobalConcurrencyLimitTable -- table", () => {
 	it("renders row data", () => {
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={vi.fn()}
 				onEditRow={vi.fn()}
 				searchValue=""
@@ -44,7 +43,7 @@ describe("GlobalConcurrencyLimitTable -- table", () => {
 
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={mockFn}
 				onEditRow={vi.fn()}
 				searchValue=""
@@ -56,7 +55,7 @@ describe("GlobalConcurrencyLimitTable -- table", () => {
 			screen.getByRole("button", { name: /open menu/i, hidden: true }),
 		);
 		await user.click(screen.getByRole("menuitem", { name: /delete/i }));
-		expect(mockFn).toHaveBeenCalledOnce();
+		expect(mockFn).toBeCalledWith(MOCK_ROW);
 	});
 	it("calls onEdit upon clicking rest action menu item", async () => {
 		const user = userEvent.setup();
@@ -64,7 +63,7 @@ describe("GlobalConcurrencyLimitTable -- table", () => {
 
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={vi.fn()}
 				onEditRow={mockFn}
 				searchValue=""
@@ -76,6 +75,47 @@ describe("GlobalConcurrencyLimitTable -- table", () => {
 			screen.getByRole("button", { name: /open menu/i, hidden: true }),
 		);
 		await user.click(screen.getByRole("menuitem", { name: /edit/i }));
-		expect(mockFn).toHaveBeenCalledOnce();
+		expect(mockFn).toHaveBeenCalledWith(MOCK_ROW);
+	});
+
+	it("toggles active switch", async () => {
+		const user = userEvent.setup();
+
+		const { rerender } = render(
+			<>
+				<Toaster />
+				<Table
+					data={[MOCK_ROW]}
+					onDeleteRow={vi.fn()}
+					onEditRow={vi.fn()}
+					searchValue=""
+					onSearchChange={vi.fn()}
+				/>
+			</>,
+			{ wrapper: createWrapper() },
+		);
+		expect(
+			screen.getByRole("switch", { name: /toggle active/i }),
+		).toBeChecked();
+
+		await user.click(
+			screen.getByRole("switch", {
+				name: /toggle active/i,
+			}),
+		);
+		expect(screen.getByText("Concurrency limit updated")).toBeVisible();
+		rerender(
+			<Table
+				data={[{ ...MOCK_ROW, active: false }]}
+				onDeleteRow={vi.fn()}
+				onEditRow={vi.fn()}
+				searchValue=""
+				onSearchChange={vi.fn()}
+			/>,
+		);
+
+		expect(
+			screen.getByRole("switch", { name: /toggle active/i }),
+		).not.toBeChecked();
 	});
 });
diff --git a/ui-v2/src/components/concurrency/task-run-concurrenct-view/data-table/data-table.test.tsx b/ui-v2/src/components/concurrency/task-run-concurrenct-view/data-table/data-table.test.tsx
index b15331a57494..85961acb10e5 100644
--- a/ui-v2/src/components/concurrency/task-run-concurrenct-view/data-table/data-table.test.tsx
+++ b/ui-v2/src/components/concurrency/task-run-concurrenct-view/data-table/data-table.test.tsx
@@ -3,22 +3,20 @@ import userEvent from "@testing-library/user-event";
 import { describe, expect, it, vi } from "vitest";
 import { Table } from "./data-table";
 
-const MOCK_DATA = [
-	{
-		id: "0",
-		created: "2021-01-01T00:00:00Z",
-		updated: "2021-01-01T00:00:00Z",
-		tag: "my tag 0",
-		concurrency_limit: 1,
-		active_slots: [] as Array<string>,
-	},
-];
+const MOCK_ROW = {
+	id: "0",
+	created: "2021-01-01T00:00:00Z",
+	updated: "2021-01-01T00:00:00Z",
+	tag: "my tag 0",
+	concurrency_limit: 1,
+	active_slots: [] as Array<string>,
+};
 
 describe("TaskRunDataTable -- table", () => {
 	it("renders row data", () => {
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={vi.fn()}
 				onResetRow={vi.fn()}
 				searchValue=""
@@ -35,7 +33,7 @@ describe("TaskRunDataTable -- table", () => {
 
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={mockFn}
 				onResetRow={vi.fn()}
 				searchValue=""
@@ -46,7 +44,7 @@ describe("TaskRunDataTable -- table", () => {
 			screen.getByRole("button", { name: /open menu/i, hidden: true }),
 		);
 		await user.click(screen.getByRole("menuitem", { name: /delete/i }));
-		expect(mockFn).toHaveBeenCalledOnce();
+		expect(mockFn).toHaveBeenCalledWith(MOCK_ROW);
 	});
 	it("calls onReset upon clicking rest action menu item", async () => {
 		const user = userEvent.setup();
@@ -54,7 +52,7 @@ describe("TaskRunDataTable -- table", () => {
 
 		render(
 			<Table
-				data={MOCK_DATA}
+				data={[MOCK_ROW]}
 				onDeleteRow={vi.fn()}
 				onResetRow={mockFn}
 				searchValue=""
@@ -65,6 +63,6 @@ describe("TaskRunDataTable -- table", () => {
 			screen.getByRole("button", { name: /open menu/i, hidden: true }),
 		);
 		await user.click(screen.getByRole("menuitem", { name: /reset/i }));
-		expect(mockFn).toHaveBeenCalledOnce();
+		expect(mockFn).toHaveBeenCalledWith(MOCK_ROW);
 	});
 });