diff --git a/apps/web/src/test/hooks/useAuth.test.tsx b/apps/web/src/test/hooks/useAuth.test.tsx index fd1430dc4..2f164d739 100644 --- a/apps/web/src/test/hooks/useAuth.test.tsx +++ b/apps/web/src/test/hooks/useAuth.test.tsx @@ -7,8 +7,18 @@ import { renderHook, act } from "@testing-library/react" import { MemoryRouter } from "react-router-dom" const mockNavigate = vi.fn() -const mockSetAuth = vi.fn() -const mockClearAuth = vi.fn() +const mockSetAuth = vi.fn((_user: any, accessToken: string, refreshToken?: string | null) => { + localStorage.setItem("access_token", accessToken) + if (refreshToken) { + localStorage.setItem("refresh_token", refreshToken) + } else { + localStorage.removeItem("refresh_token") + } +}) +const mockClearAuth = vi.fn(() => { + localStorage.removeItem("access_token") + localStorage.removeItem("refresh_token") +}) const mockMutateAsync = vi.fn() const mockQueryClear = vi.fn()