Add projects

backend/internal/config/xai_api_key_test.go

@@ -0,0 +1,95 @@
+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)
+	}
+}