wehub-resource-sync
44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v7/internal/registry"
|
|
)
|
|
|
|
func TestResolveCodexWebSearchTargetModelNeverUsesClaudeName(t *testing.T) {
|
|
got := resolveCodexWebSearchTargetModel("")
|
|
if got != defaultCodexWebSearchModel {
|
|
t.Fatalf("empty config = %q, want %q", got, defaultCodexWebSearchModel)
|
|
}
|
|
if got := resolveCodexWebSearchTargetModel("gpt-5.5"); got != "gpt-5.5" {
|
|
t.Fatalf("configured = %q", got)
|
|
}
|
|
}
|
|
|
|
func TestResolveXAIWebSearchTargetModelNeverUsesClaudeName(t *testing.T) {
|
|
got := resolveXAIWebSearchTargetModel("")
|
|
if got != defaultXAIWebSearchModel {
|
|
t.Fatalf("empty config = %q, want %q", got, defaultXAIWebSearchModel)
|
|
}
|
|
}
|
|
|
|
func TestResolveAntigravityWebSearchTargetModelConfiguredWins(t *testing.T) {
|
|
if got := resolveAntigravityWebSearchTargetModel("my-gemini", "claude-sonnet-4-6"); got != "my-gemini" {
|
|
t.Fatalf("configured = %q", got)
|
|
}
|
|
}
|
|
|
|
func TestResolveAntigravityWebSearchTargetModelFromRegistry(t *testing.T) {
|
|
reg := registry.GetGlobalRegistry()
|
|
const clientID = "test-claude-web-search-router-antigravity"
|
|
reg.RegisterClient(clientID, "antigravity", []*registry.ModelInfo{
|
|
{ID: "gemini-web-search-test", SupportsWebSearch: true},
|
|
})
|
|
t.Cleanup(func() { reg.UnregisterClient(clientID) })
|
|
got := resolveAntigravityWebSearchTargetModel("", "claude-sonnet-4-6")
|
|
if got != "gemini-web-search-test" {
|
|
t.Fatalf("fallback = %q, want gemini-web-search-test", got)
|
|
}
|
|
}
|