From d8451325ccf037115cc3049a6ad783fbc8fd9ffd Mon Sep 17 00:00:00 2001
From: Nate Kelley <nate@buster.so>
Date: Mon, 11 Aug 2025 15:52:45 -0600
Subject: [PATCH] Update vitest.setup.ts

---
 apps/web/vitest.setup.ts | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/apps/web/vitest.setup.ts b/apps/web/vitest.setup.ts
index 0176c8628..6640c594f 100644
--- a/apps/web/vitest.setup.ts
+++ b/apps/web/vitest.setup.ts
@@ -61,14 +61,14 @@ vi.mock('remark-gfm', () => ({
 }));
 
 // Mock Supabase client to prevent environment variable errors in tests
-vi.mock('@/lib/supabase/client', () => ({
-  createBrowserClient: vi.fn(() => ({
+vi.mock('@/lib/supabase/client', () => {
+  const mockClient = {
     auth: {
       refreshSession: vi.fn().mockResolvedValue({
         data: {
           session: {
             access_token: 'mock-token',
-            expires_at: Date.now() / 1000 + 3600 // 1 hour from now
+            expires_at: Math.floor(Date.now() / 1000) + 3600 // 1 hour from now
           }
         },
         error: null
@@ -76,7 +76,24 @@ vi.mock('@/lib/supabase/client', () => ({
       getUser: vi.fn().mockResolvedValue({
         data: { user: null },
         error: null
+      }),
+      getSession: vi.fn().mockResolvedValue({
+        data: {
+          session: {
+            access_token: 'mock-token',
+            expires_at: Math.floor(Date.now() / 1000) + 3600
+          }
+        },
+        error: null
+      }),
+      onAuthStateChange: vi.fn().mockReturnValue({
+        data: { subscription: { unsubscribe: vi.fn() } }
       })
     }
-  }))
-}));
+  };
+
+  return {
+    createBrowserClient: vi.fn(() => mockClient),
+    getBrowserClient: vi.fn(() => mockClient)
+  };
+});