wehub-resource-sync
a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
98 lines
3.0 KiB
Go
98 lines
3.0 KiB
Go
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)
|
|
}
|
|
}
|