1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
/**
* @vitest-environment jsdom
*/
import { cleanup, render, screen } from "@testing-library/react";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { Router } from "wouter";
import { memoryLocation } from "wouter/memory-location";
import { apiClient } from "../api/client";
import { AuthProvider } from "../stores";
import { ProtectedRoute } from "./ProtectedRoute";
vi.mock("../api/client", () => ({
apiClient: {
login: vi.fn(),
logout: vi.fn(),
isAuthenticated: vi.fn(),
getTokens: vi.fn(),
},
ApiClientError: class ApiClientError extends Error {
constructor(
message: string,
public status: number,
public code?: string,
) {
super(message);
this.name = "ApiClientError";
}
},
}));
function renderWithRouter(path: string) {
const { hook } = memoryLocation({ path });
return render(
<Router hook={hook}>
<AuthProvider>
<ProtectedRoute>
<div data-testid="protected-content">Protected Content</div>
</ProtectedRoute>
</AuthProvider>
</Router>,
);
}
beforeEach(() => {
vi.clearAllMocks();
});
afterEach(() => {
cleanup();
vi.restoreAllMocks();
});
describe("ProtectedRoute", () => {
it("shows loading state while auth is loading", () => {
vi.mocked(apiClient.getTokens).mockReturnValue(null);
vi.mocked(apiClient.isAuthenticated).mockReturnValue(false);
// The AuthProvider initially sets isLoading to true, then false after checking tokens
// Since getTokens returns null, isLoading will quickly become false
renderWithRouter("/");
// After the initial check, the component should redirect since not authenticated
expect(screen.queryByTestId("protected-content")).toBeNull();
});
it("renders children when authenticated", () => {
vi.mocked(apiClient.getTokens).mockReturnValue({
accessToken: "access-token",
refreshToken: "refresh-token",
});
vi.mocked(apiClient.isAuthenticated).mockReturnValue(true);
renderWithRouter("/");
expect(screen.getByTestId("protected-content")).toBeDefined();
expect(screen.getByText("Protected Content")).toBeDefined();
});
it("redirects to login when not authenticated", () => {
vi.mocked(apiClient.getTokens).mockReturnValue(null);
vi.mocked(apiClient.isAuthenticated).mockReturnValue(false);
renderWithRouter("/");
// Should not show protected content
expect(screen.queryByTestId("protected-content")).toBeNull();
});
});
|
