chore: import upstream snapshot with attribution

cmd/gortex/proxy_cmd_test.go

@@ -0,0 +1,97 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/zzet/gortex/internal/daemon"
+)
+
+// hermeticProxyEnv points the roster + daemon socket at a temp dir so
+// the toggle commands neither touch the user's real roster nor dial the
+// live development daemon.
+func hermeticProxyEnv(t *testing.T) string {
+	t.Helper()
+	dir := t.TempDir()
+	path := filepath.Join(dir, "servers.toml")
+	t.Setenv("GORTEX_DAEMON_SERVERS", path)
+	// A socket path that cannot exist => IsRunning() is false, so the
+	// best-effort live-reload dial is skipped.
+	t.Setenv("GORTEX_DAEMON_SOCKET", filepath.Join(dir, "nonexistent.sock"))
+	return path
+}
+
+// TestProxyToggle_OffThenOnPersists asserts `proxy off` writes
+// enabled=false (surviving a reload) and `proxy on` clears the key (back
+// to default-on).
+func TestProxyToggle_OffThenOnPersists(t *testing.T) {
+	path := hermeticProxyEnv(t)
+	roster := "[[server]]\nslug = \"r2\"\nurl = \"https://r2.example:4747\"\n"
+	if err := os.WriteFile(path, []byte(roster), 0o600); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := runProxyToggle("r2", false); err != nil {
+		t.Fatalf("proxy off: %v", err)
+	}
+	data, _ := os.ReadFile(path)
+	if !strings.Contains(string(data), "enabled = false") {
+		t.Fatalf("proxy off should write enabled = false, got:\n%s", data)
+	}
+	cfg, err := daemon.LoadServersConfig(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if cfg.Server[0].IsEnabled() {
+		t.Fatal("a disabled remote must survive a reload as disabled")
+	}
+
+	if err := runProxyToggle("r2", true); err != nil {
+		t.Fatalf("proxy on: %v", err)
+	}
+	data, _ = os.ReadFile(path)
+	if strings.Contains(string(data), "enabled") {
+		t.Fatalf("proxy on should clear the enabled key, got:\n%s", data)
+	}
+	cfg, _ = daemon.LoadServersConfig(path)
+	if !cfg.Server[0].IsEnabled() {
+		t.Fatal("re-enabled remote must reload as enabled")
+	}
+}
+
+// TestProxyToggle_UnknownSlug asserts toggling a remote not in the roster
+// is a clear error, not a silent no-op.
+func TestProxyToggle_UnknownSlug(t *testing.T) {
+	path := hermeticProxyEnv(t)
+	if err := os.WriteFile(path, []byte("[[server]]\nslug = \"r2\"\nurl = \"https://r2.example:4747\"\n"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	if err := runProxyToggle("nope", false); err == nil {
+		t.Fatal("toggling an unknown slug must error")
+	}
+}
+
+// TestProxyAdd_PositionalURL asserts `proxy add <slug> <url>` persists the
+// remote with the positional URL.
+func TestProxyAdd_PositionalURL(t *testing.T) {
+	path := hermeticProxyEnv(t)
+	daemonServerAddDefault = false
+	daemonServerAddAuthToken = ""
+	daemonServerAddAuthTokenEnv = ""
+	daemonServerAddWorkspaces = nil
+	daemonServerAddReadOnly = false
+	t.Cleanup(func() { daemonServerAddURL = "" })
+
+	if err := runProxyAdd(nil, []string{"r3", "https://r3.example:4747"}); err != nil {
+		t.Fatalf("proxy add: %v", err)
+	}
+	cfg, err := daemon.LoadServersConfig(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(cfg.Server) != 1 || cfg.Server[0].Slug != "r3" || cfg.Server[0].URL != "https://r3.example:4747" {
+		t.Fatalf("proxy add did not persist the remote: %+v", cfg.Server)
+	}
+}