139 lines
3.7 KiB
Go
139 lines
3.7 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_PreventsClaudeModelFromFallingBackToGpt54(t *testing.T) {
|
|
account := &Account{
|
|
Credentials: map[string]any{},
|
|
}
|
|
|
|
withoutDefault := normalizeCodexModel(resolveOpenAIForwardModel(account, "claude-opus-4-6", ""))
|
|
if withoutDefault != "gpt-5.4" {
|
|
t.Fatalf("normalizeCodexModel(...) = %q, want %q", withoutDefault, "gpt-5.4")
|
|
}
|
|
|
|
withDefault := normalizeCodexModel(resolveOpenAIForwardModel(account, "claude-opus-4-6", "gpt-5.4"))
|
|
if withDefault != "gpt-5.4" {
|
|
t.Fatalf("normalizeCodexModel(...) = %q, want %q", withDefault, "gpt-5.4")
|
|
}
|
|
}
|
|
|
|
func TestNormalizeCodexModel(t *testing.T) {
|
|
cases := map[string]string{
|
|
"gpt-5.3-codex-spark": "gpt-5.3-codex-spark",
|
|
"gpt-5.3-codex-spark-high": "gpt-5.3-codex-spark",
|
|
"gpt-5.3-codex-spark-xhigh": "gpt-5.3-codex-spark",
|
|
"gpt-5.3": "gpt-5.3-codex",
|
|
"gpt-image-2": "gpt-image-2",
|
|
}
|
|
|
|
for input, expected := range cases {
|
|
if got := normalizeCodexModel(input); got != expected {
|
|
t.Fatalf("normalizeCodexModel(%q) = %q, want %q", input, got, expected)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNormalizeOpenAIModelForUpstream(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
account *Account
|
|
model string
|
|
want string
|
|
}{
|
|
{
|
|
name: "oauth keeps codex normalization behavior",
|
|
account: &Account{Type: AccountTypeOAuth},
|
|
model: "gemini-3-flash-preview",
|
|
want: "gpt-5.4",
|
|
},
|
|
{
|
|
name: "apikey preserves custom compatible model",
|
|
account: &Account{Type: AccountTypeAPIKey},
|
|
model: "gemini-3-flash-preview",
|
|
want: "gemini-3-flash-preview",
|
|
},
|
|
{
|
|
name: "apikey preserves official non codex model",
|
|
account: &Account{Type: AccountTypeAPIKey},
|
|
model: "gpt-4.1",
|
|
want: "gpt-4.1",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := normalizeOpenAIModelForUpstream(tt.account, tt.model); got != tt.want {
|
|
t.Fatalf("normalizeOpenAIModelForUpstream(...) = %q, want %q", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|