38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
//go:build linux
|
|
|
|
package service
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestRenderUnit(t *testing.T) {
|
|
unit := renderUnit("/var/lib/gpu-turnstile/gpu-turnstile", "/etc/gpu-turnstile.env", "")
|
|
for _, want := range []string{
|
|
"Type=notify",
|
|
"WatchdogSec=30s",
|
|
`ExecStart="/var/lib/gpu-turnstile/gpu-turnstile" -config "/etc/gpu-turnstile.env"`,
|
|
"Restart=on-failure",
|
|
"WantedBy=multi-user.target",
|
|
"DynamicUser=yes",
|
|
"StateDirectory=gpu-turnstile",
|
|
"ProtectSystem=strict",
|
|
"NoNewPrivileges=yes",
|
|
"RestrictAddressFamilies=AF_UNIX AF_INET AF_INET6",
|
|
"SystemCallFilter=@system-service",
|
|
} {
|
|
if !strings.Contains(unit, want) {
|
|
t.Fatalf("unit missing %q:\n%s", want, unit)
|
|
}
|
|
}
|
|
if strings.Contains(unit, "BindPaths") {
|
|
t.Fatalf("unit without COMFY_DIR must not bind anything:\n%s", unit)
|
|
}
|
|
|
|
unit = renderUnit("/var/lib/gpu-turnstile/gpu-turnstile", "/etc/gpu-turnstile.env", "/home/gpu/ComfyUI")
|
|
if !strings.Contains(unit, "BindPaths=/home/gpu/ComfyUI\n") {
|
|
t.Fatalf("unit with COMFY_DIR must bind it:\n%s", unit)
|
|
}
|
|
}
|