fix: fix tests related to autocomplete changes

jaaydenh · jaaydenh · commit 418bd7684003 · 2025-12-19T18:27:21.000Z
diff --git a/site/e2e/helpers.ts b/site/e2e/helpers.ts
@@ -266,7 +266,7 @@ export const createTemplate = async (
 
 	// If the organization picker is present on the page, select the default
 	// organization.
-	const orgPicker = page.getByLabel("Belongs to *");
+	const orgPicker = page.getByTestId("organization-autocomplete");
 	const organizationsEnabled = await orgPicker.isVisible();
 	if (organizationsEnabled) {
 		if (orgName !== defaultOrganizationName) {
@@ -1233,7 +1233,7 @@ export async function createUser(
 
 	// If the organization picker is present on the page, select the default
 	// organization.
-	const orgPicker = page.getByLabel("Organization *");
+	const orgPicker = page.getByTestId("organization-autocomplete");
 	const organizationsEnabled = await orgPicker.isVisible();
 	if (organizationsEnabled) {
 		// The organization picker will be disabled if there is only one option.
diff --git a/site/src/pages/CreateUserPage/CreateUserForm.stories.tsx b/site/src/pages/CreateUserPage/CreateUserForm.stories.tsx
@@ -63,7 +63,8 @@ export const WithOrganizations: Story = {
 	},
 	play: async ({ canvasElement }) => {
 		const canvas = within(canvasElement);
-		await userEvent.click(canvas.getByLabelText("Organization *"));
+		const autocomplete = await canvas.findByTestId("organization-autocomplete");
+		await userEvent.click(autocomplete);
 	},
 };
 
