初始化仓库

2026-06-02 23:14:41 +08:00
commit 0bc3f02670
520 changed files with 191097 additions and 0 deletions
@@ -0,0 +1,125 @@
+//go:build windows
+
+package daemon
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestStrictPowerShellStopsOnCmdletErrors(t *testing.T) {
+	script := strictPowerShell("Write-Output 'ok'")
+	if !strings.HasPrefix(script, "$ErrorActionPreference = 'Stop'\n") {
+		t.Fatalf("strictPowerShell() missing stop prelude:\n%s", script)
+	}
+	if !strings.Contains(script, "Write-Output 'ok'") {
+		t.Fatalf("strictPowerShell() missing original script:\n%s", script)
+	}
+}
+
+func TestBuildWindowsTaskScript(t *testing.T) {
+	cfg := Config{
+		BinaryPath: `C:\Program Files\cc-connect\cc-connect.exe`,
+		WorkDir:    `C:\Users\me\.cc-connect`,
+		LogFile:    `C:\Users\me\.cc-connect\logs\cc-connect.log`,
+		LogMaxSize: 10 * 1024 * 1024,
+		EnvPATH:    `C:\Program Files\nodejs;C:\Users\me\AppData\Local\Programs`,
+		EnvExtra: map[string]string{
+			"HTTPS_PROXY": "http://127.0.0.1:7890",
+			"http_proxy":  "http://127.0.0.1:7890",
+		},
+	}
+
+	script := buildWindowsTaskScript(cfg)
+	for _, want := range []string{
+		`$env:CC_LOG_FILE = 'C:\Users\me\.cc-connect\logs\cc-connect.log'`,
+		`$env:CC_LOG_MAX_SIZE = '10485760'`,
+		`$env:PATH = 'C:\Program Files\nodejs;C:\Users\me\AppData\Local\Programs'`,
+		`$env:HTTPS_PROXY = 'http://127.0.0.1:7890'`,
+		`$env:http_proxy = 'http://127.0.0.1:7890'`,
+		`Set-Location -LiteralPath 'C:\Users\me\.cc-connect'`,
+		`while ($true) {`,
+		`& 'C:\Program Files\cc-connect\cc-connect.exe'`,
+		`if ($exitCode -eq 0) { exit 0 }`,
+		`Start-Sleep -Seconds 10`,
+	} {
+		if !strings.Contains(script, want) {
+			t.Fatalf("script missing %q:\n%s", want, script)
+		}
+	}
+}
+
+func TestWindowsTaskActionRunsHidden(t *testing.T) {
+	got := windowsTaskAction(`C:\Users\me\.cc-connect\cc-connect-daemon.ps1`)
+	for _, want := range []string{
+		`powershell.exe`,
+		`-WindowStyle Hidden`,
+		`-NoProfile`,
+		`-NonInteractive`,
+		`-ExecutionPolicy Bypass`,
+		`-File "C:\Users\me\.cc-connect\cc-connect-daemon.ps1"`,
+	} {
+		if !strings.Contains(got, want) {
+			t.Fatalf("windowsTaskAction() missing %q: %q", want, got)
+		}
+	}
+}
+
+func TestWindowsTaskCreateUsesLimitedInteractivePrincipal(t *testing.T) {
+	orig := runPowerShell
+	t.Cleanup(func() { runPowerShell = orig })
+
+	var script string
+	runPowerShell = func(s string) (string, error) {
+		script = s
+		return "", nil
+	}
+
+	if err := createWindowsTask(`C:\Users\me\.cc-connect\cc-connect-daemon.ps1`); err != nil {
+		t.Fatalf("createWindowsTask() error = %v", err)
+	}
+	for _, want := range []string{
+		`New-ScheduledTaskAction`,
+		`Register-ScheduledTask`,
+		`-LogonType Interactive`,
+		`-RunLevel Limited`,
+		`-WindowStyle Hidden`,
+		`C:\Users\me\.cc-connect\cc-connect-daemon.ps1`,
+	} {
+		if !strings.Contains(script, want) {
+			t.Fatalf("create script missing %q:\n%s", want, script)
+		}
+	}
+}
+
+func TestWindowsTaskMatchesActionRequiresExactAction(t *testing.T) {
+	orig := runPowerShell
+	t.Cleanup(func() { runPowerShell = orig })
+
+	var script string
+	runPowerShell = func(s string) (string, error) {
+		script = s
+		return "true", nil
+	}
+
+	if !windowsTaskMatchesAction(`C:\Users\me\.cc-connect\cc-connect-daemon.ps1`) {
+		t.Fatal("windowsTaskMatchesAction() = false, want true")
+	}
+	for _, want := range []string{
+		`$expectedArgs = '-WindowStyle Hidden -NoProfile -NonInteractive -ExecutionPolicy Bypass -File "C:\Users\me\.cc-connect\cc-connect-daemon.ps1"'`,
+		`$action.Execute -ieq 'powershell.exe'`,
+		`$action.Arguments -eq $expectedArgs`,
+	} {
+		if !strings.Contains(script, want) {
+			t.Fatalf("reuse check script missing %q:\n%s", want, script)
+		}
+	}
+}
+
+func TestPowerShellLiteralEscapesSingleQuotes(t *testing.T) {
+	got := powerShellLiteral(`C:\Users\O'Brien\.cc-connect`)
+	want := `'C:\Users\O''Brien\.cc-connect'`
+	if got != want {
+		t.Fatalf("powerShellLiteral() = %q, want %q", got, want)
+	}
+}