internal/config/xai_api_key_test.go

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

package config

import "testing"

func TestParseConfigBytesXAIConfig(t *testing.T) {
	defaultCfg, errDefault := ParseConfigBytes([]byte(`{}`))
	if errDefault != nil {
		t.Fatalf("ParseConfigBytes(default) error = %v", errDefault)
	}
	if defaultCfg.XAI.InjectXSearch {
		t.Fatal("xai.inject-x-search = true by default, want false")
	}

	enabledCfg, errEnabled := ParseConfigBytes([]byte(`xai:
  inject-x-search: true
`))
	if errEnabled != nil {
		t.Fatalf("ParseConfigBytes(enabled) error = %v", errEnabled)
	}
	if !enabledCfg.XAI.InjectXSearch {
		t.Fatal("xai.inject-x-search = false, want true")
	}
}

func TestParseConfigBytesXAIAPIKeyMatchesCodexShape(t *testing.T) {
	cfg, errParse := ParseConfigBytes([]byte(`xai-api-key:
  - api-key: " xai-key "
    priority: 3
    weight: 5
    prefix: " team-xai "
    base-url: " https://api.x.ai/v1 "
    websockets: true
    proxy-url: " http://proxy.local "
    headers:
      X-Custom: value
    models:
      - name: grok-4.5
        alias: grok-latest
        display-name: Grok Latest
        force-mapping: true
    excluded-models:
      - " grok-3-* "
    disable-cooling: true
    request-retry: 0
  - api-key: dropped
    base-url: " "
`))
	if errParse != nil {
		t.Fatalf("ParseConfigBytes() error = %v", errParse)
	}
	if len(cfg.XAIKey) != 1 {
		t.Fatalf("xai-api-key count = %d, want 1", len(cfg.XAIKey))
	}
	entry := cfg.XAIKey[0]
	if entry.APIKey != " xai-key " {
		t.Fatalf("api-key = %q, want original Codex-compatible value", entry.APIKey)
	}
	if entry.Priority != 3 {
		t.Fatalf("priority = %d, want 3", entry.Priority)
	}
	if entry.Weight == nil || *entry.Weight != 5 {
		t.Fatalf("weight = %v, want 5", entry.Weight)
	}
	if entry.Prefix != "team-xai" {
		t.Fatalf("prefix = %q, want team-xai", entry.Prefix)
	}
	if entry.BaseURL != "https://api.x.ai/v1" {
		t.Fatalf("base-url = %q, want https://api.x.ai/v1", entry.BaseURL)
	}
	if !entry.Websockets {
		t.Fatal("websockets = false, want true")
	}
	if entry.ProxyURL != " http://proxy.local " {
		t.Fatalf("proxy-url = %q, want original Codex-compatible value", entry.ProxyURL)
	}
	if entry.DisableCooling == nil || !*entry.DisableCooling {
		t.Fatalf("disable-cooling = %v, want true", entry.DisableCooling)
	}
	if entry.RequestRetry == nil || *entry.RequestRetry != 0 {
		t.Fatalf("request-retry = %v, want 0", entry.RequestRetry)
	}
	if entry.Headers["X-Custom"] != "value" {
		t.Fatalf("X-Custom header = %q, want value", entry.Headers["X-Custom"])
	}
	if len(entry.Models) != 1 {
		t.Fatalf("model count = %d, want 1", len(entry.Models))
	}
	model := entry.Models[0]
	if model.Name != "grok-4.5" || model.Alias != "grok-latest" || model.DisplayName != "Grok Latest" || !model.ForceMapping {
		t.Fatalf("unexpected model mapping: %+v", model)
	}
	if len(entry.ExcludedModels) != 1 || entry.ExcludedModels[0] != "grok-3-*" {
		t.Fatalf("excluded-models = %#v, want [grok-3-*]", entry.ExcludedModels)
	}
}