48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package main
|
|
|
|
import "testing"
|
|
|
|
func TestCodexModelsURL(t *testing.T) {
|
|
got, err := codexModelsURL(" 0.144.1 ")
|
|
if err != nil {
|
|
t.Fatalf("codexModelsURL: %v", err)
|
|
}
|
|
want := "https://chatgpt.com/backend-api/codex/models?client_version=0.144.1"
|
|
if got != want {
|
|
t.Fatalf("codexModelsURL = %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestCountModels(t *testing.T) {
|
|
count, err := countModels([]byte(`{"models":[{"slug":"a"},{"slug":"b"}]}`))
|
|
if err != nil {
|
|
t.Fatalf("countModels(valid): %v", err)
|
|
}
|
|
if count != 2 {
|
|
t.Fatalf("countModels(valid) = %d, want 2", count)
|
|
}
|
|
|
|
// Upstream dumps may omit CPA catalog-required fields; counting must still work.
|
|
count, err = countModels([]byte(`{"models":[{"slug":"gpt-5.6-sol"}]}`))
|
|
if err != nil {
|
|
t.Fatalf("countModels(incomplete upstream model): %v", err)
|
|
}
|
|
if count != 1 {
|
|
t.Fatalf("countModels(incomplete upstream model) = %d, want 1", count)
|
|
}
|
|
|
|
count, err = countModels([]byte(`{"models":[]}`))
|
|
if err != nil {
|
|
t.Fatalf("countModels(empty): %v", err)
|
|
}
|
|
if count != 0 {
|
|
t.Fatalf("countModels(empty) = %d, want 0", count)
|
|
}
|
|
|
|
if _, err := countModels([]byte(`{"models":`)); err == nil {
|
|
t.Fatal("countModels(malformed) error = nil, want error")
|
|
}
|
|
if _, err := countModels([]byte(`{}`)); err == nil {
|
|
t.Fatal("countModels(missing models) error = nil, want error")
|
|
}
|
|
}
|