opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add test

Browse source
This commit is contained in:
qiangmzsx 2026-03-01 10:34:33 +08:00
parent 632c1128f6
commit 75e433e4b8
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
pkg/providers/factory_provider_test.go
View file

@ -34,6 +34,12 @@ func TestExtractProtocol(t *testing.T) {
wantProtocol: "anthropic", wantProtocol: "anthropic",
wantModelID: "claude-sonnet-4.6", wantModelID: "claude-sonnet-4.6",
}, },
{
name: "claude with prefix",
model: "claude/claude-sonnet-4.6",
wantProtocol: "claude",
wantModelID: "claude-sonnet-4.6",
},
{ {
name: "no prefix - defaults to openai", name: "no prefix - defaults to openai",
model: "gpt-4o", model: "gpt-4o",
@ -154,6 +160,25 @@ func TestCreateProviderFromConfig_Anthropic(t *testing.T) {
} }
} }
func TestCreateProviderFromConfig_Claudue(t *testing.T) {
cfg := &config.ModelConfig{
ModelName: "test-anthropic",
Model: "claude/claude-sonnet-4.6",
APIKey: "test-key",
}
provider, modelID, err := CreateProviderFromConfig(cfg)
if err != nil {
t.Fatalf("CreateProviderFromConfig() error = %v", err)
}
if provider == nil {
t.Fatal("CreateProviderFromConfig() returned nil provider")
}
if modelID != "claude-sonnet-4.6" {
t.Errorf("modelID = %q, want %q", modelID, "claude-sonnet-4.6")
}
}
func TestCreateProviderFromConfig_Antigravity(t *testing.T) { func TestCreateProviderFromConfig_Antigravity(t *testing.T) {
cfg := &config.ModelConfig{ cfg := &config.ModelConfig{
ModelName: "test-antigravity", ModelName: "test-antigravity",