test: 为代码审核修复添加详细单元测试(7个测试文件,50+测试用例)
新增测试文件: - cors_test.go: CORS 条件化头部测试(12个测试,覆盖白名单/黑名单/通配符/凭证/多源/Vary) - gateway_helper_backoff_test.go: nextBackoff 退避测试(6个测试+基准,验证指数增长/边界/抖动/收敛) - billing_cache_jitter_test.go: jitteredTTL 抖动测试(5个测试+基准,验证范围/上界/方差/均值) - subscription_calculate_progress_test.go: calculateProgress 纯函数测试(9个测试,覆盖日/周/月限额/超限截断/过期) - openai_gateway_handler_test.go: SSE JSON 转义测试(7个子用例,验证双引号/反斜杠/换行符安全) 更新测试文件: - response_transformer_test.go: 增强 generateRandomID 测试(7个测试,含并发/字符集/降级计数器) - security_headers_test.go: 适配 GenerateNonce 新签名 - api_key_auth_test.go: 适配 NewSubscriptionService 新参数 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -3,11 +3,16 @@
|
||||
package antigravity
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
// --- Task 7: 验证 generateRandomID 和降级碰撞防护 ---
|
||||
|
||||
func TestGenerateRandomID_Uniqueness(t *testing.T) {
|
||||
seen := make(map[string]struct{}, 100)
|
||||
for i := 0; i < 100; i++ {
|
||||
@@ -19,6 +24,39 @@ func TestGenerateRandomID_Uniqueness(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestFallbackCounter_Increments(t *testing.T) {
|
||||
// 验证 fallbackCounter 的原子递增行为确保降级分支不会生成相同 seed
|
||||
before := atomic.LoadUint64(&fallbackCounter)
|
||||
cnt1 := atomic.AddUint64(&fallbackCounter, 1)
|
||||
cnt2 := atomic.AddUint64(&fallbackCounter, 1)
|
||||
require.Equal(t, before+1, cnt1, "第一次递增应为 before+1")
|
||||
require.Equal(t, before+2, cnt2, "第二次递增应为 before+2")
|
||||
require.NotEqual(t, cnt1, cnt2, "连续两次递增的计数器值应不同")
|
||||
}
|
||||
|
||||
func TestFallbackCounter_ConcurrentIncrements(t *testing.T) {
|
||||
// 验证并发递增的原子性 — 每次递增都应产生唯一值
|
||||
const goroutines = 50
|
||||
results := make([]uint64, goroutines)
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(goroutines)
|
||||
|
||||
for i := 0; i < goroutines; i++ {
|
||||
go func(idx int) {
|
||||
defer wg.Done()
|
||||
results[idx] = atomic.AddUint64(&fallbackCounter, 1)
|
||||
}(i)
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
// 所有结果应唯一
|
||||
seen := make(map[uint64]bool, goroutines)
|
||||
for _, v := range results {
|
||||
assert.False(t, seen[v], "并发递增产生了重复值: %d", v)
|
||||
seen[v] = true
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateRandomID_Charset(t *testing.T) {
|
||||
const validChars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||
validSet := make(map[byte]struct{}, len(validChars))
|
||||
@@ -34,3 +72,38 @@ func TestGenerateRandomID_Charset(t *testing.T) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateRandomID_Length(t *testing.T) {
|
||||
for i := 0; i < 100; i++ {
|
||||
id := generateRandomID()
|
||||
assert.Len(t, id, 12, "每次生成的 ID 长度应为 12")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateRandomID_ConcurrentUniqueness(t *testing.T) {
|
||||
// 验证并发调用不会产生重复 ID
|
||||
const goroutines = 100
|
||||
results := make([]string, goroutines)
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(goroutines)
|
||||
|
||||
for i := 0; i < goroutines; i++ {
|
||||
go func(idx int) {
|
||||
defer wg.Done()
|
||||
results[idx] = generateRandomID()
|
||||
}(i)
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
seen := make(map[string]bool, goroutines)
|
||||
for _, id := range results {
|
||||
assert.False(t, seen[id], "并发调用产生了重复 ID: %s", id)
|
||||
seen[id] = true
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkGenerateRandomID(b *testing.B) {
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = generateRandomID()
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user