apale7
cf3dcec399
Avoid collision with the existing Opswitch product while keeping the repository name as opencode-provider-switch. This aligns the CLI, provider key, config paths, examples, and Trellis history under one public-facing name.
455 lines
15 KiB
Go
455 lines
15 KiB
Go
package opencode
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestGlobalConfigDirIgnoresOpencodeConfigDir(t *testing.T) {
|
|
t.Setenv("OPENCODE_CONFIG_DIR", "/tmp/custom-opencode")
|
|
t.Setenv("XDG_CONFIG_HOME", "/tmp/xdg-home")
|
|
|
|
got := GlobalConfigDir()
|
|
want := filepath.Join("/tmp/xdg-home", "opencode")
|
|
if got != want {
|
|
t.Fatalf("GlobalConfigDir() = %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestResolveGlobalConfigPathPrecedence(t *testing.T) {
|
|
root := t.TempDir()
|
|
t.Setenv("XDG_CONFIG_HOME", root)
|
|
t.Setenv("OPENCODE_CONFIG_DIR", filepath.Join(root, "ignored"))
|
|
|
|
dir := filepath.Join(root, "opencode")
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatalf("mkdir: %v", err)
|
|
}
|
|
|
|
path, existed := ResolveGlobalConfigPath()
|
|
if existed {
|
|
t.Fatalf("expected existed=false before files are created")
|
|
}
|
|
wantDefault := filepath.Join(dir, "opencode.jsonc")
|
|
if path != wantDefault {
|
|
t.Fatalf("default path = %q, want %q", path, wantDefault)
|
|
}
|
|
|
|
if err := os.WriteFile(filepath.Join(dir, "config.json"), []byte("{}\n"), 0o600); err != nil {
|
|
t.Fatalf("write config.json: %v", err)
|
|
}
|
|
path, existed = ResolveGlobalConfigPath()
|
|
if !existed || path != filepath.Join(dir, "config.json") {
|
|
t.Fatalf("expected config.json, got path=%q existed=%v", path, existed)
|
|
}
|
|
|
|
if err := os.WriteFile(filepath.Join(dir, "opencode.json"), []byte("{}\n"), 0o600); err != nil {
|
|
t.Fatalf("write opencode.json: %v", err)
|
|
}
|
|
path, existed = ResolveGlobalConfigPath()
|
|
if !existed || path != filepath.Join(dir, "opencode.json") {
|
|
t.Fatalf("expected opencode.json, got path=%q existed=%v", path, existed)
|
|
}
|
|
|
|
if err := os.WriteFile(filepath.Join(dir, "opencode.jsonc"), []byte("{}\n"), 0o600); err != nil {
|
|
t.Fatalf("write opencode.jsonc: %v", err)
|
|
}
|
|
path, existed = ResolveGlobalConfigPath()
|
|
if !existed || path != filepath.Join(dir, "opencode.jsonc") {
|
|
t.Fatalf("expected opencode.jsonc, got path=%q existed=%v", path, existed)
|
|
}
|
|
}
|
|
|
|
func TestValidateOcswitchProvider(t *testing.T) {
|
|
raw := Raw{}
|
|
aliases := []string{"gpt-5.4", "gpt-5.4-mini"}
|
|
baseURL := "http://127.0.0.1:9982/v1"
|
|
apiKey := "ocswitch-local"
|
|
EnsureOcswitchProvider(raw, baseURL, apiKey, aliases)
|
|
|
|
providerRaw, _ := raw["provider"].(map[string]any)
|
|
providerEntry, _ := providerRaw[ProviderKey].(map[string]any)
|
|
opts, _ := providerEntry["options"].(map[string]any)
|
|
if got, ok := opts["setCacheKey"].(bool); !ok || !got {
|
|
t.Fatalf("provider.%s.options.setCacheKey = %#v, want true", ProviderKey, opts["setCacheKey"])
|
|
}
|
|
|
|
if err := ValidateOcswitchProvider(raw, baseURL, apiKey, aliases); err != nil {
|
|
t.Fatalf("ValidateOcswitchProvider() unexpected error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestEnsureOcswitchProviderPreservesExistingModelMetadata(t *testing.T) {
|
|
raw := Raw{
|
|
"$schema": "https://opencode.ai/config.json",
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{
|
|
"gpt-5.4": map[string]any{
|
|
"name": "custom-display-name",
|
|
"limit": map[string]any{
|
|
"context": float64(272000),
|
|
"output": float64(128000),
|
|
},
|
|
"cost": map[string]any{
|
|
"input": float64(1.75),
|
|
"output": float64(14),
|
|
},
|
|
"variants": map[string]any{
|
|
"high": map[string]any{"reasoningEffort": "high"},
|
|
},
|
|
"options": map[string]any{"serviceTier": "priority"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
changed := EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
if changed {
|
|
t.Fatal("EnsureOcswitchProvider() reported change for unchanged same-name alias")
|
|
}
|
|
|
|
providerRaw := raw["provider"].(map[string]any)
|
|
providerEntry := providerRaw[ProviderKey].(map[string]any)
|
|
models := providerEntry["models"].(map[string]any)
|
|
model := models["gpt-5.4"].(map[string]any)
|
|
if got := model["name"]; got != "custom-display-name" {
|
|
t.Fatalf("model name = %#v, want custom-display-name preserved", got)
|
|
}
|
|
if _, ok := model["limit"].(map[string]any); !ok {
|
|
t.Fatalf("model limit metadata missing: %#v", model["limit"])
|
|
}
|
|
if _, ok := model["cost"].(map[string]any); !ok {
|
|
t.Fatalf("model cost metadata missing: %#v", model["cost"])
|
|
}
|
|
if _, ok := model["variants"].(map[string]any); !ok {
|
|
t.Fatalf("model variants metadata missing: %#v", model["variants"])
|
|
}
|
|
if _, ok := model["options"].(map[string]any); !ok {
|
|
t.Fatalf("model options metadata missing: %#v", model["options"])
|
|
}
|
|
}
|
|
|
|
func TestEnsureOcswitchProviderDoesNotPanicOnComparableMetadata(t *testing.T) {
|
|
raw := Raw{
|
|
"$schema": "https://opencode.ai/config.json",
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{
|
|
"gpt-5.4": map[string]any{
|
|
"name": "custom-display-name",
|
|
"tags": []any{"reasoning", "priority"},
|
|
"variants": []any{
|
|
map[string]any{"name": "high", "effort": "high"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
t.Fatalf("EnsureOcswitchProvider() panicked with slice metadata: %v", r)
|
|
}
|
|
}()
|
|
|
|
changed := EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
if changed {
|
|
t.Fatal("EnsureOcswitchProvider() reported change for unchanged alias metadata with slices")
|
|
}
|
|
}
|
|
|
|
func TestValidateOcswitchProviderAllowsCustomModelMetadata(t *testing.T) {
|
|
raw := Raw{
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{
|
|
"gpt-5.4": map[string]any{
|
|
"name": "custom-display-name",
|
|
"options": map[string]any{"serviceTier": "priority"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
if err := ValidateOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"}); err != nil {
|
|
t.Fatalf("ValidateOcswitchProvider() unexpected error for custom metadata: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataReplacesExistingProviderOnly(t *testing.T) {
|
|
raw := Raw{
|
|
"$schema": "https://opencode.ai/config.json",
|
|
"model": "ocswitch/gpt-5.4",
|
|
"provider": map[string]any{
|
|
"anthropic": map[string]any{"npm": "@ai-sdk/anthropic"},
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{"gpt-5.4": map[string]any{"name": "gpt-5.4"}},
|
|
},
|
|
"openai": map[string]any{"npm": "@ai-sdk/openai"},
|
|
},
|
|
"small_model": "ocswitch/gpt-5.4-mini",
|
|
}
|
|
original := []byte("{\n \"model\": \"ocswitch/old\",\n \"provider\": {\n \"anthropic\": {\"npm\": \"@ai-sdk/anthropic\"},\n \"ocswitch\": {\n \"npm\": \"old\",\n \"options\": {\"baseURL\": \"http://old/v1\"},\n \"models\": {\"old\": {\"name\": \"old\"}}\n },\n \"openai\": {\"npm\": \"@ai-sdk/openai\"}\n },\n \"small_model\": \"ocswitch/old-mini\"\n}\n")
|
|
|
|
got, err := patchProviderDocument(original, raw)
|
|
if err != nil {
|
|
t.Fatalf("patchProviderDocument() error: %v", err)
|
|
}
|
|
assertValidJSON(t, got)
|
|
assertStringOrder(t, string(got), []string{`"model"`, `"provider"`, `"small_model"`})
|
|
assertStringOrder(t, string(got), []string{`"anthropic"`, `"ocswitch"`, `"openai"`})
|
|
if strings.Contains(string(got), `"npm": "old"`) {
|
|
t.Fatalf("old provider.ocswitch content still present: %s", string(got))
|
|
}
|
|
var saved Raw
|
|
if err := json.Unmarshal(got, &saved); err != nil {
|
|
t.Fatalf("unmarshal patched json: %v", err)
|
|
}
|
|
if err := ValidateOcswitchProvider(saved, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"}); err != nil {
|
|
t.Fatalf("ValidateOcswitchProvider(saved) error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataInsertsOcswitchWithoutReorderingProviderKeys(t *testing.T) {
|
|
raw := Raw{
|
|
"provider": map[string]any{
|
|
"anthropic": map[string]any{"npm": "@ai-sdk/anthropic"},
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{"gpt-5.4": map[string]any{"name": "gpt-5.4"}},
|
|
},
|
|
"openai": map[string]any{"npm": "@ai-sdk/openai"},
|
|
},
|
|
}
|
|
original := []byte("{\n \"provider\": {\n \"anthropic\": {\"npm\": \"@ai-sdk/anthropic\"},\n \"openai\": {\"npm\": \"@ai-sdk/openai\"}\n },\n \"model\": \"ocswitch/gpt-5.4\"\n}\n")
|
|
|
|
got, err := patchProviderDocument(original, raw)
|
|
if err != nil {
|
|
t.Fatalf("patchProviderDocument() error: %v", err)
|
|
}
|
|
assertValidJSON(t, got)
|
|
assertStringOrder(t, string(got), []string{`"anthropic"`, `"openai"`, `"ocswitch"`})
|
|
}
|
|
|
|
func TestRenderSaveDataInsertsProviderAtTopLevelEnd(t *testing.T) {
|
|
raw := Raw{
|
|
"model": "ocswitch/gpt-5.4",
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{"gpt-5.4": map[string]any{"name": "gpt-5.4"}},
|
|
},
|
|
},
|
|
"small_model": "ocswitch/gpt-5.4-mini",
|
|
}
|
|
original := []byte("{\n \"model\": \"ocswitch/gpt-5.4\",\n \"small_model\": \"ocswitch/gpt-5.4-mini\"\n}\n")
|
|
|
|
got, err := patchProviderDocument(original, raw)
|
|
if err != nil {
|
|
t.Fatalf("patchProviderDocument() error: %v", err)
|
|
}
|
|
assertValidJSON(t, got)
|
|
assertStringOrder(t, string(got), []string{`"model"`, `"small_model"`, `"provider"`})
|
|
}
|
|
|
|
func TestRenderSaveDataAcceptsJSONCAndProducesValidJSON(t *testing.T) {
|
|
raw := Raw{
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
"models": map[string]any{"gpt-5.4": map[string]any{"name": "gpt-5.4"}},
|
|
},
|
|
},
|
|
}
|
|
original := []byte("{\n // comment\n \"provider\": {\n \"openai\": {\"npm\": \"@ai-sdk/openai\"},\n },\n}\n")
|
|
|
|
got, err := patchProviderDocument(original, raw)
|
|
if err != nil {
|
|
t.Fatalf("patchProviderDocument() error: %v", err)
|
|
}
|
|
assertValidJSON(t, got)
|
|
if bytes.Contains(got, []byte("// comment")) {
|
|
t.Fatalf("expected normalized json output without comments, got %s", string(got))
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataRejectsInvalidJSONC(t *testing.T) {
|
|
raw := Raw{
|
|
"provider": map[string]any{
|
|
ProviderKey: map[string]any{
|
|
"npm": "@ai-sdk/openai",
|
|
"name": ProviderName,
|
|
"options": map[string]any{
|
|
"baseURL": "http://127.0.0.1:9982/v1",
|
|
"apiKey": "ocswitch-local",
|
|
"setCacheKey": true,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
if _, err := patchProviderDocument([]byte(`{"provider": {`), raw); err == nil {
|
|
t.Fatal("expected invalid json/jsonc error")
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataRejectsNonObjectProvider(t *testing.T) {
|
|
raw := Raw{}
|
|
EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
|
|
if _, err := patchProviderDocument([]byte(`{"provider":"bad"}`), raw); err == nil {
|
|
t.Fatal("expected provider object error")
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataRejectsNonObjectTopLevel(t *testing.T) {
|
|
raw := Raw{}
|
|
EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
|
|
if _, err := patchProviderDocument([]byte(`[]`), raw); err == nil {
|
|
t.Fatal("expected top-level object error")
|
|
}
|
|
if _, err := patchProviderDocument([]byte(`{} trailing`), raw); err == nil {
|
|
t.Fatal("expected single top-level object error")
|
|
}
|
|
}
|
|
|
|
func TestRenderSaveDataWritesValidJSONToDisk(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "opencode.jsonc")
|
|
if err := os.WriteFile(path, []byte("{\n \"model\": \"ocswitch/gpt-5.4\",\n \"provider\": {\n \"openai\": {\"npm\": \"@ai-sdk/openai\"}\n }\n}\n"), 0o600); err != nil {
|
|
t.Fatalf("write seed config: %v", err)
|
|
}
|
|
raw := Raw{}
|
|
EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
|
|
if err := Save(path, raw); err != nil {
|
|
t.Fatalf("Save() error: %v", err)
|
|
}
|
|
got, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("read saved config: %v", err)
|
|
}
|
|
assertValidJSON(t, got)
|
|
loaded, err := Load(path)
|
|
if err != nil {
|
|
t.Fatalf("Load(saved) error: %v", err)
|
|
}
|
|
if err := ValidateOcswitchProvider(loaded, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"}); err != nil {
|
|
t.Fatalf("ValidateOcswitchProvider(loaded) error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSavePreservesExistingModelMetadataForSameAlias(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "opencode.jsonc")
|
|
seed := []byte("{\n \"$schema\": \"https://opencode.ai/config.json\",\n \"provider\": {\n \"ocswitch\": {\n \"npm\": \"@ai-sdk/openai\",\n \"name\": \"OpenCode Provider Switch CLI\",\n \"options\": {\n \"baseURL\": \"http://127.0.0.1:9982/v1\",\n \"apiKey\": \"ocswitch-local\",\n \"setCacheKey\": true\n },\n \"models\": {\n \"gpt-5.4\": {\n \"name\": \"custom-display-name\",\n \"limit\": {\n \"context\": 272000,\n \"output\": 128000\n },\n \"options\": {\n \"serviceTier\": \"priority\"\n }\n }\n }\n }\n }\n}\n")
|
|
if err := os.WriteFile(path, seed, 0o600); err != nil {
|
|
t.Fatalf("write seed config: %v", err)
|
|
}
|
|
|
|
raw, err := Load(path)
|
|
if err != nil {
|
|
t.Fatalf("Load(seed) error: %v", err)
|
|
}
|
|
changed := EnsureOcswitchProvider(raw, "http://127.0.0.1:9982/v1", "ocswitch-local", []string{"gpt-5.4"})
|
|
if changed {
|
|
t.Fatal("EnsureOcswitchProvider() reported change for preserved same-name alias metadata")
|
|
}
|
|
if err := Save(path, raw); err != nil {
|
|
t.Fatalf("Save() error: %v", err)
|
|
}
|
|
|
|
loaded, err := Load(path)
|
|
if err != nil {
|
|
t.Fatalf("Load(saved) error: %v", err)
|
|
}
|
|
providerRaw := loaded["provider"].(map[string]any)
|
|
providerEntry := providerRaw[ProviderKey].(map[string]any)
|
|
models := providerEntry["models"].(map[string]any)
|
|
model := models["gpt-5.4"].(map[string]any)
|
|
if got := model["name"]; got != "custom-display-name" {
|
|
t.Fatalf("saved model name = %#v, want custom-display-name preserved", got)
|
|
}
|
|
if _, ok := model["limit"].(map[string]any); !ok {
|
|
t.Fatalf("saved limit metadata missing: %#v", model["limit"])
|
|
}
|
|
if _, ok := model["options"].(map[string]any); !ok {
|
|
t.Fatalf("saved options metadata missing: %#v", model["options"])
|
|
}
|
|
}
|
|
|
|
func assertValidJSON(t *testing.T, data []byte) {
|
|
t.Helper()
|
|
if !json.Valid(data) {
|
|
t.Fatalf("invalid json output: %s", string(data))
|
|
}
|
|
}
|
|
|
|
func assertStringOrder(t *testing.T, body string, parts []string) {
|
|
t.Helper()
|
|
last := -1
|
|
for _, part := range parts {
|
|
idx := strings.Index(body, part)
|
|
if idx < 0 {
|
|
t.Fatalf("missing %q in output: %s", part, body)
|
|
}
|
|
if idx < last {
|
|
t.Fatalf("order mismatch for %q in output: %s", part, body)
|
|
}
|
|
last = idx
|
|
}
|
|
}
|