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>
This commit is contained in:
yangjianbo
317
backend/internal/integration/e2e_user_flow_test.go
Normal file
317
backend/internal/integration/e2e_user_flow_test.go
Normal file
@@ -0,0 +1,317 @@
|
||||
//go:build e2e
|
||||
|
||||
package integration
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// E2E 用户流程测试
|
||||
// 测试完整的用户操作链路：注册 → 登录 → 创建 API Key → 调用网关 → 查询用量
|
||||
|
||||
var (
|
||||
testUserEmail = "e2e-test-" + fmt.Sprintf("%d", time.Now().UnixMilli()) + "@test.local"
|
||||
testUserPassword = "E2eTest@12345"
|
||||
testUserName = "e2e-test-user"
|
||||
)
|
||||
|
||||
// TestUserRegistrationAndLogin 测试用户注册和登录流程
|
||||
func TestUserRegistrationAndLogin(t *testing.T) {
|
||||
// 步骤 1: 注册新用户
|
||||
t.Run("注册新用户", func(t *testing.T) {
|
||||
payload := map[string]string{
|
||||
"email": testUserEmail,
|
||||
"password": testUserPassword,
|
||||
"username": testUserName,
|
||||
}
|
||||
body, _ := json.Marshal(payload)
|
||||
|
||||
resp, err := doRequest(t, "POST", "/api/auth/register", body, "")
|
||||
if err != nil {
|
||||
t.Skipf("注册接口不可用，跳过用户流程测试: %v", err)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
|
||||
// 注册可能返回 200（成功）或 400（邮箱已存在）或 403（注册已关闭）
|
||||
switch resp.StatusCode {
|
||||
case 200:
|
||||
t.Logf("✅ 用户注册成功: %s", testUserEmail)
|
||||
case 400:
|
||||
t.Logf("⚠️ 用户可能已存在: %s", string(respBody))
|
||||
case 403:
|
||||
t.Skipf("注册功能已关闭: %s", string(respBody))
|
||||
default:
|
||||
t.Logf("⚠️ 注册返回 HTTP %d: %s（继续尝试登录）", resp.StatusCode, string(respBody))
|
||||
}
|
||||
})
|
||||
|
||||
// 步骤 2: 登录获取 JWT
|
||||
var accessToken string
|
||||
t.Run("用户登录获取JWT", func(t *testing.T) {
|
||||
payload := map[string]string{
|
||||
"email": testUserEmail,
|
||||
"password": testUserPassword,
|
||||
}
|
||||
body, _ := json.Marshal(payload)
|
||||
|
||||
resp, err := doRequest(t, "POST", "/api/auth/login", body, "")
|
||||
if err != nil {
|
||||
t.Fatalf("登录请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
t.Skipf("登录失败 HTTP %d: %s（可能需要先注册用户）", resp.StatusCode, string(respBody))
|
||||
return
|
||||
}
|
||||
|
||||
var result map[string]any
|
||||
if err := json.Unmarshal(respBody, &result); err != nil {
|
||||
t.Fatalf("解析登录响应失败: %v", err)
|
||||
}
|
||||
|
||||
// 尝试从标准响应格式获取 token
|
||||
if token, ok := result["access_token"].(string); ok && token != "" {
|
||||
accessToken = token
|
||||
} else if data, ok := result["data"].(map[string]any); ok {
|
||||
if token, ok := data["access_token"].(string); ok {
|
||||
accessToken = token
|
||||
}
|
||||
}
|
||||
|
||||
if accessToken == "" {
|
||||
t.Skipf("未获取到 access_token，响应: %s", string(respBody))
|
||||
return
|
||||
}
|
||||
|
||||
// 验证 token 不为空且格式基本正确
|
||||
if len(accessToken) < 10 {
|
||||
t.Fatalf("access_token 格式异常: %s", accessToken)
|
||||
}
|
||||
|
||||
t.Logf("✅ 登录成功，获取 JWT（长度: %d）", len(accessToken))
|
||||
})
|
||||
|
||||
if accessToken == "" {
|
||||
t.Skip("未获取到 JWT，跳过后续测试")
|
||||
return
|
||||
}
|
||||
|
||||
// 步骤 3: 使用 JWT 获取当前用户信息
|
||||
t.Run("获取当前用户信息", func(t *testing.T) {
|
||||
resp, err := doRequest(t, "GET", "/api/user/me", nil, accessToken)
|
||||
if err != nil {
|
||||
t.Fatalf("请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
body, _ := io.ReadAll(resp.Body)
|
||||
t.Fatalf("HTTP %d: %s", resp.StatusCode, string(body))
|
||||
}
|
||||
|
||||
t.Logf("✅ 成功获取用户信息")
|
||||
})
|
||||
}
|
||||
|
||||
// TestAPIKeyLifecycle 测试 API Key 的创建和使用
|
||||
func TestAPIKeyLifecycle(t *testing.T) {
|
||||
// 先登录获取 JWT
|
||||
accessToken := loginTestUser(t)
|
||||
if accessToken == "" {
|
||||
t.Skip("无法登录，跳过 API Key 生命周期测试")
|
||||
return
|
||||
}
|
||||
|
||||
var apiKey string
|
||||
|
||||
// 步骤 1: 创建 API Key
|
||||
t.Run("创建API_Key", func(t *testing.T) {
|
||||
payload := map[string]string{
|
||||
"name": "e2e-test-key-" + fmt.Sprintf("%d", time.Now().UnixMilli()),
|
||||
}
|
||||
body, _ := json.Marshal(payload)
|
||||
|
||||
resp, err := doRequest(t, "POST", "/api/keys", body, accessToken)
|
||||
if err != nil {
|
||||
t.Fatalf("创建 API Key 请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
t.Skipf("创建 API Key 失败 HTTP %d: %s", resp.StatusCode, string(respBody))
|
||||
return
|
||||
}
|
||||
|
||||
var result map[string]any
|
||||
if err := json.Unmarshal(respBody, &result); err != nil {
|
||||
t.Fatalf("解析响应失败: %v", err)
|
||||
}
|
||||
|
||||
// 从响应中提取 key
|
||||
if key, ok := result["key"].(string); ok {
|
||||
apiKey = key
|
||||
} else if data, ok := result["data"].(map[string]any); ok {
|
||||
if key, ok := data["key"].(string); ok {
|
||||
apiKey = key
|
||||
}
|
||||
}
|
||||
|
||||
if apiKey == "" {
|
||||
t.Skipf("未获取到 API Key，响应: %s", string(respBody))
|
||||
return
|
||||
}
|
||||
|
||||
// 验证 API Key 脱敏日志（只显示前 8 位）
|
||||
masked := apiKey
|
||||
if len(masked) > 8 {
|
||||
masked = masked[:8] + "..."
|
||||
}
|
||||
t.Logf("✅ API Key 创建成功: %s", masked)
|
||||
})
|
||||
|
||||
if apiKey == "" {
|
||||
t.Skip("未创建 API Key，跳过后续测试")
|
||||
return
|
||||
}
|
||||
|
||||
// 步骤 2: 使用 API Key 调用网关（需要 Claude 或 Gemini 可用）
|
||||
t.Run("使用API_Key调用网关", func(t *testing.T) {
|
||||
// 尝试调用 models 列表（最轻量的 API 调用）
|
||||
resp, err := doRequest(t, "GET", "/v1/models", nil, apiKey)
|
||||
if err != nil {
|
||||
t.Fatalf("网关请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
|
||||
// 可能返回 200（成功）或 402（余额不足）或 403（无可用账户）
|
||||
switch {
|
||||
case resp.StatusCode == 200:
|
||||
t.Logf("✅ API Key 网关调用成功")
|
||||
case resp.StatusCode == 402:
|
||||
t.Logf("⚠️ 余额不足，但 API Key 认证通过")
|
||||
case resp.StatusCode == 403:
|
||||
t.Logf("⚠️ 无可用账户，但 API Key 认证通过")
|
||||
default:
|
||||
t.Logf("⚠️ 网关返回 HTTP %d: %s", resp.StatusCode, string(respBody))
|
||||
}
|
||||
})
|
||||
|
||||
// 步骤 3: 查询用量记录
|
||||
t.Run("查询用量记录", func(t *testing.T) {
|
||||
resp, err := doRequest(t, "GET", "/api/usage/dashboard", nil, accessToken)
|
||||
if err != nil {
|
||||
t.Fatalf("用量查询请求失败: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
body, _ := io.ReadAll(resp.Body)
|
||||
t.Logf("⚠️ 用量查询返回 HTTP %d: %s", resp.StatusCode, string(body))
|
||||
return
|
||||
}
|
||||
|
||||
t.Logf("✅ 用量查询成功")
|
||||
})
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 辅助函数
|
||||
// =============================================================================
|
||||
|
||||
func doRequest(t *testing.T, method, path string, body []byte, token string) (*http.Response, error) {
|
||||
t.Helper()
|
||||
|
||||
url := baseURL + path
|
||||
var bodyReader io.Reader
|
||||
if body != nil {
|
||||
bodyReader = bytes.NewReader(body)
|
||||
}
|
||||
|
||||
req, err := http.NewRequest(method, url, bodyReader)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建请求失败: %w", err)
|
||||
}
|
||||
|
||||
if body != nil {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
if token != "" {
|
||||
req.Header.Set("Authorization", "Bearer "+token)
|
||||
}
|
||||
|
||||
client := &http.Client{Timeout: 30 * time.Second}
|
||||
return client.Do(req)
|
||||
}
|
||||
|
||||
func loginTestUser(t *testing.T) string {
|
||||
t.Helper()
|
||||
|
||||
// 先尝试用管理员账户登录
|
||||
adminEmail := getEnv("ADMIN_EMAIL", "admin@sub2api.local")
|
||||
adminPassword := getEnv("ADMIN_PASSWORD", "")
|
||||
|
||||
if adminPassword == "" {
|
||||
// 尝试用测试用户
|
||||
adminEmail = testUserEmail
|
||||
adminPassword = testUserPassword
|
||||
}
|
||||
|
||||
payload := map[string]string{
|
||||
"email": adminEmail,
|
||||
"password": adminPassword,
|
||||
}
|
||||
body, _ := json.Marshal(payload)
|
||||
|
||||
resp, err := doRequest(t, "POST", "/api/auth/login", body, "")
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != 200 {
|
||||
return ""
|
||||
}
|
||||
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
var result map[string]any
|
||||
if err := json.Unmarshal(respBody, &result); err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
if token, ok := result["access_token"].(string); ok {
|
||||
return token
|
||||
}
|
||||
if data, ok := result["data"].(map[string]any); ok {
|
||||
if token, ok := data["access_token"].(string); ok {
|
||||
return token
|
||||
}
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
// redactAPIKey API Key 脱敏，只显示前 8 位
|
||||
func redactAPIKey(key string) string {
|
||||
key = strings.TrimSpace(key)
|
||||
if len(key) <= 8 {
|
||||
return "***"
|
||||
}
|
||||
return key[:8] + "..."
|
||||
}
|
||||
Reference in New Issue
Block a user