chore: update tests

jaaydenh · jaaydenh · commit f97261d5d42e · 2024-12-12T19:49:40.000Z
diff --git a/site/e2e/tests/deployment/idpOrgSync.spec.ts b/site/e2e/tests/deployment/idpOrgSync.spec.ts
@@ -78,7 +78,7 @@ test.describe("IdpOrgSyncPage", () => {
 		).toBeVisible();
 	});
 
-	test("toggle default organization assignment", async ({ page }) => {
+	test("toggle off default organization assignment", async ({ page }) => {
 		requiresLicense();
 		await page.goto("/deployment/idp-org-sync", {
 			waitUntil: "domcontentloaded",
@@ -89,6 +89,12 @@ test.describe("IdpOrgSyncPage", () => {
 		});
 		await toggle.click();
 
+		const dialog = page.getByRole("dialog");
+		await expect(dialog).toBeVisible();
+
+		await dialog.getByRole("button", { name: "Confirm" }).click();
+		await expect(dialog).not.toBeVisible();
+
 		await expect(
 			page.getByText("Organization sync settings updated."),
 		).toBeVisible();
diff --git a/site/src/pages/DeploymentSettingsPage/IdpOrgSyncPage/IdpOrgSyncPageView.stories.tsx b/site/src/pages/DeploymentSettingsPage/IdpOrgSyncPage/IdpOrgSyncPageView.stories.tsx
@@ -1,4 +1,5 @@
 import type { Meta, StoryObj } from "@storybook/react";
+import { userEvent, within } from "@storybook/test";
 import {
 	MockOrganization,
 	MockOrganization2,
@@ -48,3 +49,18 @@ export const MissingGroups: Story = {
 		organizationSyncSettings: MockOrganizationSyncSettings,
 	},
 };
+
+export const AssignDefaultOrgWarningDialog: Story = {
+	args: {
+		organizationSyncSettings: MockOrganizationSyncSettings,
+		organizations: [MockOrganization, MockOrganization2],
+	},
+	play: async ({ canvasElement }) => {
+		const canvas = within(canvasElement);
+		await userEvent.click(
+			canvas.getByRole("switch", {
+				name: "Assign Default Organization",
+			}),
+		);
+	},
+};
