126 lines
3.7 KiB
Go
126 lines
3.7 KiB
Go
//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)
|
|
}
|
|
}
|