internal/config/oauth_request_scoped_errors_test.go

Rama main · réplica de lectura del repositorio original.

package 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"
  meta:
    - status: 400
      match:
        - "context_length_exceeded"
      action: "stop"
`

	cfg, err := ParseConfigBytes([]byte(yamlConfig))
	if err != nil {
		t.Fatalf("ParseConfigFromBytes failed: %v", err)
	}

	if len(cfg.OAuthRequestScopedErrors) != 8 {
		t.Fatalf("cfg.OAuthRequestScopedErrors len = %d, want 8", 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)
	}
}