1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
package e2e
import (
"os"
"os/exec"
"path/filepath"
"testing"
"time"
)
func waitForPids(t *testing.T, pidsDir string, expectedCount int) {
timeout := time.After(10 * time.Second)
tick := time.NewTicker(100 * time.Millisecond)
defer tick.Stop()
for {
select {
case <-timeout:
files, err := os.ReadDir(pidsDir)
if err != nil {
t.Fatalf("Failed to read pids dir during timeout: %v", err)
}
t.Fatalf("Timed out waiting for %d PID files, got %d", expectedCount, len(files))
case <-tick.C:
files, err := os.ReadDir(pidsDir)
if err != nil {
if os.IsNotExist(err) {
continue
}
t.Fatalf("Failed to read pids dir: %v", err)
}
if len(files) == expectedCount {
return
}
}
}
}
func waitForLifecycle(t *testing.T, binaryPath, runtimeDir, profile string, expectedActive bool) {
timeout := time.After(10 * time.Second)
tick := time.NewTicker(100 * time.Millisecond)
defer tick.Stop()
for {
select {
case <-timeout:
t.Fatalf("Timed out waiting for lifecycle state: expected active=%v", expectedActive)
case <-tick.C:
cmd := exec.Command(binaryPath, "test-lifecycle", "--profile", profile)
cmd.Env = SetEnvOverrides(map[string]string{"XDG_RUNTIME_DIR": runtimeDir})
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
err := cmd.Run()
isActive := err == nil
if isActive == expectedActive {
return
}
}
}
}
func TestNamespaceLifecycleAutomation(t *testing.T) {
// 1. Setup Environment
binaryPath := EnsureBinary(t)
// 2. Override the runtime base dir to a temporary location
tmpRuntimeDir := t.TempDir()
profile := "default"
pidsDir := filepath.Join(tmpRuntimeDir, "profiles", profile, "pids")
// Clean up before starting
if err := os.RemoveAll(filepath.Join(tmpRuntimeDir, "profiles", profile)); err != nil {
t.Fatalf("failed to remove profile directory: %v", err)
}
t.Run("ReferenceCounting", func(t *testing.T) {
// Start a process that exits quickly
cmd1 := exec.Command(binaryPath, "--profile", "default", "--", "sleep", "1.0")
cmd1.Env = SetEnvOverrides(map[string]string{"XDG_RUNTIME_DIR": tmpRuntimeDir})
cmd1.Stdout = os.Stdout
cmd1.Stderr = os.Stderr
if err := cmd1.Start(); err != nil {
t.Fatalf("Failed to start cmd1: %v", err)
}
// Verify PID file exists using polling
waitForLifecycle(t, binaryPath, tmpRuntimeDir, "default", true)
// Start a second process using the same profile with a longer sleep
cmd2 := exec.Command(binaryPath, "--profile", "default", "--", "sleep", "5.0")
cmd2.Env = SetEnvOverrides(map[string]string{"XDG_RUNTIME_DIR": tmpRuntimeDir})
cmd2.Stdout = os.Stdout
cmd2.Stderr = os.Stderr
if err := cmd2.Start(); err != nil {
t.Fatalf("Failed to start cmd2: %v", err)
}
waitForPids(t, pidsDir, 2)
// Wait for first process to exit naturally (triggering defer)
if err := cmd1.Wait(); err != nil {
t.Fatalf("cmd1 failed: %v", err)
}
// Poll for the count to drop back to 1 (cmd2 should still be running)
waitForLifecycle(t, binaryPath, tmpRuntimeDir, "default", true)
// Wait for second process to exit naturally
if err := cmd2.Wait(); err != nil {
t.Fatalf("cmd2 failed: %v", err)
}
// Verify a clean state (expect 0 files)
waitForLifecycle(t, binaryPath, tmpRuntimeDir, "default", false)
})
}
|
