docs: audit reports + feature docs + skills + admin-v2 + config sync
Some checks failed
CI / Lint & TypeCheck (push) Has been cancelled
CI / Unit Tests (push) Has been cancelled
CI / Build Frontend (push) Has been cancelled
CI / Rust Check (push) Has been cancelled
CI / Security Scan (push) Has been cancelled
CI / E2E Tests (push) Has been cancelled

Update audit tracker, roadmap, architecture docs,
add admin-v2 Roles page + Billing tests,
sync CLAUDE.md, Cargo.toml, docker-compose.yml,
add deep-research / frontend-design / chart-visualization skills

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
iven
2026-04-02 19:25:00 +08:00
parent 28299807b6
commit 8898bb399e
48 changed files with 7388 additions and 173 deletions

View File

@@ -0,0 +1,219 @@
// ============================================================
// Config 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Config from '@/pages/Config'
// ── Mock data ────────────────────────────────────────────────
const mockConfigItems = [
{
id: 'cfg-001',
category: 'general',
key_path: 'general.app_name',
value_type: 'string',
current_value: 'ZCLAW',
default_value: 'ZCLAW',
source: 'database',
description: '应用程序名称',
requires_restart: false,
created_at: '2026-01-01T00:00:00Z',
updated_at: '2026-01-01T00:00:00Z',
},
{
id: 'cfg-002',
category: 'general',
key_path: 'general.debug_mode',
value_type: 'boolean',
current_value: 'false',
default_value: 'false',
source: 'default',
description: '调试模式开关',
requires_restart: true,
created_at: '2026-01-01T00:00:00Z',
updated_at: '2026-01-01T00:00:00Z',
},
{
id: 'cfg-003',
category: 'general',
key_path: 'general.max_connections',
value_type: 'integer',
current_value: null,
default_value: '100',
source: 'default',
description: '最大连接数',
requires_restart: false,
created_at: '2026-01-01T00:00:00Z',
updated_at: '2026-01-01T00:00:00Z',
},
]
const mockResponse = {
items: mockConfigItems,
total: 3,
page: 1,
page_size: 50,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Config page', () => {
it('renders page header', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
expect(screen.getByText('系统配置')).toBeInTheDocument()
expect(screen.getByText('管理系统运行参数和功能开关')).toBeInTheDocument()
})
it('fetches and displays config items', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
await waitFor(() => {
expect(screen.getByText('general.app_name')).toBeInTheDocument()
})
expect(screen.getByText('general.debug_mode')).toBeInTheDocument()
})
it('shows loading spinner while fetching', async () => {
server.use(
http.get('*/api/v1/config/items', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
// Ant Design Spin component renders a .ant-spin element
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('general.app_name')).toBeInTheDocument()
})
})
it('shows error state on API failure', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '服务器内部错误' },
{ status: 500 },
)
}),
)
renderWithProviders(<Config />)
// Config page does not have a dedicated ErrorState; the ProTable simply
// renders empty when the query fails. We verify the page header is still
// rendered and the table body has no data rows (shows "暂无数据").
await waitFor(() => {
const emptyElements = screen.queryAllByText('暂无数据')
expect(emptyElements.length).toBeGreaterThanOrEqual(1)
})
// Page header is still present even on error
expect(screen.getByText('系统配置')).toBeInTheDocument()
})
it('renders config key_path and current_value columns', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
// key_path values are rendered in <code> elements
await waitFor(() => {
expect(screen.getByText('general.app_name')).toBeInTheDocument()
})
expect(screen.getByText('general.debug_mode')).toBeInTheDocument()
// current_value "ZCLAW" appears in both the current_value column and default_value column
const zclawElements = screen.getAllByText('ZCLAW')
expect(zclawElements.length).toBeGreaterThanOrEqual(1)
})
it('renders requires_restart column with tags', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
await waitFor(() => {
expect(screen.getByText('general.app_name')).toBeInTheDocument()
})
// requires_restart=true renders "是" (orange tag)
expect(screen.getByText('是')).toBeInTheDocument()
// requires_restart=false renders "否" (may appear multiple times for two items)
const noTags = screen.getAllByText('否')
expect(noTags.length).toBeGreaterThanOrEqual(1)
})
it('renders category tabs', async () => {
server.use(
http.get('*/api/v1/config/items', () => {
return HttpResponse.json(mockResponse)
}),
)
renderWithProviders(<Config />)
expect(screen.getByText('通用')).toBeInTheDocument()
expect(screen.getByText('认证')).toBeInTheDocument()
expect(screen.getByText('中转')).toBeInTheDocument()
expect(screen.getByText('模型')).toBeInTheDocument()
})
})

View File

@@ -0,0 +1,242 @@
// ============================================================
// Dashboard 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Dashboard from '@/pages/Dashboard'
// ── Mock data ────────────────────────────────────────────────
const mockStats = {
total_accounts: 12,
active_accounts: 8,
tasks_today: 156,
active_providers: 3,
active_models: 7,
tokens_today_input: 24000,
tokens_today_output: 8500,
}
const mockLogs = {
items: [
{
id: 1,
account_id: 'acc-001',
action: 'login',
target_type: 'account',
target_id: 'acc-001',
details: null,
ip_address: '192.168.1.1',
created_at: '2026-03-30T10:00:00Z',
},
{
id: 2,
account_id: 'acc-002',
action: 'create_provider',
target_type: 'provider',
target_id: 'prov-001',
details: { name: 'OpenAI' },
ip_address: '10.0.0.1',
created_at: '2026-03-30T09:30:00Z',
},
],
total: 2,
page: 1,
page_size: 10,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Dashboard page', () => {
it('renders page header', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
expect(screen.getByText('仪表盘')).toBeInTheDocument()
expect(screen.getByText('系统概览与最近活动')).toBeInTheDocument()
})
it('renders stat cards with correct values', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
await waitFor(() => {
expect(screen.getByText('12')).toBeInTheDocument()
})
// Stat titles
expect(screen.getByText('总账号')).toBeInTheDocument()
expect(screen.getByText('活跃服务商')).toBeInTheDocument()
expect(screen.getByText('活跃模型')).toBeInTheDocument()
expect(screen.getByText('今日请求')).toBeInTheDocument()
expect(screen.getByText('今日 Token')).toBeInTheDocument()
// Token total: 24000 + 8500 = 32500
expect(screen.getByText('32,500')).toBeInTheDocument()
})
it('renders recent logs table with action labels', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
// Wait for action labels from constants/status.ts
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
expect(screen.getByText('创建服务商')).toBeInTheDocument()
})
it('renders target types in logs table', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
expect(screen.getByText('account')).toBeInTheDocument()
expect(screen.getByText('provider')).toBeInTheDocument()
})
it('shows loading spinner before stats load', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
// Ant Design Spin component renders a .ant-spin element
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('总账号')).toBeInTheDocument()
})
})
it('shows error state when stats request fails', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '服务器内部错误' },
{ status: 500 },
)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
await waitFor(() => {
expect(screen.getByText('服务器内部错误')).toBeInTheDocument()
})
})
it('renders stat cards with zero values when stats are null', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json({})
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json({ items: [], total: 0, page: 1, page_size: 10 })
}),
)
renderWithProviders(<Dashboard />)
// All stats should fallback to 0
await waitFor(() => {
const zeros = screen.getAllByText('0')
expect(zeros.length).toBeGreaterThanOrEqual(2)
})
})
it('renders recent logs section header', async () => {
server.use(
http.get('*/api/v1/stats/dashboard', () => {
return HttpResponse.json(mockStats)
}),
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Dashboard />)
await waitFor(() => {
expect(screen.getByText('最近操作日志')).toBeInTheDocument()
})
})
})

View File

@@ -0,0 +1,219 @@
// ============================================================
// Login 页面测试
// ============================================================
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import userEvent from '@testing-library/user-event'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import { MemoryRouter } from 'react-router-dom'
import Login from '@/pages/Login'
// ── Mock data ────────────────────────────────────────────────
const mockLoginResponse = {
token: 'jwt-token-123',
refresh_token: 'refresh-token-456',
account: {
id: 'acc-001',
username: 'testadmin',
email: 'admin@zclaw.ai',
display_name: 'Admin',
role: 'super_admin',
status: 'active',
totp_enabled: false,
last_login_at: null,
created_at: '2026-01-01T00:00:00Z',
llm_routing: 'relay',
},
}
const mockAccount = {
id: 'acc-001',
username: 'testadmin',
email: 'admin@zclaw.ai',
display_name: 'Admin',
role: 'super_admin',
status: 'active',
totp_enabled: false,
last_login_at: null,
created_at: '2026-01-01T00:00:00Z',
llm_routing: 'relay',
}
// ── Hoisted mocks ────────────────────────────────────────────
const { mockLogin, mockNavigate, mockAuthServiceLogin } = vi.hoisted(() => ({
mockLogin: vi.fn(),
mockNavigate: vi.fn(),
mockAuthServiceLogin: vi.fn(),
}))
vi.mock('@/stores/authStore', () => ({
useAuthStore: Object.assign(
vi.fn((selector: (s: Record<string, unknown>) => unknown) =>
selector({ login: mockLogin }),
),
{ getState: () => ({ token: null, refreshToken: null, logout: vi.fn() }) },
),
}))
vi.mock('@/services/auth', () => ({
authService: {
login: mockAuthServiceLogin,
},
}))
vi.mock('react-router-dom', async () => {
const actual = await vi.importActual<typeof import('react-router-dom')>('react-router-dom')
return {
...actual,
useNavigate: () => mockNavigate,
}
})
beforeEach(() => {
mockLogin.mockClear()
mockNavigate.mockClear()
mockAuthServiceLogin.mockClear()
})
// ── Helper: render with providers ────────────────────────────
function renderLogin(initialEntries = ['/login']) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
<MemoryRouter initialEntries={initialEntries}>
<Login />
</MemoryRouter>
</QueryClientProvider>,
)
}
/** Click the LoginForm submit button (Ant Design renders "登 录" with a space) */
function getSubmitButton(): HTMLElement {
const btn = document.querySelector<HTMLButtonElement>(
'button.ant-btn-primary[type="button"]',
)
if (!btn) throw new Error('Submit button not found')
return btn
}
// ── Tests ────────────────────────────────────────────────────
describe('Login page', () => {
it('renders the login form with username and password fields', () => {
renderLogin()
expect(screen.getByText('登录到 ZCLAW')).toBeInTheDocument()
expect(screen.getByPlaceholderText('请输入用户名')).toBeInTheDocument()
expect(screen.getByPlaceholderText('请输入密码')).toBeInTheDocument()
const submitButton = getSubmitButton()
expect(submitButton).toBeTruthy()
})
it('shows the ZCLAW brand logo', () => {
renderLogin()
expect(screen.getByText('Z')).toBeInTheDocument()
expect(screen.getByText(/ZCLAW Admin/)).toBeInTheDocument()
})
it('successful login calls authStore.login and navigates to /', async () => {
const user = userEvent.setup()
mockAuthServiceLogin.mockResolvedValue(mockLoginResponse)
renderLogin()
await user.type(screen.getByPlaceholderText('请输入用户名'), 'testadmin')
await user.type(screen.getByPlaceholderText('请输入密码'), 'password123')
await user.click(getSubmitButton())
await waitFor(() => {
expect(mockLogin).toHaveBeenCalledWith(
'jwt-token-123',
'refresh-token-456',
mockAccount,
)
})
expect(mockNavigate).toHaveBeenCalledWith('/', { replace: true })
})
it('navigates to redirect path after login', async () => {
const user = userEvent.setup()
mockAuthServiceLogin.mockResolvedValue(mockLoginResponse)
renderLogin(['/login?from=/settings'])
await user.type(screen.getByPlaceholderText('请输入用户名'), 'testadmin')
await user.type(screen.getByPlaceholderText('请输入密码'), 'password123')
await user.click(getSubmitButton())
await waitFor(() => {
expect(mockNavigate).toHaveBeenCalledWith('/settings', { replace: true })
})
})
it('shows TOTP field when server returns TOTP-related error', async () => {
const user = userEvent.setup()
const error = new Error('请输入两步验证码 (TOTP)')
Object.assign(error, { status: 403 })
mockAuthServiceLogin.mockRejectedValue(error)
renderLogin()
// Initially no TOTP field
expect(screen.queryByPlaceholderText('请输入 6 位验证码')).not.toBeInTheDocument()
await user.type(screen.getByPlaceholderText('请输入用户名'), 'testadmin')
await user.type(screen.getByPlaceholderText('请输入密码'), 'password123')
await user.click(getSubmitButton())
// After TOTP error, TOTP field appears
await waitFor(() => {
expect(screen.getByPlaceholderText('请输入 6 位验证码')).toBeInTheDocument()
})
})
it('shows error message on invalid credentials', async () => {
const user = userEvent.setup()
const error = new Error('用户名或密码错误')
mockAuthServiceLogin.mockRejectedValue(error)
renderLogin()
await user.type(screen.getByPlaceholderText('请输入用户名'), 'wrong')
await user.type(screen.getByPlaceholderText('请输入密码'), 'wrong')
await user.click(getSubmitButton())
await waitFor(() => {
expect(screen.getByText('用户名或密码错误')).toBeInTheDocument()
})
})
it('does not call authStore.login on failed login', async () => {
const user = userEvent.setup()
const error = new Error('用户名或密码错误')
mockAuthServiceLogin.mockRejectedValue(error)
renderLogin()
await user.type(screen.getByPlaceholderText('请输入用户名'), 'wrong')
await user.type(screen.getByPlaceholderText('请输入密码'), 'wrong')
await user.click(getSubmitButton())
await waitFor(() => {
expect(screen.getByText('用户名或密码错误')).toBeInTheDocument()
})
expect(mockLogin).not.toHaveBeenCalled()
expect(mockNavigate).not.toHaveBeenCalled()
})
})

View File

@@ -0,0 +1,210 @@
// ============================================================
// Logs 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Logs from '@/pages/Logs'
// ── Mock data ────────────────────────────────────────────────
const mockLogs = {
items: [
{
id: 1,
account_id: 'acc-001',
action: 'login',
target_type: 'account',
target_id: 'acc-001',
details: null,
ip_address: '192.168.1.1',
created_at: '2026-03-30T10:00:00Z',
},
{
id: 2,
account_id: 'acc-002',
action: 'create_provider',
target_type: 'provider',
target_id: 'prov-001',
details: { name: 'OpenAI' },
ip_address: '10.0.0.1',
created_at: '2026-03-30T09:30:00Z',
},
{
id: 3,
account_id: 'acc-001',
action: 'delete_model',
target_type: 'model',
target_id: 'mdl-001',
details: null,
ip_address: '192.168.1.1',
created_at: '2026-03-29T14:00:00Z',
},
],
total: 3,
page: 1,
page_size: 20,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Logs page', () => {
it('renders page header', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
expect(screen.getByText('操作日志')).toBeInTheDocument()
expect(screen.getByText('系统审计与操作记录')).toBeInTheDocument()
})
it('fetches and displays log entries', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
// Wait for action labels rendered from constants/status.ts
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
expect(screen.getByText('创建服务商')).toBeInTheDocument()
expect(screen.getByText('删除模型')).toBeInTheDocument()
})
it('shows loading spinner while fetching', async () => {
server.use(
http.get('*/api/v1/logs/operations', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
// Ant Design Spin component renders a .ant-spin element
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
})
it('shows ErrorState on API failure with retry button', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '服务器内部错误' },
{ status: 500 },
)
}),
)
renderWithProviders(<Logs />)
// ErrorState renders the error message
await waitFor(() => {
expect(screen.getByText('服务器内部错误')).toBeInTheDocument()
})
// Ant Design Button splits two-character text with a space: "重 试"
const retryButton = screen.getByRole('button', { name: /重.?试/ })
expect(retryButton).toBeInTheDocument()
})
it('renders action as a colored tag', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
// Verify the action tags have the correct Ant Design color classes
const loginTag = screen.getByText('登录').closest('.ant-tag')
expect(loginTag).toBeTruthy()
// actionColors.login = 'green' → Ant Design renders ant-tag-green or ant-tag-color-green
expect(loginTag?.className).toMatch(/green/)
})
it('renders IP address column', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
// 192.168.1.1 appears twice (two log entries from the same IP)
const ip1Elements = screen.getAllByText('192.168.1.1')
expect(ip1Elements.length).toBeGreaterThanOrEqual(1)
expect(screen.getByText('10.0.0.1')).toBeInTheDocument()
})
it('renders target_type column', async () => {
server.use(
http.get('*/api/v1/logs/operations', () => {
return HttpResponse.json(mockLogs)
}),
)
renderWithProviders(<Logs />)
await waitFor(() => {
expect(screen.getByText('登录')).toBeInTheDocument()
})
expect(screen.getByText('account')).toBeInTheDocument()
expect(screen.getByText('provider')).toBeInTheDocument()
expect(screen.getByText('model')).toBeInTheDocument()
})
})

View File

@@ -0,0 +1,184 @@
// ============================================================
// ModelServices 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import ModelServices from '@/pages/ModelServices'
// ── Mock data ────────────────────────────────────────────────
const mockProviders = {
items: [
{
id: 'prov-001',
name: 'openai',
display_name: 'OpenAI',
base_url: 'https://api.openai.com/v1',
api_protocol: 'openai',
enabled: true,
rate_limit_rpm: 500,
rate_limit_tpm: null,
created_at: '2026-01-01T00:00:00Z',
updated_at: '2026-03-15T10:00:00Z',
},
{
id: 'prov-002',
name: 'anthropic',
display_name: 'Anthropic',
base_url: 'https://api.anthropic.com',
api_protocol: 'anthropic',
enabled: false,
rate_limit_rpm: 200,
rate_limit_tpm: null,
created_at: '2026-02-01T00:00:00Z',
updated_at: '2026-03-01T00:00:00Z',
},
{
id: 'prov-003',
name: 'deepseek',
display_name: 'DeepSeek',
base_url: 'https://api.deepseek.com/v1',
api_protocol: 'openai',
enabled: true,
rate_limit_rpm: null,
rate_limit_tpm: null,
created_at: '2026-03-01T00:00:00Z',
updated_at: '2026-03-01T00:00:00Z',
},
],
total: 3,
page: 1,
page_size: 20,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('ModelServices page', () => {
it('renders page header', async () => {
server.use(
http.get('*/api/v1/providers', () => {
return HttpResponse.json(mockProviders)
}),
)
renderWithProviders(<ModelServices />)
expect(screen.getByText('模型服务')).toBeInTheDocument()
expect(screen.getByText('管理 AI 服务商、模型配置和 Key 池')).toBeInTheDocument()
})
it('fetches and displays providers', async () => {
server.use(
http.get('*/api/v1/providers', () => {
return HttpResponse.json(mockProviders)
}),
)
renderWithProviders(<ModelServices />)
await waitFor(() => {
expect(screen.getByText('OpenAI')).toBeInTheDocument()
})
expect(screen.getByText('Anthropic')).toBeInTheDocument()
expect(screen.getByText('DeepSeek')).toBeInTheDocument()
// Provider identifiers rendered as code
// openai also appears in base_url, so use getAllByText
expect(screen.getAllByText('openai').length).toBeGreaterThanOrEqual(1)
expect(screen.getAllByText('anthropic').length).toBeGreaterThanOrEqual(1)
expect(screen.getAllByText('deepseek').length).toBeGreaterThanOrEqual(1)
})
it('shows loading spinner before data arrives', async () => {
server.use(
http.get('*/api/v1/providers', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockProviders)
}),
)
renderWithProviders(<ModelServices />)
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('OpenAI')).toBeInTheDocument()
})
})
it('renders provider status as tag', async () => {
server.use(
http.get('*/api/v1/providers', () => {
return HttpResponse.json(mockProviders)
}),
)
renderWithProviders(<ModelServices />)
await waitFor(() => {
expect(screen.getByText('OpenAI')).toBeInTheDocument()
})
// enabled: true -> "启用" tag, enabled: false -> "禁用" tag
const enabledTags = screen.getAllByText('启用')
expect(enabledTags.length).toBe(2) // openai + deepseek
expect(screen.getByText('禁用')).toBeInTheDocument() // anthropic
})
it('shows empty table on API failure', async () => {
server.use(
http.get('*/api/v1/providers', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '获取服务商列表失败' },
{ status: 500 },
)
}),
)
renderWithProviders(<ModelServices />)
// Page header should still render
expect(screen.getByText('模型服务')).toBeInTheDocument()
// Provider names should NOT be rendered
await waitFor(() => {
expect(screen.queryByText('OpenAI')).not.toBeInTheDocument()
})
})
})

View File

@@ -0,0 +1,178 @@
// ============================================================
// Prompts 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Prompts from '@/pages/Prompts'
// ── Mock data ────────────────────────────────────────────────
const mockPrompts = {
items: [
{
id: 'pt-001',
name: 'system-default',
category: 'system',
description: 'Default system prompt for all agents',
source: 'builtin' as const,
current_version: 3,
status: 'active' as const,
created_at: '2026-01-15T08:00:00Z',
updated_at: '2026-03-20T12:00:00Z',
},
{
id: 'pt-002',
name: 'custom-research',
category: 'tool',
description: 'Custom research prompt template',
source: 'custom' as const,
current_version: 1,
status: 'active' as const,
created_at: '2026-03-01T10:00:00Z',
updated_at: '2026-03-01T10:00:00Z',
},
{
id: 'pt-003',
name: 'legacy-summary',
category: 'system',
description: 'Legacy summary prompt',
source: 'builtin' as const,
current_version: 5,
status: 'archived' as const,
created_at: '2025-06-01T00:00:00Z',
updated_at: '2026-02-28T00:00:00Z',
},
],
total: 3,
page: 1,
page_size: 20,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Prompts page', () => {
it('renders page title and create button', async () => {
server.use(
http.get('*/api/v1/prompts', () => {
return HttpResponse.json(mockPrompts)
}),
)
renderWithProviders(<Prompts />)
expect(screen.getByText('提示词管理')).toBeInTheDocument()
expect(screen.getByText('管理系统提示词模板和版本历史')).toBeInTheDocument()
expect(screen.getByText('新建提示词')).toBeInTheDocument()
})
it('fetches and displays prompt templates', async () => {
server.use(
http.get('*/api/v1/prompts', () => {
return HttpResponse.json(mockPrompts)
}),
)
renderWithProviders(<Prompts />)
await waitFor(() => {
expect(screen.getByText('system-default')).toBeInTheDocument()
})
expect(screen.getByText('custom-research')).toBeInTheDocument()
expect(screen.getByText('legacy-summary')).toBeInTheDocument()
// Category "tool" appears once in data
expect(screen.getByText('tool')).toBeInTheDocument()
})
it('shows loading spinner before data arrives', async () => {
server.use(
http.get('*/api/v1/prompts', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockPrompts)
}),
)
renderWithProviders(<Prompts />)
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('system-default')).toBeInTheDocument()
})
})
it('renders source as tag with correct labels', async () => {
server.use(
http.get('*/api/v1/prompts', () => {
return HttpResponse.json(mockPrompts)
}),
)
renderWithProviders(<Prompts />)
await waitFor(() => {
expect(screen.getByText('system-default')).toBeInTheDocument()
})
// sourceLabels: { builtin: '内置', custom: '自定义' }
// '内置' appears twice (2 builtin items), '自定义' appears once
const builtinTags = screen.getAllByText('内置')
expect(builtinTags.length).toBe(2)
expect(screen.getByText('自定义')).toBeInTheDocument()
})
it('shows error state on API failure', async () => {
server.use(
http.get('*/api/v1/prompts', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '获取提示词列表失败' },
{ status: 500 },
)
}),
)
renderWithProviders(<Prompts />)
// React Query error propagation: ProTable receives empty data
// but the query error should be visible via the table state
// Check that no prompt names are rendered
await waitFor(() => {
expect(screen.queryByText('system-default')).not.toBeInTheDocument()
})
})
})

View File

@@ -0,0 +1,234 @@
// ============================================================
// Relay 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Relay from '@/pages/Relay'
// ── Mock data ────────────────────────────────────────────────
const mockRelayTasks = {
items: [
{
id: 'task-001-abcdef',
account_id: 'acc-001',
provider_id: 'prov-001',
model_id: 'gpt-4o',
status: 'completed',
priority: 0,
attempt_count: 1,
max_attempts: 3,
input_tokens: 1500,
output_tokens: 800,
error_message: null,
queued_at: '2026-03-30T10:00:00Z',
started_at: '2026-03-30T10:00:01Z',
completed_at: '2026-03-30T10:00:05Z',
created_at: '2026-03-30T10:00:00Z',
},
{
id: 'task-002-ghijkl',
account_id: 'acc-002',
provider_id: 'prov-002',
model_id: 'claude-3.5-sonnet',
status: 'failed',
priority: 0,
attempt_count: 3,
max_attempts: 3,
input_tokens: 2000,
output_tokens: 0,
error_message: 'Rate limit exceeded',
queued_at: '2026-03-30T09:00:00Z',
started_at: '2026-03-30T09:00:01Z',
completed_at: '2026-03-30T09:01:00Z',
created_at: '2026-03-30T09:00:00Z',
},
{
id: 'task-003-mnopqr',
account_id: 'acc-001',
provider_id: 'prov-001',
model_id: 'gpt-4o-mini',
status: 'queued',
priority: 1,
attempt_count: 0,
max_attempts: 3,
input_tokens: 0,
output_tokens: 0,
error_message: null,
queued_at: '2026-03-30T11:00:00Z',
started_at: null,
completed_at: null,
created_at: '2026-03-30T11:00:00Z',
},
],
total: 3,
page: 1,
page_size: 20,
}
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Relay page', () => {
it('renders page header', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
expect(screen.getByText('中转任务')).toBeInTheDocument()
expect(screen.getByText('查看和管理 AI 模型中转请求')).toBeInTheDocument()
})
it('fetches and displays relay tasks', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
await waitFor(() => {
expect(screen.getByText('已完成')).toBeInTheDocument()
})
expect(screen.getByText('失败')).toBeInTheDocument()
expect(screen.getByText('排队中')).toBeInTheDocument()
})
it('shows loading spinner while fetching', async () => {
server.use(
http.get('*/api/v1/relay/tasks', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
// Ant Design Spin component renders a .ant-spin element
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('已完成')).toBeInTheDocument()
})
})
it('shows ErrorState on API failure with retry button', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '服务器内部错误' },
{ status: 500 },
)
}),
)
renderWithProviders(<Relay />)
await waitFor(() => {
expect(screen.getByText('服务器内部错误')).toBeInTheDocument()
})
// Ant Design Button splits two-character text with a space: "重 试"
const retryButton = screen.getByRole('button', { name: /重.?试/ })
expect(retryButton).toBeInTheDocument()
})
it('renders status as colored tag', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
await waitFor(() => {
expect(screen.getByText('已完成')).toBeInTheDocument()
})
// Verify the status tags have correct Ant Design color classes
const completedTag = screen.getByText('已完成').closest('.ant-tag')
expect(completedTag).toBeTruthy()
// statusColors.completed = 'green'
expect(completedTag?.className).toMatch(/green/)
const failedTag = screen.getByText('失败').closest('.ant-tag')
expect(failedTag).toBeTruthy()
// statusColors.failed = 'red'
expect(failedTag?.className).toMatch(/red/)
})
it('renders model_id column', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
await waitFor(() => {
expect(screen.getByText('已完成')).toBeInTheDocument()
})
expect(screen.getByText('gpt-4o')).toBeInTheDocument()
expect(screen.getByText('claude-3.5-sonnet')).toBeInTheDocument()
expect(screen.getByText('gpt-4o-mini')).toBeInTheDocument()
})
it('renders token count column', async () => {
server.use(
http.get('*/api/v1/relay/tasks', () => {
return HttpResponse.json(mockRelayTasks)
}),
)
renderWithProviders(<Relay />)
await waitFor(() => {
expect(screen.getByText('已完成')).toBeInTheDocument()
})
// Token (入/出): 1,500 / 800
expect(screen.getByText(/1,500 \/ 800/)).toBeInTheDocument()
// 2,000 / 0
expect(screen.getByText(/2,000 \/ 0/)).toBeInTheDocument()
})
})

View File

@@ -0,0 +1,248 @@
// ============================================================
// Usage 页面测试
// ============================================================
import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { render, screen, waitFor } from '@testing-library/react'
import { http, HttpResponse } from 'msw'
import { setupServer } from 'msw/node'
import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
import Usage from '@/pages/Usage'
// ── Mock data ────────────────────────────────────────────────
const mockDailyStats = [
{
day: '2026-03-28',
request_count: 120,
input_tokens: 24000,
output_tokens: 8000,
unique_devices: 5,
},
{
day: '2026-03-29',
request_count: 80,
input_tokens: 16000,
output_tokens: 5000,
unique_devices: 3,
},
{
day: '2026-03-30',
request_count: 200,
input_tokens: 40000,
output_tokens: 12000,
unique_devices: 7,
},
]
const mockModelStats = [
{
model_id: 'gpt-4o',
request_count: 300,
input_tokens: 60000,
output_tokens: 18000,
avg_latency_ms: 450.3,
success_rate: 0.98,
},
{
model_id: 'claude-sonnet-4-20250514',
request_count: 100,
input_tokens: 20000,
output_tokens: 7000,
avg_latency_ms: 620.7,
success_rate: 0.95,
},
]
// ── MSW server ───────────────────────────────────────────────
const server = setupServer()
beforeEach(() => {
server.listen({ onUnhandledRequest: 'bypass' })
})
afterEach(() => {
server.close()
})
// ── Helper: render with QueryClient ──────────────────────────
function renderWithProviders(ui: React.ReactElement) {
const queryClient = new QueryClient({
defaultOptions: {
queries: { retry: false },
},
})
return render(
<QueryClientProvider client={queryClient}>
{ui}
</QueryClientProvider>,
)
}
// ── Tests ────────────────────────────────────────────────────
describe('Usage page', () => {
it('renders page title and summary cards', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', () => {
return HttpResponse.json(mockDailyStats)
}),
http.get('*/api/v1/telemetry/stats', () => {
return HttpResponse.json(mockModelStats)
}),
)
renderWithProviders(<Usage />)
expect(screen.getByText('用量统计')).toBeInTheDocument()
expect(screen.getByText('查看模型使用情况和 Token 消耗')).toBeInTheDocument()
// Summary card titles
expect(screen.getByText('总请求数')).toBeInTheDocument()
expect(screen.getByText('总 Token 数')).toBeInTheDocument()
// Total requests: 120 + 80 + 200 = 400
await waitFor(() => {
expect(screen.getByText('400')).toBeInTheDocument()
})
// Total tokens: (24000+8000) + (16000+5000) + (40000+12000) = 105,000
expect(screen.getByText('105,000')).toBeInTheDocument()
})
it('fetches and displays daily stats table', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', () => {
return HttpResponse.json(mockDailyStats)
}),
http.get('*/api/v1/telemetry/stats', () => {
return HttpResponse.json(mockModelStats)
}),
)
renderWithProviders(<Usage />)
// Table column headers
expect(screen.getByText('每日统计')).toBeInTheDocument()
// Wait for data rows to render
await waitFor(() => {
expect(screen.getByText('2026-03-28')).toBeInTheDocument()
})
// Formatted request counts
expect(screen.getByText('120')).toBeInTheDocument()
expect(screen.getByText('80')).toBeInTheDocument()
expect(screen.getByText('200')).toBeInTheDocument()
// Device counts
expect(screen.getByText('5')).toBeInTheDocument()
})
it('fetches and displays model stats table', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', () => {
return HttpResponse.json(mockDailyStats)
}),
http.get('*/api/v1/telemetry/stats', () => {
return HttpResponse.json(mockModelStats)
}),
)
renderWithProviders(<Usage />)
expect(screen.getByText('按模型统计')).toBeInTheDocument()
await waitFor(() => {
expect(screen.getByText('gpt-4o')).toBeInTheDocument()
})
expect(screen.getByText('claude-sonnet-4-20250514')).toBeInTheDocument()
// Success rate: 0.98 -> "98.0%"
expect(screen.getByText('98.0%')).toBeInTheDocument()
// Avg latency: 450.3 -> "450ms"
expect(screen.getByText('450ms')).toBeInTheDocument()
})
it('shows loading spinner before data loads', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockDailyStats)
}),
http.get('*/api/v1/telemetry/stats', async () => {
await new Promise((resolve) => setTimeout(resolve, 500))
return HttpResponse.json(mockModelStats)
}),
)
renderWithProviders(<Usage />)
// Ant Design Spin component renders a .ant-spin element
const spinner = document.querySelector('.ant-spin')
expect(spinner).toBeTruthy()
// Wait for loading to complete so afterEach cleanup is clean
await waitFor(() => {
expect(screen.getByText('用量统计')).toBeInTheDocument()
})
})
it('shows ErrorState when daily stats request fails', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', () => {
return HttpResponse.json(
{ error: 'internal_error', message: '服务器内部错误' },
{ status: 500 },
)
}),
http.get('*/api/v1/telemetry/stats', () => {
return HttpResponse.json(mockModelStats)
}),
)
renderWithProviders(<Usage />)
await waitFor(() => {
expect(screen.getByText('服务器内部错误')).toBeInTheDocument()
})
// ErrorState renders a retry button (antd v6 may split Chinese characters)
expect(screen.getByRole('button', { name: /重.*试/ })).toBeInTheDocument()
})
it('calculates totals correctly from daily data', async () => {
server.use(
http.get('*/api/v1/telemetry/daily', () => {
return HttpResponse.json([
{
day: '2026-03-30',
request_count: 1500,
input_tokens: 10000,
output_tokens: 3000,
unique_devices: 2,
},
])
}),
http.get('*/api/v1/telemetry/stats', () => {
return HttpResponse.json([])
}),
)
renderWithProviders(<Usage />)
// Total requests: 1500 (formatted as "1,500" by Statistic)
await waitFor(() => {
const elements = screen.getAllByText('1,500')
expect(elements.length).toBeGreaterThanOrEqual(1)
})
// Total tokens: 10000 + 3000 = 13,000
expect(screen.getAllByText('13,000').length).toBeGreaterThanOrEqual(1)
})
})