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
78
backend/internal/testutil/fixtures.go
Normal file
78
backend/internal/testutil/fixtures.go
Normal file
@@ -0,0 +1,78 @@
|
||||
//go:build unit
|
||||
|
||||
package testutil
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||||
)
|
||||
|
||||
// NewTestUser 创建一个可用的测试用户，可通过 opts 覆盖默认值。
|
||||
func NewTestUser(opts ...func(*service.User)) *service.User {
|
||||
u := &service.User{
|
||||
ID: 1,
|
||||
Email: "test@example.com",
|
||||
Username: "testuser",
|
||||
Role: "user",
|
||||
Balance: 100.0,
|
||||
Concurrency: 5,
|
||||
Status: service.StatusActive,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(u)
|
||||
}
|
||||
return u
|
||||
}
|
||||
|
||||
// NewTestAccount 创建一个可用的测试账户，可通过 opts 覆盖默认值。
|
||||
func NewTestAccount(opts ...func(*service.Account)) *service.Account {
|
||||
a := &service.Account{
|
||||
ID: 1,
|
||||
Name: "test-account",
|
||||
Platform: service.PlatformAnthropic,
|
||||
Status: service.StatusActive,
|
||||
Schedulable: true,
|
||||
Concurrency: 5,
|
||||
Priority: 1,
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(a)
|
||||
}
|
||||
return a
|
||||
}
|
||||
|
||||
// NewTestAPIKey 创建一个可用的测试 API Key，可通过 opts 覆盖默认值。
|
||||
func NewTestAPIKey(opts ...func(*service.APIKey)) *service.APIKey {
|
||||
groupID := int64(1)
|
||||
k := &service.APIKey{
|
||||
ID: 1,
|
||||
UserID: 1,
|
||||
Key: "sk-test-key-12345678",
|
||||
Name: "test-key",
|
||||
GroupID: &groupID,
|
||||
Status: service.StatusActive,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(k)
|
||||
}
|
||||
return k
|
||||
}
|
||||
|
||||
// NewTestGroup 创建一个可用的测试分组，可通过 opts 覆盖默认值。
|
||||
func NewTestGroup(opts ...func(*service.Group)) *service.Group {
|
||||
g := &service.Group{
|
||||
ID: 1,
|
||||
Platform: service.PlatformAnthropic,
|
||||
Status: service.StatusActive,
|
||||
Hydrated: true,
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(g)
|
||||
}
|
||||
return g
|
||||
}
|
||||
35
backend/internal/testutil/httptest.go
Normal file
35
backend/internal/testutil/httptest.go
Normal file
@@ -0,0 +1,35 @@
|
||||
//go:build unit
|
||||
|
||||
package testutil
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func init() {
|
||||
gin.SetMode(gin.TestMode)
|
||||
}
|
||||
|
||||
// NewGinTestContext 创建一个 Gin 测试上下文和 ResponseRecorder。
|
||||
// body 为空字符串时创建无 body 的请求。
|
||||
func NewGinTestContext(method, path, body string) (*gin.Context, *httptest.ResponseRecorder) {
|
||||
rec := httptest.NewRecorder()
|
||||
c, _ := gin.CreateTestContext(rec)
|
||||
|
||||
var bodyReader io.Reader
|
||||
if body != "" {
|
||||
bodyReader = strings.NewReader(body)
|
||||
}
|
||||
|
||||
c.Request = httptest.NewRequest(method, path, bodyReader)
|
||||
if method == http.MethodPost || method == http.MethodPut || method == http.MethodPatch {
|
||||
c.Request.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
|
||||
return c, rec
|
||||
}
|
||||
137
backend/internal/testutil/stubs.go
Normal file
137
backend/internal/testutil/stubs.go
Normal file
@@ -0,0 +1,137 @@
|
||||
//go:build unit
|
||||
|
||||
// Package testutil 提供单元测试共享的 Stub、Fixture 和辅助函数。
|
||||
// 所有文件使用 //go:build unit 标签，确保不会被生产构建包含。
|
||||
package testutil
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||||
)
|
||||
|
||||
// ============================================================
|
||||
// StubConcurrencyCache — service.ConcurrencyCache 的空实现
|
||||
// ============================================================
|
||||
|
||||
// 编译期接口断言
|
||||
var _ service.ConcurrencyCache = StubConcurrencyCache{}
|
||||
|
||||
// StubConcurrencyCache 是 ConcurrencyCache 的默认空实现，所有方法返回零值。
|
||||
type StubConcurrencyCache struct{}
|
||||
|
||||
func (c StubConcurrencyCache) AcquireAccountSlot(_ context.Context, _ int64, _ int, _ string) (bool, error) {
|
||||
return true, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) ReleaseAccountSlot(_ context.Context, _ int64, _ string) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubConcurrencyCache) GetAccountConcurrency(_ context.Context, _ int64) (int, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) IncrementAccountWaitCount(_ context.Context, _ int64, _ int) (bool, error) {
|
||||
return true, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) DecrementAccountWaitCount(_ context.Context, _ int64) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubConcurrencyCache) GetAccountWaitingCount(_ context.Context, _ int64) (int, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) AcquireUserSlot(_ context.Context, _ int64, _ int, _ string) (bool, error) {
|
||||
return true, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) ReleaseUserSlot(_ context.Context, _ int64, _ string) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubConcurrencyCache) GetUserConcurrency(_ context.Context, _ int64) (int, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) IncrementWaitCount(_ context.Context, _ int64, _ int) (bool, error) {
|
||||
return true, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) DecrementWaitCount(_ context.Context, _ int64) error { return nil }
|
||||
func (c StubConcurrencyCache) GetAccountsLoadBatch(_ context.Context, accounts []service.AccountWithConcurrency) (map[int64]*service.AccountLoadInfo, error) {
|
||||
result := make(map[int64]*service.AccountLoadInfo, len(accounts))
|
||||
for _, acc := range accounts {
|
||||
result[acc.ID] = &service.AccountLoadInfo{AccountID: acc.ID, LoadRate: 0}
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) GetUsersLoadBatch(_ context.Context, users []service.UserWithConcurrency) (map[int64]*service.UserLoadInfo, error) {
|
||||
result := make(map[int64]*service.UserLoadInfo, len(users))
|
||||
for _, u := range users {
|
||||
result[u.ID] = &service.UserLoadInfo{UserID: u.ID, LoadRate: 0}
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
func (c StubConcurrencyCache) CleanupExpiredAccountSlots(_ context.Context, _ int64) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// StubGatewayCache — service.GatewayCache 的空实现
|
||||
// ============================================================
|
||||
|
||||
var _ service.GatewayCache = StubGatewayCache{}
|
||||
|
||||
type StubGatewayCache struct{}
|
||||
|
||||
func (c StubGatewayCache) GetSessionAccountID(_ context.Context, _ int64, _ string) (int64, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubGatewayCache) SetSessionAccountID(_ context.Context, _ int64, _ string, _ int64, _ time.Duration) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubGatewayCache) RefreshSessionTTL(_ context.Context, _ int64, _ string, _ time.Duration) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubGatewayCache) DeleteSessionAccountID(_ context.Context, _ int64, _ string) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubGatewayCache) IncrModelCallCount(_ context.Context, _ int64, _ string) (int64, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubGatewayCache) GetModelLoadBatch(_ context.Context, _ []int64, _ string) (map[int64]*service.ModelLoadInfo, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (c StubGatewayCache) FindGeminiSession(_ context.Context, _ int64, _, _ string) (string, int64, bool) {
|
||||
return "", 0, false
|
||||
}
|
||||
func (c StubGatewayCache) SaveGeminiSession(_ context.Context, _ int64, _, _, _ string, _ int64) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// StubSessionLimitCache — service.SessionLimitCache 的空实现
|
||||
// ============================================================
|
||||
|
||||
var _ service.SessionLimitCache = StubSessionLimitCache{}
|
||||
|
||||
type StubSessionLimitCache struct{}
|
||||
|
||||
func (c StubSessionLimitCache) RegisterSession(_ context.Context, _ int64, _ string, _ int, _ time.Duration) (bool, error) {
|
||||
return true, nil
|
||||
}
|
||||
func (c StubSessionLimitCache) RefreshSession(_ context.Context, _ int64, _ string, _ time.Duration) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubSessionLimitCache) GetActiveSessionCount(_ context.Context, _ int64) (int, error) {
|
||||
return 0, nil
|
||||
}
|
||||
func (c StubSessionLimitCache) GetActiveSessionCountBatch(_ context.Context, _ []int64, _ map[int64]time.Duration) (map[int64]int, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (c StubSessionLimitCache) IsSessionActive(_ context.Context, _ int64, _ string) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
func (c StubSessionLimitCache) GetWindowCost(_ context.Context, _ int64) (float64, bool, error) {
|
||||
return 0, false, nil
|
||||
}
|
||||
func (c StubSessionLimitCache) SetWindowCost(_ context.Context, _ int64, _ float64) error {
|
||||
return nil
|
||||
}
|
||||
func (c StubSessionLimitCache) GetWindowCostBatch(_ context.Context, _ []int64) (map[int64]float64, error) {
|
||||
return nil, nil
|
||||
}
|
||||
Reference in New Issue
Block a user