2588fa6a8f
基于 backend-code-audit 审计报告，修复剩余 P0/P1/P2 共 34 项问题： P0 生产 Bug： - 修复 time.Since(time.Now()) 计时逻辑错误 (P0-03) - generateRandomID 改用 crypto/rand 替代固定索引 (P0-04) - IncrementQuotaUsed 重写为 Ent 原子操作消除 TOCTOU 竞态 (P0-05) 安全加固： - gateway/openai handler 错误响应替换为泛化消息，防止内部信息泄露 (P1-14) - usage_log_repo dateFormat 参数改用白名单映射，防止 SQL 注入 (P1-16) - 默认配置安全加固：sslmode=prefer、response_headers=true、mode=release (P1-18/19, P2-15) 性能优化： - gateway handler 循环内 defer 替换为显式 releaseWait 闭包 (P1-02) - group_repo/promo_code_repo Count 前 Clone 查询避免状态污染 (P1-03) - usage_log_repo 四个查询添加 LIMIT 10000 防止 OOM (P1-07) - GetBatchUsageStats 添加时间范围参数，默认最近 30 天 (P1-10) - ip.go CIDR 预编译为包级变量 (P1-11) - BatchUpdateCredentials 重构为先验证后更新 (P1-13) 缓存一致性： - billing_cache 添加 jitteredTTL 防止缓存雪崩 (P2-10) - DeductUserBalance/UpdateSubscriptionUsage 错误传播修复 (P2-12) - UserService.UpdateBalance 成功后异步失效 billingCache (P2-13) 代码质量： - search 截断改为按 rune 处理，支持多字节字符 (P2-01) - TLS Handshake 改为 HandshakeContext 支持 context 取消 (P2-07) - CORS 预检添加 Access-Control-Max-Age: 86400 (P2-16) 测试覆盖： - 新增 user_service_test.go（UpdateBalance 缓存失效 6 个用例） - 新增 batch_update_credentials_test.go（fail-fast + 类型验证 7 个用例） - 新增 response_transformer_test.go、ip_test.go、usage_log_repo_unit_test.go、search_truncate_test.go - 集成测试：IncrementQuotaUsed 并发测试、billing_cache 错误传播测试 - config_test.go 补充 server.mode/sslmode 默认值断言 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
//go:build unit
|
||
|
||
package admin
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
// truncateSearchByRune 模拟 user_handler.go 中的 search 截断逻辑
|
||
func truncateSearchByRune(search string, maxRunes int) string {
|
||
if runes := []rune(search); len(runes) > maxRunes {
|
||
return string(runes[:maxRunes])
|
||
}
|
||
return search
|
||
}
|
||
|
||
func TestTruncateSearchByRune(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
input string
|
||
maxRunes int
|
||
wantLen int // 期望的 rune 长度
|
||
}{
|
||
{
|
||
name: "纯中文超长",
|
||
input: string(make([]rune, 150)),
|
||
maxRunes: 100,
|
||
wantLen: 100,
|
||
},
|
||
{
|
||
name: "纯 ASCII 超长",
|
||
input: string(make([]byte, 150)),
|
||
maxRunes: 100,
|
||
wantLen: 100,
|
||
},
|
||
{
|
||
name: "空字符串",
|
||
input: "",
|
||
maxRunes: 100,
|
||
wantLen: 0,
|
||
},
|
||
{
|
||
name: "恰好 100 个字符",
|
||
input: string(make([]rune, 100)),
|
||
maxRunes: 100,
|
||
wantLen: 100,
|
||
},
|
||
{
|
||
name: "不足 100 字符不截断",
|
||
input: "hello世界",
|
||
maxRunes: 100,
|
||
wantLen: 7,
|
||
},
|
||
}
|
||
|
||
for _, tc := range tests {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
result := truncateSearchByRune(tc.input, tc.maxRunes)
|
||
require.Equal(t, tc.wantLen, len([]rune(result)))
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestTruncateSearchByRune_PreservesMultibyte(t *testing.T) {
|
||
// 101 个中文字符，截断到 100 个后应该仍然是有效 UTF-8
|
||
input := ""
|
||
for i := 0; i < 101; i++ {
|
||
input += "中"
|
||
}
|
||
result := truncateSearchByRune(input, 100)
|
||
|
||
require.Equal(t, 100, len([]rune(result)))
|
||
// 验证截断结果是有效的 UTF-8（每个中文字符 3 字节）
|
||
require.Equal(t, 300, len(result))
|
||
}
|
||
|
||
func TestTruncateSearchByRune_MixedASCIIAndMultibyte(t *testing.T) {
|
||
// 50 个 ASCII + 51 个中文 = 101 个 rune
|
||
input := ""
|
||
for i := 0; i < 50; i++ {
|
||
input += "a"
|
||
}
|
||
for i := 0; i < 51; i++ {
|
||
input += "中"
|
||
}
|
||
result := truncateSearchByRune(input, 100)
|
||
|
||
runes := []rune(result)
|
||
require.Equal(t, 100, len(runes))
|
||
// 前 50 个应该是 'a'，后 50 个应该是 '中'
|
||
require.Equal(t, 'a', runes[0])
|
||
require.Equal(t, 'a', runes[49])
|
||
require.Equal(t, '中', runes[50])
|
||
require.Equal(t, '中', runes[99])
|
||
}
|