📦 EqualifyEverything / equalify-dashboard

📄 protected.test.tsx · 94 lines
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
90
91
92
93
94import { describe, expect, it, vi } from 'vitest';

import * as useAuthModule from '~/hooks/useAuth';
import { Account, Login, Protected } from '~/routes';
import { render, screen } from '../../customRender';

vi.mock('~/hooks/useAuth', () => ({
  useAuth: vi.fn(),
}));

const mockUseAuth = (
  overrides: Partial<ReturnType<typeof useAuthModule.useAuth>>,
) => {
  vi.mocked(useAuthModule.useAuth).mockReturnValue({
    signUp: vi.fn(),
    confirmSignUp: vi.fn(),
    resendSignUpCode: vi.fn(),
    signIn: vi.fn(),
    signOut: vi.fn(),
    deleteUser: vi.fn(),
    user: null,
    isAuthenticated: false,
    needsConfirmation: false,
    pendingUsername: null,
    loading: false,
    error: null,
    ...overrides,
  });
};

describe('Protected Routes', () => {
  describe('Account Page', () => {
    it('redirects to login when user is not authenticated', async () => {
      mockUseAuth({ isAuthenticated: false });

      render(<Account />, {
        routeOptions: {
          routes: [
            {
              path: '/account',
              element: (
                <Protected>
                  <Account />
                </Protected>
              ),
            },
            {
              path: '/login',
              element: <Login />,
            },
          ],
          initialEntries: ['/account'],
        },
      });

      const heading = await screen.findByRole('heading', {
        name: /login/i,
        level: 1,
      });
      expect(heading).toBeInTheDocument();
    });

    it('renders the account page when user is authenticated', async () => {
      mockUseAuth({ isAuthenticated: true });

      render(<Account />, {
        routeOptions: {
          routes: [
            {
              path: '/account',
              element: (
                <Protected>
                  <Account />
                </Protected>
              ),
            },
            {
              path: '/login',
              element: <Login />,
            },
          ],
          initialEntries: ['/account'],
        },
      });

      const heading = await screen.findByRole('heading', {
        name: /your account/i,
        level: 1,
      });
      expect(heading).toBeInTheDocument();
    });
  });
});