朱子纯 / test6

Commit 3826bed765e68c13ffee4ea5754cc72e184087a4

Authored by zichun
1 parent f3bba97e

feat(fe-login): 登录成功落地登录态与跳转 REQ-USR-004

Inline Side-by-side
Showing 1 changed file with 78 additions and 0 deletions
frontend/tests/unit/LoginPage.success.test.tsx 0 → 100644
  View file @3826bed
  1 +import { describe, it, expect, vi, beforeEach } from 'vitest';
  2 +import { screen, waitFor } from '@testing-library/react';
  3 +import userEvent from '@testing-library/user-event';
  4 +
  5 +// 捕获 navigate 调用
  6 +const navigateSpy = vi.fn();
  7 +vi.mock('react-router-dom', async () => {
  8 + const actual = await vi.importActual<typeof import('react-router-dom')>('react-router-dom');
  9 + return { ...actual, useNavigate: () => navigateSpy };
  10 +});
  11 +
  12 +// 捕获 message 调用：保留 antd 其余真实导出，仅桩 App.useApp 返回的 message
  13 +const messageSpy = { success: vi.fn(), error: vi.fn() };
  14 +vi.mock('antd', async () => {
  15 + const actual = await vi.importActual<typeof import('antd')>('antd');
  16 + return {
  17 + ...actual,
  18 + App: Object.assign(actual.App, { useApp: () => ({ message: messageSpy }) }),
  19 + };
  20 +});
  21 +
  22 +vi.mock('../../src/api/usrApi', () => ({
  23 + fetchCompanies: vi.fn(),
  24 + login: vi.fn(),
  25 +}));
  26 +
  27 +import { fetchCompanies, login } from '../../src/api/usrApi';
  28 +import LoginPage from '../../src/pages/usr/Login/LoginPage';
  29 +import { TOKEN_STORAGE_KEY } from '../../src/api/request';
  30 +import { renderWithProviders } from './renderLogin';
  31 +
  32 +const mockedFetch = fetchCompanies as unknown as ReturnType<typeof vi.fn>;
  33 +const mockedLogin = login as unknown as ReturnType<typeof vi.fn>;
  34 +
  35 +describe('LoginPage 登录成功', () => {
  36 + beforeEach(() => {
  37 + vi.clearAllMocks();
  38 + localStorage.clear();
  39 + mockedFetch.mockResolvedValue([{ id: 1, sCompanyName: '甲公司', sVersion: '标准版' }]);
  40 + });
  41 +
  42 + async function submitLogin() {
  43 + const user = userEvent.setup();
  44 + await user.type(screen.getByPlaceholderText('请输入你的用户名'), 'admin');
  45 + await user.type(screen.getByPlaceholderText('请输入你的密码'), 'secret');
  46 + await user.click(screen.getByRole('button', { name: /登\s*录/ }));
  47 + }
  48 +
  49 + it('dispatches setCredentials and persists token on success', async () => {
  50 + mockedLogin.mockResolvedValue({
  51 + token: 'tk',
  52 + user: { id: 1, sUserName: 'admin', sUserType: '超级管理员', sLanguage: '中文' },
  53 + });
  54 + const { getState } = renderWithProviders(<LoginPage />);
  55 + expect(await screen.findByText('甲公司（标准版）')).toBeInTheDocument();
  56 + await submitLogin();
  57 + await waitFor(() => expect(getState().auth.token).toBe('tk'));
  58 + expect(getState().auth.user).toEqual({
  59 + id: 1,
  60 + sUserName: 'admin',
  61 + sUserType: '超级管理员',
  62 + sLanguage: '中文',
  63 + });
  64 + expect(localStorage.getItem(TOKEN_STORAGE_KEY)).toBe('tk');
  65 + });
  66 +
  67 + it('shows success message and navigates to / with replace', async () => {
  68 + mockedLogin.mockResolvedValue({
  69 + token: 'tk',
  70 + user: { id: 1, sUserName: 'admin', sUserType: '超级管理员', sLanguage: '中文' },
  71 + });
  72 + renderWithProviders(<LoginPage />);
  73 + expect(await screen.findByText('甲公司（标准版）')).toBeInTheDocument();
  74 + await submitLogin();
  75 + await waitFor(() => expect(messageSpy.success).toHaveBeenCalledWith('登录成功'));
  76 + expect(navigateSpy).toHaveBeenCalledWith('/', { replace: true });
  77 + });
  78 +});