mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-03 14:39:26 +08:00
a80e8082ef
- Introduced `disable-codex-cloaking` to allow disabling enforced `User-Agent` and `Originator` headers for Codex requests. - Updated header application logic to conditionally include `codexUserAgent` and `codexOriginator` based on configuration. - Enhanced config diff tracking to detect changes in `disable-codex-cloaking`. - Expanded tests to cover new config behavior and header application scenarios.
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestLoadConfigOptional_CodexHeaderDefaults(t *testing.T) {
|
|
dir := t.TempDir()
|
|
configPath := filepath.Join(dir, "config.yaml")
|
|
configYAML := []byte(`
|
|
codex-header-defaults:
|
|
user-agent: " my-codex-client/1.0 "
|
|
beta-features: " feature-a,feature-b "
|
|
`)
|
|
if err := os.WriteFile(configPath, configYAML, 0o600); err != nil {
|
|
t.Fatalf("failed to write config: %v", err)
|
|
}
|
|
|
|
cfg, err := LoadConfigOptional(configPath, false)
|
|
if err != nil {
|
|
t.Fatalf("LoadConfigOptional() error = %v", err)
|
|
}
|
|
|
|
if got := cfg.CodexHeaderDefaults.UserAgent; got != "my-codex-client/1.0" {
|
|
t.Fatalf("UserAgent = %q, want %q", got, "my-codex-client/1.0")
|
|
}
|
|
if got := cfg.CodexHeaderDefaults.BetaFeatures; got != "feature-a,feature-b" {
|
|
t.Fatalf("BetaFeatures = %q, want %q", got, "feature-a,feature-b")
|
|
}
|
|
if cfg.Codex.DisableCodexCloaking {
|
|
t.Fatal("DisableCodexCloaking = true, want default false")
|
|
}
|
|
}
|
|
|
|
func TestLoadConfigOptional_CodexIdentityConfuse(t *testing.T) {
|
|
dir := t.TempDir()
|
|
configPath := filepath.Join(dir, "config.yaml")
|
|
configYAML := []byte(`
|
|
codex:
|
|
identity-confuse: true
|
|
disable-codex-cloaking: true
|
|
optimize-multi-agent-v2: true
|
|
`)
|
|
if err := os.WriteFile(configPath, configYAML, 0o600); err != nil {
|
|
t.Fatalf("failed to write config: %v", err)
|
|
}
|
|
|
|
cfg, err := LoadConfigOptional(configPath, false)
|
|
if err != nil {
|
|
t.Fatalf("LoadConfigOptional() error = %v", err)
|
|
}
|
|
|
|
if !cfg.Codex.IdentityConfuse {
|
|
t.Fatalf("IdentityConfuse = false, want true")
|
|
}
|
|
if !cfg.Codex.DisableCodexCloaking {
|
|
t.Fatal("DisableCodexCloaking = false, want true")
|
|
}
|
|
if !cfg.Codex.OptimizeMultiAgentV2 {
|
|
t.Fatalf("OptimizeMultiAgentV2 = false, want true")
|
|
}
|
|
}
|