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>
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
package repository
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
// --- Task 6.1 验证: math/rand/v2 迁移后 jitteredTTL 行为正确 ---
|
||
|
||
func TestJitteredTTL_WithinExpectedRange(t *testing.T) {
|
||
// jitteredTTL 使用减法抖动: billingCacheTTL - [0, billingCacheJitter)
|
||
// 所以结果应在 [billingCacheTTL - billingCacheJitter, billingCacheTTL] 范围内
|
||
lowerBound := billingCacheTTL - billingCacheJitter // 5min - 30s = 4min30s
|
||
upperBound := billingCacheTTL // 5min
|
||
|
||
for i := 0; i < 200; i++ {
|
||
ttl := jitteredTTL()
|
||
assert.GreaterOrEqual(t, int64(ttl), int64(lowerBound),
|
||
"TTL 不应低于 %v，实际得到 %v", lowerBound, ttl)
|
||
assert.LessOrEqual(t, int64(ttl), int64(upperBound),
|
||
"TTL 不应超过 %v（上界不变保证），实际得到 %v", upperBound, ttl)
|
||
}
|
||
}
|
||
|
||
func TestJitteredTTL_NeverExceedsBase(t *testing.T) {
|
||
// 关键安全性测试：jitteredTTL 使用减法抖动，确保永远不超过 billingCacheTTL
|
||
for i := 0; i < 500; i++ {
|
||
ttl := jitteredTTL()
|
||
assert.LessOrEqual(t, int64(ttl), int64(billingCacheTTL),
|
||
"jitteredTTL 不应超过基础 TTL（上界预期不被打破）")
|
||
}
|
||
}
|
||
|
||
func TestJitteredTTL_HasVariance(t *testing.T) {
|
||
// 验证抖动确实产生了不同的值
|
||
results := make(map[time.Duration]bool)
|
||
for i := 0; i < 100; i++ {
|
||
ttl := jitteredTTL()
|
||
results[ttl] = true
|
||
}
|
||
|
||
require.Greater(t, len(results), 1,
|
||
"jitteredTTL 应产生不同的值（抖动生效），但 100 次调用结果全部相同")
|
||
}
|
||
|
||
func TestJitteredTTL_AverageNearCenter(t *testing.T) {
|
||
// 验证平均值大约在抖动范围中间
|
||
var sum time.Duration
|
||
runs := 1000
|
||
for i := 0; i < runs; i++ {
|
||
sum += jitteredTTL()
|
||
}
|
||
|
||
avg := sum / time.Duration(runs)
|
||
expectedCenter := billingCacheTTL - billingCacheJitter/2 // 4min45s
|
||
|
||
// 允许 ±5s 的误差
|
||
tolerance := 5 * time.Second
|
||
assert.InDelta(t, float64(expectedCenter), float64(avg), float64(tolerance),
|
||
"平均 TTL 应接近抖动范围中心 %v", expectedCenter)
|
||
}
|
||
|
||
func TestBillingKeyGeneration(t *testing.T) {
|
||
t.Run("balance_key", func(t *testing.T) {
|
||
key := billingBalanceKey(12345)
|
||
assert.Equal(t, "billing:balance:12345", key)
|
||
})
|
||
|
||
t.Run("sub_key", func(t *testing.T) {
|
||
key := billingSubKey(100, 200)
|
||
assert.Equal(t, "billing:sub:100:200", key)
|
||
})
|
||
}
|
||
|
||
func BenchmarkJitteredTTL(b *testing.B) {
|
||
for i := 0; i < b.N; i++ {
|
||
_ = jitteredTTL()
|
||
}
|
||
}
|