95 lines
2.9 KiB
Go
95 lines
2.9 KiB
Go
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)
|
|
}
|
|
}
|