bb5a5dd65e
系统性地修复、补充和强化项目的自动化测试能力： 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>
121 lines
4.0 KiB
Go
121 lines
4.0 KiB
Go
//go:build unit
|
||
|
||
package service
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
// TestDecrementWaitCount_NilCache 确保 nil cache 不会 panic
|
||
func TestDecrementWaitCount_NilCache(t *testing.T) {
|
||
svc := &ConcurrencyService{cache: nil}
|
||
// 不应 panic
|
||
svc.DecrementWaitCount(context.Background(), 1)
|
||
}
|
||
|
||
// TestDecrementWaitCount_CacheError 确保 cache 错误不会传播
|
||
func TestDecrementWaitCount_CacheError(t *testing.T) {
|
||
cache := &stubConcurrencyCacheForTest{}
|
||
svc := NewConcurrencyService(cache)
|
||
// DecrementWaitCount 使用 background context，错误只记录日志不传播
|
||
svc.DecrementWaitCount(context.Background(), 1)
|
||
}
|
||
|
||
// TestDecrementAccountWaitCount_NilCache 确保 nil cache 不会 panic
|
||
func TestDecrementAccountWaitCount_NilCache(t *testing.T) {
|
||
svc := &ConcurrencyService{cache: nil}
|
||
svc.DecrementAccountWaitCount(context.Background(), 1)
|
||
}
|
||
|
||
// TestDecrementAccountWaitCount_CacheError 确保 cache 错误不会传播
|
||
func TestDecrementAccountWaitCount_CacheError(t *testing.T) {
|
||
cache := &stubConcurrencyCacheForTest{}
|
||
svc := NewConcurrencyService(cache)
|
||
svc.DecrementAccountWaitCount(context.Background(), 1)
|
||
}
|
||
|
||
// TestWaitingQueueFlow_IncrementThenDecrement 测试完整的等待队列增减流程
|
||
func TestWaitingQueueFlow_IncrementThenDecrement(t *testing.T) {
|
||
cache := &stubConcurrencyCacheForTest{waitAllowed: true}
|
||
svc := NewConcurrencyService(cache)
|
||
|
||
// 进入等待队列
|
||
allowed, err := svc.IncrementWaitCount(context.Background(), 1, 25)
|
||
require.NoError(t, err)
|
||
require.True(t, allowed)
|
||
|
||
// 离开等待队列（不应 panic）
|
||
svc.DecrementWaitCount(context.Background(), 1)
|
||
}
|
||
|
||
// TestWaitingQueueFlow_AccountLevel 测试账号级等待队列流程
|
||
func TestWaitingQueueFlow_AccountLevel(t *testing.T) {
|
||
cache := &stubConcurrencyCacheForTest{waitAllowed: true}
|
||
svc := NewConcurrencyService(cache)
|
||
|
||
// 进入账号等待队列
|
||
allowed, err := svc.IncrementAccountWaitCount(context.Background(), 42, 10)
|
||
require.NoError(t, err)
|
||
require.True(t, allowed)
|
||
|
||
// 离开账号等待队列
|
||
svc.DecrementAccountWaitCount(context.Background(), 42)
|
||
}
|
||
|
||
// TestWaitingQueueFull_Returns429Signal 测试等待队列满时返回 false
|
||
func TestWaitingQueueFull_Returns429Signal(t *testing.T) {
|
||
// waitAllowed=false 模拟队列已满
|
||
cache := &stubConcurrencyCacheForTest{waitAllowed: false}
|
||
svc := NewConcurrencyService(cache)
|
||
|
||
// 用户级等待队列满
|
||
allowed, err := svc.IncrementWaitCount(context.Background(), 1, 25)
|
||
require.NoError(t, err)
|
||
require.False(t, allowed, "等待队列满时应返回 false（调用方根据此返回 429）")
|
||
|
||
// 账号级等待队列满
|
||
allowed, err = svc.IncrementAccountWaitCount(context.Background(), 1, 10)
|
||
require.NoError(t, err)
|
||
require.False(t, allowed, "账号等待队列满时应返回 false")
|
||
}
|
||
|
||
// TestWaitingQueue_FailOpen_OnCacheError 测试 Redis 故障时 fail-open
|
||
func TestWaitingQueue_FailOpen_OnCacheError(t *testing.T) {
|
||
cache := &stubConcurrencyCacheForTest{waitErr: errors.New("redis connection refused")}
|
||
svc := NewConcurrencyService(cache)
|
||
|
||
// 用户级：Redis 错误时允许通过
|
||
allowed, err := svc.IncrementWaitCount(context.Background(), 1, 25)
|
||
require.NoError(t, err, "Redis 错误不应向调用方传播")
|
||
require.True(t, allowed, "Redis 故障时应 fail-open 放行")
|
||
|
||
// 账号级：同样 fail-open
|
||
allowed, err = svc.IncrementAccountWaitCount(context.Background(), 1, 10)
|
||
require.NoError(t, err, "Redis 错误不应向调用方传播")
|
||
require.True(t, allowed, "Redis 故障时应 fail-open 放行")
|
||
}
|
||
|
||
// TestCalculateMaxWait_Scenarios 测试最大等待队列大小计算
|
||
func TestCalculateMaxWait_Scenarios(t *testing.T) {
|
||
tests := []struct {
|
||
concurrency int
|
||
expected int
|
||
}{
|
||
{5, 25}, // 5 + 20
|
||
{10, 30}, // 10 + 20
|
||
{1, 21}, // 1 + 20
|
||
{0, 21}, // min(1) + 20
|
||
{-1, 21}, // min(1) + 20
|
||
{-10, 21}, // min(1) + 20
|
||
{100, 120}, // 100 + 20
|
||
}
|
||
for _, tt := range tests {
|
||
result := CalculateMaxWait(tt.concurrency)
|
||
require.Equal(t, tt.expected, result, "CalculateMaxWait(%d)", tt.concurrency)
|
||
}
|
||
}
|