fix: 修复 useAuth 测试 mock 支持 getState() 方法

2026-08-17 14:48:44 +08:00
1 changed files with 21 additions and 8 deletions
@@ -31,14 +31,27 @@ vi.mock("react-router-dom", async () => {
 })

 vi.mock("@/store/authStore", () => ({
-  useAuthStore: (selector: any) =>
-    selector({
-      user: { id: "1", username: "testuser" },
-      token: "mock-token",
-      isAuthenticated: true,
-      setAuth: mockSetAuth,
-      clearAuth: mockClearAuth,
-    }),
+  useAuthStore: Object.assign(
+    (selector: any) =>
+      selector({
+        user: { id: "1", username: "testuser" },
+        token: "mock-token",
+        refreshToken: "mock-refresh-token",
+        isAuthenticated: true,
+        setAuth: mockSetAuth,
+        clearAuth: mockClearAuth,
+      }),
+    {
+      getState: () => ({
+        user: { id: "1", username: "testuser" },
+        token: "mock-token",
+        refreshToken: "mock-refresh-token",
+        isAuthenticated: true,
+        setAuth: mockSetAuth,
+        clearAuth: mockClearAuth,
+      }),
+    },
+  ),
 }))

 vi.mock("@tanstack/react-query", () => ({