mirror of
https://gh-proxy.org/https://github.com/klinsmaya/CLIProxyAPI.git
synced 2026-11-09 18:06:50 +08:00
test(models, executor): add XAI video model test and fix Codex User-Agent assertions
This commit is contained in:
@@ -0,0 +1,18 @@
|
||||
package registry
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestWithXAIBuiltinsIncludesVideoPreviewModel(t *testing.T) {
|
||||
models := WithXAIBuiltins(nil)
|
||||
|
||||
for _, model := range models {
|
||||
if model == nil {
|
||||
continue
|
||||
}
|
||||
if model.ID == xaiBuiltinVideo15PreviewModelID {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
t.Fatalf("expected xAI builtin model %s", xaiBuiltinVideo15PreviewModelID)
|
||||
}
|
||||
@@ -163,11 +163,11 @@ func TestApplyCodexWebsocketHeadersDefaultsToCurrentResponsesBeta(t *testing.T)
|
||||
if !strings.HasPrefix(codexUserAgent, codexOriginator+"/") {
|
||||
t.Fatalf("default Codex User-Agent = %s, want prefix %s/", codexUserAgent, codexOriginator)
|
||||
}
|
||||
if strings.HasPrefix(codexUserAgent, "codex-tui/") {
|
||||
t.Fatalf("default Codex User-Agent = %s, must not use stale codex-tui prefix", codexUserAgent)
|
||||
if !strings.HasPrefix(codexUserAgent, "codex-tui/") {
|
||||
t.Fatalf("default Codex User-Agent = %s, want codex-tui prefix", codexUserAgent)
|
||||
}
|
||||
if strings.Contains(codexUserAgent, "(codex-tui;") {
|
||||
t.Fatalf("default Codex User-Agent = %s, must not include stale codex-tui suffix", codexUserAgent)
|
||||
if !strings.Contains(codexUserAgent, "(codex-tui;") {
|
||||
t.Fatalf("default Codex User-Agent = %s, want codex-tui suffix", codexUserAgent)
|
||||
}
|
||||
if got := headers.Get("Originator"); got != codexOriginator {
|
||||
t.Fatalf("Originator = %s, want %s", got, codexOriginator)
|
||||
|
||||
Reference in New Issue
Block a user