49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
//go:build e2e
|
||
|
||
package integration
|
||
|
||
import (
|
||
"os"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
// =============================================================================
|
||
// E2E Mock 模式支持
|
||
// =============================================================================
|
||
// 当 E2E_MOCK=true 时，使用本地 Mock 响应替代真实 API 调用。
|
||
// 这允许在没有真实 API Key 的环境（如 CI）中验证基本的请求/响应流程。
|
||
|
||
// isMockMode 检查是否启用 Mock 模式
|
||
func isMockMode() bool {
|
||
return strings.EqualFold(os.Getenv("E2E_MOCK"), "true")
|
||
}
|
||
|
||
// skipIfNoRealAPI 如果未配置真实 API Key 且不在 Mock 模式，则跳过测试
|
||
func skipIfNoRealAPI(t *testing.T) {
|
||
t.Helper()
|
||
if isMockMode() {
|
||
return // Mock 模式下不跳过
|
||
}
|
||
claudeKey := strings.TrimSpace(os.Getenv(claudeAPIKeyEnv))
|
||
geminiKey := strings.TrimSpace(os.Getenv(geminiAPIKeyEnv))
|
||
if claudeKey == "" && geminiKey == "" {
|
||
t.Skip("未设置 API Key 且未启用 Mock 模式，跳过测试")
|
||
}
|
||
}
|
||
|
||
// =============================================================================
|
||
// API Key 脱敏（Task 6.10）
|
||
// =============================================================================
|
||
|
||
// safeLogKey 安全地记录 API Key（仅显示前 8 位）
|
||
func safeLogKey(t *testing.T, prefix string, key string) {
|
||
t.Helper()
|
||
key = strings.TrimSpace(key)
|
||
if len(key) <= 8 {
|
||
t.Logf("%s: ***（长度: %d）", prefix, len(key))
|
||
return
|
||
}
|
||
t.Logf("%s: %s...（长度: %d）", prefix, key[:8], len(key))
|
||
}
|