RequireSuperAdmin.test.tsx
2.04 KB
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
import { describe, it, expect } from 'vitest';
import { render, screen } from '@testing-library/react';
import { MemoryRouter, Routes, Route } from 'react-router-dom';
import { Provider } from 'react-redux';
import { configureStore } from '@reduxjs/toolkit';
import authReducer, { setSession } from '../store/slices/authSlice';
import RequireSuperAdmin from './RequireSuperAdmin';
function makeStore(opts: { token?: string; userType?: 'NORMAL' | 'SUPER_ADMIN' } = {}) {
const store = configureStore({ reducer: { auth: authReducer } });
if (opts.token) {
store.dispatch(
setSession({
accessToken: opts.token,
userInfo: {
userId: 1,
username: 'alice',
userType: opts.userType ?? 'NORMAL',
language: 'zh-CN',
companyCode: 'HQ',
},
}),
);
}
return store;
}
function renderRoutes(store: ReturnType<typeof makeStore>, entry: string) {
return render(
<Provider store={store}>
<MemoryRouter initialEntries={[entry]}>
<Routes>
<Route
path="/users"
element={
<RequireSuperAdmin>
<div data-testid="admin-only">ADMIN</div>
</RequireSuperAdmin>
}
/>
<Route path="/login" element={<div data-testid="login">LOGIN</div>} />
</Routes>
</MemoryRouter>
</Provider>,
);
}
describe('RequireSuperAdmin', () => {
it('no token → redirects to /login', () => {
renderRoutes(makeStore(), '/users');
expect(screen.getByTestId('login')).toBeInTheDocument();
});
it('NORMAL user token → shows 403 Result', () => {
renderRoutes(makeStore({ token: 'jwt', userType: 'NORMAL' }), '/users');
expect(screen.getByTestId('forbidden-result')).toBeInTheDocument();
expect(screen.queryByTestId('admin-only')).toBeNull();
});
it('SUPER_ADMIN token → renders children', () => {
renderRoutes(makeStore({ token: 'jwt', userType: 'SUPER_ADMIN' }), '/users');
expect(screen.getByTestId('admin-only')).toBeInTheDocument();
});
});