Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
4.1 kB · 123 lines
Go
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124package config
import ( "testing")
func TestParseConfigRequestScopedErrors(t *testing.T) { const yamlConfig = `gemini-api-key: - api-key: gemini-key-1 request-scoped-errors: - status: 400 match: - "maximum_context_length" - "context_length_exceeded" match-regexr: - "maximum_context_length$" - "^context_length_exceeded" action: stop
interactions-api-key: - api-key: interactions-key-1 request-scoped-errors: - status: 400 match: - "invalid_argument" action: continue
codex-api-key: - api-key: codex-key-1 base-url: https://api.openai.com/v1 request-scoped-errors: - status: 400 match: - "context_window_exceeded" action: stop-and-cooldown
xai-api-key: - api-key: xai-key-1 base-url: https://api.x.ai/v1 request-scoped-errors: - status: 500 match: - "rate_limit_exceeded" action: continue-and-cooldown
claude-api-key: - api-key: claude-key-1 request-scoped-errors: - status: 400 match: - "prompt is too long" action: stop
openai-compatibility: - name: test-openai-compat base-url: https://api.openai.compat/v1 api-key-entries: - api-key: compat-key-1 request-scoped-errors: - status: 400 match: - maximum_context_length - context_length_exceeded match-regexr: - "maximum_context_length$" - "^context_length_exceeded" action: stop`
cfg, errParse := ParseConfigBytes([]byte(yamlConfig)) if errParse != nil { t.Fatalf("ParseConfigBytes() error = %v", errParse) }
if len(cfg.GeminiKey) != 1 || len(cfg.GeminiKey[0].RequestScopedErrors) != 1 { t.Fatalf("gemini[0].request-scoped-errors len = %d, want 1", len(cfg.GeminiKey[0].RequestScopedErrors)) } gRule := cfg.GeminiKey[0].RequestScopedErrors[0] if gRule.Status != 400 || len(gRule.Match) != 2 || len(gRule.MatchRegexr) != 2 || gRule.Action != "stop" { t.Fatalf("unexpected gemini rule: %+v", gRule) }
if len(cfg.InteractionsKey) != 1 || len(cfg.InteractionsKey[0].RequestScopedErrors) != 1 { t.Fatalf("interactions[0].request-scoped-errors len = %d, want 1", len(cfg.InteractionsKey[0].RequestScopedErrors)) } iRule := cfg.InteractionsKey[0].RequestScopedErrors[0] if iRule.Status != 400 || len(iRule.Match) != 1 || iRule.Action != "continue" { t.Fatalf("unexpected interactions rule: %+v", iRule) }
if len(cfg.CodexKey) != 1 || len(cfg.CodexKey[0].RequestScopedErrors) != 1 { t.Fatalf("codex[0].request-scoped-errors len = %d, want 1", len(cfg.CodexKey[0].RequestScopedErrors)) } codexRule := cfg.CodexKey[0].RequestScopedErrors[0] if codexRule.Status != 400 || codexRule.Action != "stop-and-cooldown" { t.Fatalf("unexpected codex rule: %+v", codexRule) }
if len(cfg.XAIKey) != 1 || len(cfg.XAIKey[0].RequestScopedErrors) != 1 { t.Fatalf("xai[0].request-scoped-errors len = %d, want 1", len(cfg.XAIKey[0].RequestScopedErrors)) } xaiRule := cfg.XAIKey[0].RequestScopedErrors[0] if xaiRule.Status != 500 || xaiRule.Action != "continue-and-cooldown" { t.Fatalf("unexpected xai rule: %+v", xaiRule) }
if len(cfg.ClaudeKey) != 1 || len(cfg.ClaudeKey[0].RequestScopedErrors) != 1 { t.Fatalf("claude[0].request-scoped-errors len = %d, want 1", len(cfg.ClaudeKey[0].RequestScopedErrors)) } claudeRule := cfg.ClaudeKey[0].RequestScopedErrors[0] if claudeRule.Status != 400 || claudeRule.Action != "stop" { t.Fatalf("unexpected claude rule: %+v", claudeRule) }
if len(cfg.OpenAICompatibility) != 1 || len(cfg.OpenAICompatibility[0].RequestScopedErrors) != 1 { t.Fatalf("openai-compatibility[0].request-scoped-errors len = %d, want 1", len(cfg.OpenAICompatibility[0].RequestScopedErrors)) } compatRule := cfg.OpenAICompatibility[0].RequestScopedErrors[0] if compatRule.Status != 400 || len(compatRule.Match) != 2 || len(compatRule.MatchRegexr) != 2 || compatRule.Action != "stop" { t.Fatalf("unexpected openai-compatibility rule: %+v", compatRule) }}