104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
package service
|
|
|
|
import "testing"
|
|
|
|
func TestResolveOpenAIForwardModel(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
account *Account
|
|
requestedModel string
|
|
defaultMappedModel string
|
|
expectedModel string
|
|
}{
|
|
{
|
|
name: "falls back to group default when account has no mapping",
|
|
account: &Account{
|
|
Credentials: map[string]any{},
|
|
},
|
|
requestedModel: "gpt-5.4",
|
|
defaultMappedModel: "gpt-4o-mini",
|
|
expectedModel: "gpt-4o-mini",
|
|
},
|
|
{
|
|
name: "preserves exact passthrough mapping instead of group default",
|
|
account: &Account{
|
|
Credentials: map[string]any{
|
|
"model_mapping": map[string]any{
|
|
"gpt-5.4": "gpt-5.4",
|
|
},
|
|
},
|
|
},
|
|
requestedModel: "gpt-5.4",
|
|
defaultMappedModel: "gpt-4o-mini",
|
|
expectedModel: "gpt-5.4",
|
|
},
|
|
{
|
|
name: "preserves wildcard passthrough mapping instead of group default",
|
|
account: &Account{
|
|
Credentials: map[string]any{
|
|
"model_mapping": map[string]any{
|
|
"gpt-*": "gpt-5.4",
|
|
},
|
|
},
|
|
},
|
|
requestedModel: "gpt-5.4",
|
|
defaultMappedModel: "gpt-4o-mini",
|
|
expectedModel: "gpt-5.4",
|
|
},
|
|
{
|
|
name: "uses account remap when explicit target differs",
|
|
account: &Account{
|
|
Credentials: map[string]any{
|
|
"model_mapping": map[string]any{
|
|
"gpt-5": "gpt-5.4",
|
|
},
|
|
},
|
|
},
|
|
requestedModel: "gpt-5",
|
|
defaultMappedModel: "gpt-4o-mini",
|
|
expectedModel: "gpt-5.4",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := resolveOpenAIForwardModel(tt.account, tt.requestedModel, tt.defaultMappedModel); got != tt.expectedModel {
|
|
t.Fatalf("resolveOpenAIForwardModel(...) = %q, want %q", got, tt.expectedModel)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestResolveOpenAIForwardModel_PreventsClaudeModelFromFallingBackToGpt51(t *testing.T) {
|
|
account := &Account{
|
|
Credentials: map[string]any{},
|
|
}
|
|
|
|
withoutDefault := resolveOpenAIUpstreamModel(resolveOpenAIForwardModel(account, "claude-opus-4-6", ""))
|
|
if withoutDefault != "gpt-5.1" {
|
|
t.Fatalf("resolveOpenAIUpstreamModel(...) = %q, want %q", withoutDefault, "gpt-5.1")
|
|
}
|
|
|
|
withDefault := resolveOpenAIUpstreamModel(resolveOpenAIForwardModel(account, "claude-opus-4-6", "gpt-5.4"))
|
|
if withDefault != "gpt-5.4" {
|
|
t.Fatalf("resolveOpenAIUpstreamModel(...) = %q, want %q", withDefault, "gpt-5.4")
|
|
}
|
|
}
|
|
|
|
func TestResolveOpenAIUpstreamModel(t *testing.T) {
|
|
cases := map[string]string{
|
|
"gpt-5.3-codex-spark": "gpt-5.3-codex-spark",
|
|
"gpt 5.3 codex spark": "gpt-5.3-codex-spark",
|
|
" openai/gpt-5.3-codex-spark ": "gpt-5.3-codex-spark",
|
|
"gpt-5.3-codex-spark-high": "gpt-5.3-codex",
|
|
"gpt-5.3-codex-spark-xhigh": "gpt-5.3-codex",
|
|
"gpt-5.3": "gpt-5.3-codex",
|
|
}
|
|
|
|
for input, expected := range cases {
|
|
if got := resolveOpenAIUpstreamModel(input); got != expected {
|
|
t.Fatalf("resolveOpenAIUpstreamModel(%q) = %q, want %q", input, got, expected)
|
|
}
|
|
}
|
|
}
|