test: 完善自动化测试体系（7个模块，73个任务）

系统性地修复、补充和强化项目的自动化测试能力： 1. 测试基础设施修复 - 修复 stubConcurrencyCache 缺失方法和构造函数参数不匹配 - 创建 testutil 共享包（stubs.go, fixtures.go, httptest.go） - 为所有 Stub 添加编译期接口断言 2. 中间件测试补充 - 新增 JWT 认证中间件测试（有效/过期/篡改/缺失 Token） - 补充 rate_limiter 和 recovery 中间件测试场景 3. 网关核心路径测试 - 新增账户选择、等待队列、流式响应、并发控制、计费、Claude Code 检测测试 - 覆盖负载均衡、粘性会话、SSE 转发、槽位管理等关键逻辑 4. 前端测试体系（11个新测试文件，163个测试用例） - Pinia stores: auth, app, subscriptions - API client: 请求拦截器、响应拦截器、401 刷新 - Router guards: 认证重定向、管理员权限、简易模式限制 - Composables: useForm, useTableLoader, useClipboard - Components: LoginForm, ApiKeyCreate, Dashboard 5. CI/CD 流水线重构 - 重构 backend-ci.yml 为统一的 ci.yml - 前后端 4 个并行 Job + Postgres/Redis services - Race 检测、覆盖率收集与门禁、Docker 构建验证 6. E2E 自动化测试 - e2e-test.sh 自动化脚本（Docker 启动→健康检查→测试→清理） - 用户注册→登录→API Key→网关调用完整链路测试 - Mock 模式和 API Key 脱敏支持 7. 修复预存问题 - tlsfingerprint dialer_test.go 缺失 build tag 导致集成测试编译冲突 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-08 12:05:39 +08:00
parent 53e1c8b268
commit bb5a5dd65e
41 changed files with 5101 additions and 182 deletions
--- a/frontend/src/stores/tests/auth.spec.ts
+++ b/frontend/src/stores/tests/auth.spec.ts
@@ -0,0 +1,289 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
+import { setActivePinia, createPinia } from 'pinia'
+import { useAuthStore } from '@/stores/auth'
+
+// Mock authAPI
+const mockLogin = vi.fn()
+const mockLogin2FA = vi.fn()
+const mockLogout = vi.fn()
+const mockGetCurrentUser = vi.fn()
+const mockRegister = vi.fn()
+const mockRefreshToken = vi.fn()
+
+vi.mock('@/api', () => ({
+  authAPI: {
+    login: (...args: any[]) => mockLogin(...args),
+    login2FA: (...args: any[]) => mockLogin2FA(...args),
+    logout: (...args: any[]) => mockLogout(...args),
+    getCurrentUser: (...args: any[]) => mockGetCurrentUser(...args),
+    register: (...args: any[]) => mockRegister(...args),
+    refreshToken: (...args: any[]) => mockRefreshToken(...args),
+  },
+  isTotp2FARequired: (response: any) => response?.requires_2fa === true,
+}))
+
+const fakeUser = {
+  id: 1,
+  username: 'testuser',
+  email: 'test@example.com',
+  role: 'user' as const,
+  balance: 100,
+  concurrency: 5,
+  status: 'active' as const,
+  allowed_groups: null,
+  created_at: '2024-01-01',
+  updated_at: '2024-01-01',
+}
+
+const fakeAdminUser = {
+  ...fakeUser,
+  id: 2,
+  username: 'admin',
+  email: 'admin@example.com',
+  role: 'admin' as const,
+}
+
+const fakeAuthResponse = {
+  access_token: 'test-token-123',
+  refresh_token: 'refresh-token-456',
+  expires_in: 3600,
+  token_type: 'Bearer',
+  user: { ...fakeUser },
+}
+
+describe('useAuthStore', () => {
+  beforeEach(() => {
+    setActivePinia(createPinia())
+    localStorage.clear()
+    vi.useFakeTimers()
+    vi.clearAllMocks()
+  })
+
+  afterEach(() => {
+    vi.useRealTimers()
+  })
+
+  // --- login ---
+
+  describe('login', () => {
+    it('成功登录后设置 token 和 user', async () => {
+      mockLogin.mockResolvedValue(fakeAuthResponse)
+      const store = useAuthStore()
+
+      await store.login({ email: 'test@example.com', password: '123456' })
+
+      expect(store.token).toBe('test-token-123')
+      expect(store.user).toEqual(fakeUser)
+      expect(store.isAuthenticated).toBe(true)
+      expect(localStorage.getItem('auth_token')).toBe('test-token-123')
+      expect(localStorage.getItem('auth_user')).toBe(JSON.stringify(fakeUser))
+    })
+
+    it('登录失败时清除状态并抛出错误', async () => {
+      mockLogin.mockRejectedValue(new Error('Invalid credentials'))
+      const store = useAuthStore()
+
+      await expect(store.login({ email: 'test@example.com', password: 'wrong' })).rejects.toThrow(
+        'Invalid credentials'
+      )
+
+      expect(store.token).toBeNull()
+      expect(store.user).toBeNull()
+      expect(store.isAuthenticated).toBe(false)
+    })
+
+    it('需要 2FA 时返回响应但不设置认证状态', async () => {
+      const twoFAResponse = { requires_2fa: true, temp_token: 'temp-123' }
+      mockLogin.mockResolvedValue(twoFAResponse)
+      const store = useAuthStore()
+
+      const result = await store.login({ email: 'test@example.com', password: '123456' })
+
+      expect(result).toEqual(twoFAResponse)
+      expect(store.token).toBeNull()
+      expect(store.isAuthenticated).toBe(false)
+    })
+  })
+
+  // --- login2FA ---
+
+  describe('login2FA', () => {
+    it('2FA 验证成功后设置认证状态', async () => {
+      mockLogin2FA.mockResolvedValue(fakeAuthResponse)
+      const store = useAuthStore()
+
+      const user = await store.login2FA('temp-123', '654321')
+
+      expect(store.token).toBe('test-token-123')
+      expect(store.user).toEqual(fakeUser)
+      expect(user).toEqual(fakeUser)
+      expect(mockLogin2FA).toHaveBeenCalledWith({
+        temp_token: 'temp-123',
+        totp_code: '654321',
+      })
+    })
+
+    it('2FA 验证失败时清除状态并抛出错误', async () => {
+      mockLogin2FA.mockRejectedValue(new Error('Invalid TOTP'))
+      const store = useAuthStore()
+
+      await expect(store.login2FA('temp-123', '000000')).rejects.toThrow('Invalid TOTP')
+      expect(store.token).toBeNull()
+      expect(store.isAuthenticated).toBe(false)
+    })
+  })
+
+  // --- logout ---
+
+  describe('logout', () => {
+    it('注销后清除所有状态和 localStorage', async () => {
+      mockLogin.mockResolvedValue(fakeAuthResponse)
+      mockLogout.mockResolvedValue(undefined)
+      const store = useAuthStore()
+
+      // 先登录
+      await store.login({ email: 'test@example.com', password: '123456' })
+      expect(store.isAuthenticated).toBe(true)
+
+      // 注销
+      await store.logout()
+
+      expect(store.token).toBeNull()
+      expect(store.user).toBeNull()
+      expect(store.isAuthenticated).toBe(false)
+      expect(localStorage.getItem('auth_token')).toBeNull()
+      expect(localStorage.getItem('auth_user')).toBeNull()
+      expect(localStorage.getItem('refresh_token')).toBeNull()
+      expect(localStorage.getItem('token_expires_at')).toBeNull()
+    })
+  })
+
+  // --- checkAuth ---
+
+  describe('checkAuth', () => {
+    it('从 localStorage 恢复持久化状态', () => {
+      localStorage.setItem('auth_token', 'saved-token')
+      localStorage.setItem('auth_user', JSON.stringify(fakeUser))
+
+      // Mock refreshUser (getCurrentUser) 防止后台刷新报错
+      mockGetCurrentUser.mockResolvedValue({ data: fakeUser })
+
+      const store = useAuthStore()
+      store.checkAuth()
+
+      expect(store.token).toBe('saved-token')
+      expect(store.user).toEqual(fakeUser)
+      expect(store.isAuthenticated).toBe(true)
+    })
+
+    it('localStorage 无数据时保持未认证状态', () => {
+      const store = useAuthStore()
+      store.checkAuth()
+
+      expect(store.token).toBeNull()
+      expect(store.user).toBeNull()
+      expect(store.isAuthenticated).toBe(false)
+    })
+
+    it('localStorage 中用户数据损坏时清除状态', () => {
+      localStorage.setItem('auth_token', 'saved-token')
+      localStorage.setItem('auth_user', 'invalid-json{{{')
+
+      const store = useAuthStore()
+      store.checkAuth()
+
+      expect(store.token).toBeNull()
+      expect(store.user).toBeNull()
+      expect(localStorage.getItem('auth_token')).toBeNull()
+    })
+
+    it('恢复 refresh token 和过期时间', () => {
+      const futureTs = String(Date.now() + 3600_000)
+      localStorage.setItem('auth_token', 'saved-token')
+      localStorage.setItem('auth_user', JSON.stringify(fakeUser))
+      localStorage.setItem('refresh_token', 'saved-refresh')
+      localStorage.setItem('token_expires_at', futureTs)
+
+      mockGetCurrentUser.mockResolvedValue({ data: fakeUser })
+
+      const store = useAuthStore()
+      store.checkAuth()
+
+      expect(store.isAuthenticated).toBe(true)
+    })
+  })
+
+  // --- isAdmin ---
+
+  describe('isAdmin', () => {
+    it('管理员用户返回 true', async () => {
+      const adminResponse = { ...fakeAuthResponse, user: { ...fakeAdminUser } }
+      mockLogin.mockResolvedValue(adminResponse)
+      const store = useAuthStore()
+
+      await store.login({ email: 'admin@example.com', password: '123456' })
+
+      expect(store.isAdmin).toBe(true)
+    })
+
+    it('普通用户返回 false', async () => {
+      mockLogin.mockResolvedValue(fakeAuthResponse)
+      const store = useAuthStore()
+
+      await store.login({ email: 'test@example.com', password: '123456' })
+
+      expect(store.isAdmin).toBe(false)
+    })
+
+    it('未登录时返回 false', () => {
+      const store = useAuthStore()
+      expect(store.isAdmin).toBe(false)
+    })
+  })
+
+  // --- refreshUser ---
+
+  describe('refreshUser', () => {
+    it('刷新用户数据并更新 localStorage', async () => {
+      mockLogin.mockResolvedValue(fakeAuthResponse)
+      const store = useAuthStore()
+      await store.login({ email: 'test@example.com', password: '123456' })
+
+      const updatedUser = { ...fakeUser, username: 'updated-name' }
+      mockGetCurrentUser.mockResolvedValue({ data: updatedUser })
+
+      const result = await store.refreshUser()
+
+      expect(result).toEqual(updatedUser)
+      expect(store.user).toEqual(updatedUser)
+      expect(JSON.parse(localStorage.getItem('auth_user')!)).toEqual(updatedUser)
+    })
+
+    it('未认证时抛出错误', async () => {
+      const store = useAuthStore()
+      await expect(store.refreshUser()).rejects.toThrow('Not authenticated')
+    })
+  })
+
+  // --- isSimpleMode ---
+
+  describe('isSimpleMode', () => {
+    it('run_mode 为 simple 时返回 true', async () => {
+      const simpleResponse = {
+        ...fakeAuthResponse,
+        user: { ...fakeUser, run_mode: 'simple' as const },
+      }
+      mockLogin.mockResolvedValue(simpleResponse)
+      const store = useAuthStore()
+
+      await store.login({ email: 'test@example.com', password: '123456' })
+
+      expect(store.isSimpleMode).toBe(true)
+    })
+
+    it('默认为 standard 模式', () => {
+      const store = useAuthStore()
+      expect(store.isSimpleMode).toBe(false)
+    })
+  })
+})