103 lines
3.3 KiB
Go
103 lines
3.3 KiB
Go
package provider
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"slices"
|
|
"testing"
|
|
)
|
|
|
|
// A Command Code account on the Go plan lists every model Command Code
|
|
// lets Go use, not ten (01huadalang on Discord: "command 账号登录只有十个
|
|
// 模型，正确的应该是几十个"): its list is the Provider API's, asked without
|
|
// the key, less the models Go is refused; the CLI's own table stands in
|
|
// before it is fetched. testdata/commandcode_models.json is that list as
|
|
// api.commandcode.ai answered it (2026-10-01, 86 models).
|
|
func TestCommandCodeGoModels(t *testing.T) {
|
|
home := signIn(t)
|
|
writeFile(t, filepath.Join(home, ".commandcode", "auth.json"), map[string]any{"apiKey": "go-key", "userName": "gouser"})
|
|
list, err := os.ReadFile(filepath.Join("testdata", "commandcode_models.json"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.URL.Path {
|
|
case "/provider/v1/models":
|
|
if r.Header.Get("Authorization") != "" || r.Header.Get("x-api-key") != "" {
|
|
t.Error("the list was asked with the Go key")
|
|
}
|
|
_, _ = w.Write(list)
|
|
case "/alpha/billing/subscriptions":
|
|
_, _ = w.Write([]byte(`{"success":true,"data":{"planId":"individual-go-monthly","status":"active"}}`))
|
|
default:
|
|
w.WriteHeader(404)
|
|
}
|
|
}))
|
|
defer srv.Close()
|
|
oldAPI := cmdAPI
|
|
cmdAPI = srv.URL
|
|
defer func() { cmdAPI = oldAPI }()
|
|
cmdPlansSeen.Lock()
|
|
cmdPlansSeen.m = map[string]cmdSeen{}
|
|
cmdPlansSeen.Unlock()
|
|
|
|
p, found := find(All(), CommandCodePlanID)
|
|
if !found {
|
|
t.Fatal("no Command Code account")
|
|
}
|
|
// before a fetch, once the plan is read: the CLI's table, far more
|
|
// than ten
|
|
if _, _, ok := CommandCodeGenerate(context.Background(), p); !ok {
|
|
t.Fatal("not on Go")
|
|
}
|
|
p, _ = find(All(), CommandCodePlanID)
|
|
before := p.Available()
|
|
if len(before) != len(cmdGoModels) || len(before) < 40 {
|
|
t.Fatalf("before a fetch: %d models", len(before))
|
|
}
|
|
for _, m := range cmdGoModels {
|
|
if cmdGoRefused[m.ID] {
|
|
t.Errorf("the table offers %s, which Go is refused", m.ID)
|
|
}
|
|
}
|
|
|
|
ms, err := p.Fetch(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var got []string
|
|
for _, m := range ms {
|
|
got = append(got, m.ID)
|
|
}
|
|
if len(got) != 56 {
|
|
t.Fatalf("Go lists %d models: %v", len(got), got)
|
|
}
|
|
for _, id := range []string{"gpt-6-luna", "deepseek/deepseek-v4.1-flash", "moonshotai/Kimi-K2.6", "zai-org/GLM-5.2", "Qwen/Qwen3.7-Max", "stepfun/Step-5-Preview", "tencent/hy4-preview", "google/gemini-3.6-flash", "xai/grok-4.5"} {
|
|
if !slices.Contains(got, id) {
|
|
t.Errorf("%s missing from %v", id, got)
|
|
}
|
|
}
|
|
for _, id := range []string{"claude-opus-5-5", "claude-sonnet-5-5", "gpt-6-sol", "gpt-5.6-sol", "xai/grok-4.7", "google/gemini-3.8-flash", "xiaomi/mimo-v2.6-pro-ultraspeed"} {
|
|
if slices.Contains(got, id) {
|
|
t.Errorf("%s, which Go is refused, is listed", id)
|
|
}
|
|
}
|
|
// kept, and shown with the table's reasoning levels and pictures
|
|
p, _ = find(All(), CommandCodePlanID)
|
|
after := p.Available()
|
|
if len(after) != 56 {
|
|
t.Fatalf("after a fetch: %d models", len(after))
|
|
}
|
|
if e := p.Efforts("moonshotai/Kimi-K3"); !slices.Equal(e, []string{"low", "high", "max"}) {
|
|
t.Errorf("Kimi K3 efforts: %v", e)
|
|
}
|
|
for _, m := range after {
|
|
if m.ID == "moonshotai/Kimi-K3" && !m.Images {
|
|
t.Error("Kimi K3 takes no pictures")
|
|
}
|
|
}
|
|
}
|