58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package claudecode
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/chenhg5/cc-connect/core"
|
|
)
|
|
|
|
func TestConfiguredModels_BoundaryConditions(t *testing.T) {
|
|
a := &Agent{
|
|
providers: []core.ProviderConfig{
|
|
{Models: []core.ModelOption{{Name: "first"}}},
|
|
{Models: []core.ModelOption{{Name: "second"}}},
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
activeIdx int
|
|
wantNil bool
|
|
wantName string
|
|
}{
|
|
{name: "negative index", activeIdx: -1, wantNil: true},
|
|
{name: "out of range", activeIdx: 2, wantNil: true},
|
|
{name: "valid index", activeIdx: 1, wantName: "second"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
a.activeIdx = tt.activeIdx
|
|
got := a.configuredModels()
|
|
if tt.wantNil {
|
|
if got != nil {
|
|
t.Fatalf("configuredModels() = %v, want nil", got)
|
|
}
|
|
return
|
|
}
|
|
if len(got) != 1 || got[0].Name != tt.wantName {
|
|
t.Fatalf("configuredModels() = %v, want %q", got, tt.wantName)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetModel_PrefersActiveProviderModel(t *testing.T) {
|
|
a := &Agent{
|
|
model: "sonnet",
|
|
providers: []core.ProviderConfig{
|
|
{Name: "anthropic", Model: "opus"},
|
|
},
|
|
activeIdx: 0,
|
|
}
|
|
|
|
if got := a.GetModel(); got != "opus" {
|
|
t.Fatalf("GetModel() = %q, want opus", got)
|
|
}
|
|
}
|