Files
ai-gateway-go/internal/platform/config/config_test.go
T
superidou 9501751792 0.10.1: 安全与业务逻辑加固、新品牌与部署加固
三轮审查修复(60+ 项),相对远端 main(b536672)的关键变更:
- 安全: 数据面 SSRF 拨号防护(防 DNS rebinding)/上游凭据剥离/登录防枚举
  与锁定态统一/可信代理(X-Forwarded-For)限流加固/会话版本失效机制/
  撤销即时传播/弱密钥拒绝启动/脱敏字节级重写(保签名契约)
- 业务逻辑: 裸 body 上传 panic/bootstrap 审计管线卡死/定价通配符优先级/
  全局工具可见性/调度器停机补跑/TOTP 挑战令牌消费顺序/熔断探针语义/
  >4MB 响应 token 计量/管理员重置密码作废会话 等
- 前端: 新 logo(语枢 AI 网关主题)/Provider 凭据异常警示/删除入口/
  后端错误消息透传/localStorage 敏感数据收敛
- 部署: CREDENTIAL_MASTER_KEY 持久化与弱值拒绝/Provider DELETE 接口/
  nginx 安全头/worker 内存限制
- 新增迁移 000029(key_hash 索引)/000030(usage_daily 币种维度)
2026-08-13 10:50:51 +08:00

46 lines
1.4 KiB
Go

package config
import "testing"
func TestProductionRequiresCoreDependencies(t *testing.T) {
t.Setenv("APP_ENV", "production")
t.Setenv("DATABASE_URL", "")
t.Setenv("REDIS_CRITICAL_URL", "")
t.Setenv("GATEWAY_BOOTSTRAP_API_KEY", "short")
t.Setenv("UPSTREAM_API_KEY", "")
cfg, err := Load()
if err != nil {
t.Fatalf("unexpected structural configuration error: %v", err)
}
if err := cfg.ValidateRuntime(); err == nil {
t.Fatal("expected production validation error")
}
}
func TestLocalDefaultsAreValid(t *testing.T) {
t.Setenv("APP_ENV", "local")
t.Setenv("UPSTREAM_BASE_URL", "https://example.com")
if _, err := Load(); err != nil {
t.Fatalf("unexpected error: %v", err)
}
}
func TestProductionCanDisableBootstrapCompatibility(t *testing.T) {
t.Setenv("APP_ENV", "production")
t.Setenv("DATABASE_URL", "postgres://gateway:secret@db.example/gateway?sslmode=require")
t.Setenv("REDIS_CRITICAL_URL", "rediss://redis.example/0")
t.Setenv("GATEWAY_BOOTSTRAP_API_KEY", "")
t.Setenv("GATEWAY_BOOTSTRAP_API_KEY_ENABLED", "false")
t.Setenv("UPSTREAM_FALLBACK_ENABLED", "false")
t.Setenv("CREDENTIAL_MASTER_KEY", "yP7sK9xR2mV4nQ8wT1uB3cE5fG6hJ0kL=")
t.Setenv("UPSTREAM_BASE_URL", "https://example.com")
cfg, err := Load()
if err != nil {
t.Fatal(err)
}
if err := cfg.ValidateRuntime(); err != nil {
t.Fatalf("disabled bootstrap compatibility should not require a key: %v", err)
}
}