Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
2.9 kB · 115 lines
Go
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116package config
import ( "testing")
func TestParseConfigOAuthRequestScopedErrors(t *testing.T) { const yamlConfig = `oauth-request-scoped-errors: vertex: - status: 400 match: - "maximum_context_length" - "context_length_exceeded" match-regexr: - "maximum_context_length$" - "^context_length_exceeded" action: "stop" aistudio: - status: 400 match: - "invalid_argument" action: "continue" antigravity: - status: 500 match: - "internal_server_error" action: "stop-and-cooldown" claude: - status: 429 match: - "rate_limit" action: "continue-and-cooldown" codex: - status: 400 match: - "context_window_exceeded" action: "stop" kimi: - status: 400 match: - "length_limit" action: "stop" xai: - status: 400 match: - "max_tokens_exceeded" action: "stop"`
cfg, err := ParseConfigBytes([]byte(yamlConfig)) if err != nil { t.Fatalf("ParseConfigFromBytes failed: %v", err) }
if len(cfg.OAuthRequestScopedErrors) != 7 { t.Fatalf("cfg.OAuthRequestScopedErrors len = %d, want 7", len(cfg.OAuthRequestScopedErrors)) }
vertexRules, ok := cfg.OAuthRequestScopedErrors["vertex"] if !ok || len(vertexRules) != 1 { t.Fatalf("vertex rules missing or len != 1: %#v", vertexRules) } rule := vertexRules[0] if rule.Status != 400 || rule.Action != "stop" { t.Errorf("unexpected vertex rule: %+v", rule) } if len(rule.Match) != 2 || len(rule.MatchRegexr) != 2 { t.Errorf("unexpected vertex match len: %+v", rule) }}
func TestSanitizeOAuthRequestScopedErrors(t *testing.T) { cfg := &Config{ OAuthRequestScopedErrors: map[string][]RequestScopedErrorRule{ " Vertex ": { { Status: 400, Match: []string{" context_length ", ""}, MatchRegexr: []string{" ^error.* ", ""}, Action: " STOP ", }, { Status: 0, // invalid status Match: []string{"foo"}, Action: "stop", }, { Status: 400, // missing match / action }, }, " empty-channel ": {}, }, }
cfg.SanitizeOAuthRequestScopedErrors()
if len(cfg.OAuthRequestScopedErrors) != 1 { t.Fatalf("expected 1 sanitized channel, got %d", len(cfg.OAuthRequestScopedErrors)) }
rules := cfg.OAuthRequestScopedErrors["vertex"] if len(rules) != 1 { t.Fatalf("expected 1 rule for vertex, got %d", len(rules)) } if rules[0].Status != 400 || rules[0].Action != "stop" { t.Errorf("unexpected sanitized rule: %+v", rules[0]) } if len(rules[0].Match) != 1 || rules[0].Match[0] != "context_length" { t.Errorf("unexpected sanitized match: %+v", rules[0].Match) } if len(rules[0].MatchRegexr) != 1 || rules[0].MatchRegexr[0] != "^error.*" { t.Errorf("unexpected sanitized regexr: %+v", rules[0].MatchRegexr) }}