From f0abe4135805edc8b5d1925cce104e194cdcae3b Mon Sep 17 00:00:00 2001
From: pablodanswer <pablo@danswer.ai>
Date: Mon, 2 Dec 2024 08:21:23 -0800
Subject: [PATCH] minor update

---
 web/tests/e2e/chat.spec.ts | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/web/tests/e2e/chat.spec.ts b/web/tests/e2e/chat.spec.ts
index e1e720f3dfd..e5d096ab0f6 100644
--- a/web/tests/e2e/chat.spec.ts
+++ b/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();
   }
 );