Update Chromatic Tests

web/tests/e2e/admin_auth.setup.ts

@@ -6,9 +6,9 @@ import { TEST_CREDENTIALS } from "./constants";
 setup("authenticate", async ({ page }) => {
   const { email, password } = TEST_CREDENTIALS;
 
-  await page.goto("http://localhost:3000/search");
+  await page.goto("http://localhost:3000/chat");
 
-  await page.waitForURL("http://localhost:3000/auth/login?next=%2Fsearch");
+  await page.waitForURL("http://localhost:3000/auth/login?next=%2Fchat");
 
   await expect(page).toHaveTitle("Danswer");
 
@@ -18,7 +18,7 @@ setup("authenticate", async ({ page }) => {
   // Click the login button
   await page.click('button[type="submit"]');
 
-  await page.waitForURL("http://localhost:3000/search");
+  await page.waitForURL("http://localhost:3000/chat");
 
   await page.context().storageState({ path: "admin_auth.json" });
 });

web/tests/e2e/chat.spec.ts

@@ -8,12 +8,20 @@ test(
   async ({ page }, testInfo) => {
     // Test simple loading
     await page.goto("http://localhost:3000/chat");
-    await expect(page.locator("div.text-2xl").nth(0)).toHaveText("General");
-    await expect(page.getByRole("button", { name: "Search S" })).toHaveClass(
-      /text-text-application-untoggled/
-    );
-    await expect(page.getByRole("button", { name: "Chat D" })).toHaveClass(
-      /text-text-application-toggled/
-    );
+
+    // Check for the "General" text in the new UI element
+    await expect(
+      page.locator("div.flex.items-center span.font-bold")
+    ).toHaveText("General");
+
+    // Check for the presence of the new UI element
+    await expect(
+      page.locator("div.flex.justify-center div.bg-black.rounded-full")
+    ).toBeVisible();
+
+    // Check for the SVG icon
+    await expect(
+      page.locator("div.flex.justify-center svg.w-5.h-5")
+    ).toBeVisible();
   }
 );

web/tests/e2e/home.spec.ts

@@ -12,9 +12,9 @@ test(
     // Test redirect to login, and redirect to search after login
     const { email, password } = TEST_CREDENTIALS;
 
-    await page.goto("http://localhost:3000/search");
+    await page.goto("http://localhost:3000/chat");
 
-    await page.waitForURL("http://localhost:3000/auth/login?next=%2Fsearch");
+    await page.waitForURL("http://localhost:3000/auth/login?next=%2Fchat");
 
     await expect(page).toHaveTitle("Danswer");
 
@@ -26,6 +26,6 @@ test(
     // Click the login button
     await page.click('button[type="submit"]');
 
-    await page.waitForURL("http://localhost:3000/search");
+    await page.waitForURL("http://localhost:3000/chat");
   }
 );