初始化仓库

2026-06-02 23:14:41 +08:00
commit 0bc3f02670
520 changed files with 191097 additions and 0 deletions
@@ -0,0 +1,215 @@
+package codex
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+)
+
+func TestEnsureCodexProviderConfig_CreatesNewFile(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+
+	err := ensureCodexProviderConfig(home, "shengsuanyun",
+		"https://router.shengsuanyun.com/api/v1", "responses",
+		map[string]string{"HTTP-Referer": "https://openai.com/zh-Hans-CN/codex/", "X-Title": "CodeX"})
+	if err != nil {
+		t.Fatalf("ensureCodexProviderConfig: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "config.toml"))
+	if err != nil {
+		t.Fatalf("read config.toml: %v", err)
+	}
+	content := string(data)
+
+	for _, want := range []string{
+		`[model_providers.shengsuanyun]`,
+		`env_key = "OPENAI_API_KEY"`,
+		`wire_api = "responses"`,
+		`base_url = "https://router.shengsuanyun.com/api/v1"`,
+		`[model_providers.shengsuanyun.http_headers]`,
+		`"HTTP-Referer" = "https://openai.com/zh-Hans-CN/codex/"`,
+		`"X-Title" = "CodeX"`,
+	} {
+		if !strings.Contains(content, want) {
+			t.Errorf("config.toml missing %q\ngot:\n%s", want, content)
+		}
+	}
+}
+
+func TestEnsureCodexProviderConfig_UpdatesExistingSection(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+	if err := os.MkdirAll(home, 0o755); err != nil {
+		t.Fatal(err)
+	}
+
+	initial := `model = "gpt-5.4"
+
+[model_providers.shengsuanyun]
+name = "shengsuanyun"
+env_key = "OLD_KEY"
+wire_api = "chat"
+
+[some_other_section]
+key = "value"
+`
+	if err := os.WriteFile(filepath.Join(home, "config.toml"), []byte(initial), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	err := ensureCodexProviderConfig(home, "shengsuanyun",
+		"https://router.shengsuanyun.com/api/v1", "responses", nil)
+	if err != nil {
+		t.Fatalf("ensureCodexProviderConfig: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "config.toml"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	content := string(data)
+
+	if !strings.Contains(content, `env_key = "OPENAI_API_KEY"`) {
+		t.Errorf("updated config missing new env_key\ngot:\n%s", content)
+	}
+	if strings.Contains(content, `env_key = "OLD_KEY"`) {
+		t.Errorf("updated config still has old env_key\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `wire_api = "responses"`) {
+		t.Errorf("updated config missing new wire_api\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `[some_other_section]`) {
+		t.Errorf("updated config lost other section\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `model = "gpt-5.4"`) {
+		t.Errorf("updated config lost top-level key\ngot:\n%s", content)
+	}
+}
+
+func TestEnsureCodexProviderConfig_DefaultEnvKey(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+
+	err := ensureCodexProviderConfig(home, "dmxapi", "https://www.dmxapi.cn/v1", "responses", nil)
+	if err != nil {
+		t.Fatalf("ensureCodexProviderConfig: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "config.toml"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	content := string(data)
+
+	if !strings.Contains(content, `env_key = "OPENAI_API_KEY"`) {
+		t.Errorf("config should contain default env_key OPENAI_API_KEY\ngot:\n%s", content)
+	}
+	if strings.Contains(content, "requires_openai_auth") {
+		t.Errorf("config should NOT contain requires_openai_auth\ngot:\n%s", content)
+	}
+}
+
+func TestEnsureCodexProviderConfig_PreservesOtherProviders(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+	if err := os.MkdirAll(home, 0o755); err != nil {
+		t.Fatal(err)
+	}
+
+	initial := `[model_providers.other]
+name = "other"
+env_key = "OTHER_KEY"
+
+[model_providers.other.http_headers]
+"X-Custom" = "val"
+`
+	if err := os.WriteFile(filepath.Join(home, "config.toml"), []byte(initial), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	err := ensureCodexProviderConfig(home, "shengsuanyun", "", "responses", nil)
+	if err != nil {
+		t.Fatalf("ensureCodexProviderConfig: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "config.toml"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	content := string(data)
+
+	if !strings.Contains(content, `[model_providers.other]`) {
+		t.Errorf("lost other provider section\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `[model_providers.shengsuanyun]`) {
+		t.Errorf("new provider not added\ngot:\n%s", content)
+	}
+}
+
+func TestEnsureCodexProviderConfig_SkipsWhenEmpty(t *testing.T) {
+	err := ensureCodexProviderConfig("", "", "", "", nil)
+	if err != nil {
+		t.Fatalf("unexpected error for empty name: %v", err)
+	}
+}
+
+func TestEnsureCodexAuth_WritesAuthJSON(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+
+	err := ensureCodexAuth(home, "sk-test-key-123")
+	if err != nil {
+		t.Fatalf("ensureCodexAuth: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "auth.json"))
+	if err != nil {
+		t.Fatalf("read auth.json: %v", err)
+	}
+	content := string(data)
+
+	for _, want := range []string{
+		`"OPENAI_API_KEY": "sk-test-key-123"`,
+		`"auth_mode": "apikey"`,
+	} {
+		if !strings.Contains(content, want) {
+			t.Errorf("auth.json missing %q\ngot:\n%s", want, content)
+		}
+	}
+}
+
+func TestEnsureCodexAuth_SkipsEmptyKey(t *testing.T) {
+	err := ensureCodexAuth(t.TempDir(), "")
+	if err != nil {
+		t.Fatalf("unexpected error for empty key: %v", err)
+	}
+}
+
+func TestEnsureCodexAuth_OverwritesExisting(t *testing.T) {
+	home := filepath.Join(t.TempDir(), ".codex")
+	if err := os.MkdirAll(home, 0o755); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(home, "auth.json"), []byte(`{"auth_mode":"chatgpt"}`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	err := ensureCodexAuth(home, "new-api-key")
+	if err != nil {
+		t.Fatalf("ensureCodexAuth: %v", err)
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, "auth.json"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	content := string(data)
+
+	if strings.Contains(content, "chatgpt") {
+		t.Errorf("auth.json still has old auth_mode\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `"auth_mode": "apikey"`) {
+		t.Errorf("auth.json missing apikey mode\ngot:\n%s", content)
+	}
+	if !strings.Contains(content, `"OPENAI_API_KEY": "new-api-key"`) {
+		t.Errorf("auth.json missing new key\ngot:\n%s", content)
+	}
+}