Carlos Florêncio
121 lines
3.6 KiB
Go
121 lines
3.6 KiB
Go
package launcher
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBuildManifestUsesSameBinaryBackendMode(t *testing.T) {
|
|
env := []string{
|
|
"KANDEV_SERVER_PORT=1234",
|
|
"KANDEV_DESKTOP_HEALTH_TOKEN=health-token",
|
|
"KANDEV_CONSOLE_LOG_LEVEL=warn",
|
|
"KANDEV_WEB_TITLE_PREFIX=Debug",
|
|
"UNRELATED=value",
|
|
}
|
|
manifest := buildManifest("/opt/kandev/bin/kandev", []string{"__backend"}, "/opt/kandev/bin", env, "/tmp/home", 1234, "run")
|
|
|
|
if manifest.BackendExecutable != "/opt/kandev/bin/kandev" {
|
|
t.Fatalf("BackendExecutable = %q", manifest.BackendExecutable)
|
|
}
|
|
if len(manifest.Argv) != 1 || manifest.Argv[0] != "__backend" {
|
|
t.Fatalf("Argv = %v, want [__backend]", manifest.Argv)
|
|
}
|
|
if _, ok := manifest.Env["UNRELATED"]; ok {
|
|
t.Fatalf("manifest contains unrelated env: %+v", manifest.Env)
|
|
}
|
|
if manifest.Env["KANDEV_SERVER_PORT"] != "1234" {
|
|
t.Fatalf("KANDEV_SERVER_PORT = %q", manifest.Env["KANDEV_SERVER_PORT"])
|
|
}
|
|
if manifest.Env["KANDEV_CONSOLE_LOG_LEVEL"] != "warn" {
|
|
t.Fatalf("KANDEV_CONSOLE_LOG_LEVEL = %q", manifest.Env["KANDEV_CONSOLE_LOG_LEVEL"])
|
|
}
|
|
if manifest.Env["KANDEV_DESKTOP_HEALTH_TOKEN"] != "health-token" {
|
|
t.Fatalf("KANDEV_DESKTOP_HEALTH_TOKEN = %q", manifest.Env["KANDEV_DESKTOP_HEALTH_TOKEN"])
|
|
}
|
|
if manifest.Env["KANDEV_WEB_TITLE_PREFIX"] != "Debug" {
|
|
t.Fatalf("KANDEV_WEB_TITLE_PREFIX = %q", manifest.Env["KANDEV_WEB_TITLE_PREFIX"])
|
|
}
|
|
}
|
|
|
|
func TestPrepareSupervisorEnvWritesExpectedPaths(t *testing.T) {
|
|
home := t.TempDir()
|
|
env, socket, manifest, err := prepareSupervisorEnv(nil, home)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if socket != filepath.Join(home, "supervisor", "control.sock") {
|
|
t.Fatalf("socket = %q", socket)
|
|
}
|
|
if manifest != filepath.Join(home, "supervisor", "launch.json") {
|
|
t.Fatalf("manifest = %q", manifest)
|
|
}
|
|
got := allowedSupervisorEnv(env)
|
|
if got["KANDEV_RESTART_ADAPTER"] != "supervisor" {
|
|
t.Fatalf("restart adapter env = %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestListenControlSocketUsesOwnerOnlyMode(t *testing.T) {
|
|
const unixSocketTempRoot = "/tmp"
|
|
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("unix sockets are not supported on windows")
|
|
}
|
|
shortRoot, err := os.MkdirTemp(unixSocketTempRoot, "kandev-inherited-tmp-*")
|
|
if err != nil {
|
|
t.Fatalf("create temporary directory under /tmp: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = os.RemoveAll(shortRoot) })
|
|
longTMPDIR := filepath.Join(shortRoot, "this-inherited-tmpdir-is-deliberately-long-to-exceed-the-unix-socket-path-limit")
|
|
if err := os.Mkdir(longTMPDIR, 0o700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("TMPDIR", longTMPDIR)
|
|
|
|
dir, err := os.MkdirTemp(unixSocketTempRoot, "kandev-sock-*")
|
|
if err != nil {
|
|
t.Fatalf("create socket directory under /tmp: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = os.RemoveAll(dir) })
|
|
if filepath.Dir(dir) != unixSocketTempRoot {
|
|
t.Fatalf("socket directory = %q, want it directly under %s", dir, unixSocketTempRoot)
|
|
}
|
|
socket := filepath.Join(dir, "control.sock")
|
|
ln, err := listenControlSocket(socket)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = ln.Close() })
|
|
|
|
info, err := os.Stat(socket)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := info.Mode().Perm(); got != 0o600 {
|
|
t.Fatalf("socket mode = %#o, want 0600", got)
|
|
}
|
|
}
|
|
|
|
func TestRestartFailureNotifiesLauncherExit(t *testing.T) {
|
|
backend := &restartableBackend{
|
|
command: filepath.Join(t.TempDir(), "missing-kandev"),
|
|
supervisor: newSupervisor(),
|
|
exitCh: make(chan int, 1),
|
|
}
|
|
|
|
backend.restart()
|
|
|
|
select {
|
|
case code := <-backend.exitCh:
|
|
if code == 0 {
|
|
t.Fatal("restart failure reported successful exit")
|
|
}
|
|
case <-time.After(500 * time.Millisecond):
|
|
t.Fatal("restart failure did not notify launcher exit")
|
|
}
|
|
}
|