fix(test): 修复useAuth测试中mock setAuth不写localStorage的问题

2026-07-21 13:16:35 +08:00
parent 5e3a3099cf
commit 7a4ffcf1d4
1 changed files with 12 additions and 2 deletions
@@ -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()