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
|
package e2e
import (
"fmt"
"os"
"os/exec"
"path/filepath"
"testing"
"time"
)
func TestNamespaceLifecycleAutomation(t *testing.T) {
// 1. Setup Environment
binaryPath, err := GetBinaryPath()
if err != nil {
t.Skipf("Skipping test: %v", err)
}
// 2. Override the runtime base dir to a temporary location
tmpRuntimeDir := t.TempDir()
profile := "e2e-lifecycle-test"
pidsDir := filepath.Join(tmpRuntimeDir, "profiles", profile, "pids")
// Clean up before starting
os.RemoveAll(filepath.Join(tmpRuntimeDir, "profiles", profile))
t.Run("ReferenceCounting", func(t *testing.T) {
// Start a process that exits quickly
cmd1 := exec.Command(binaryPath, "--profile", profile, "--", "sleep", "0.1")
cmd1.Env = append(os.Environ(), fmt.Sprintf("XDG_RUNTIME_DIR=%s", tmpRuntimeDir))
if err := cmd1.Start(); err != nil {
t.Fatalf("Failed to start cmd1: %v", err)
}
// Allow a moment for the bootstrap loop to complete and register the PID
time.Sleep(500 * time.Millisecond)
// Verify PID file exists
files, err := os.ReadDir(pidsDir)
if err != nil {
t.Fatalf("Failed to read pids dir: %v", err)
}
if len(files) != 1 {
t.Errorf("Expected 1 PID file, got %d", len(files))
}
// Start a second process using the same profile
cmd2 := exec.Command(binaryPath, "--profile", profile, "--", "sleep", "0.1")
cmd2.Env = append(os.Environ(), fmt.Sprintf("XDG_RUNTIME_DIR=%s", tmpRuntimeDir))
if err := cmd2.Start(); err != nil {
t.Fatalf("Failed to start cmd2: %v", err)
}
time.Sleep(500 * time.Millisecond)
files, err = os.ReadDir(pidsDir)
if err != nil {
t.Fatalf("Failed to read pids dir: %v", err)
}
if len(files) != 2 {
t.Errorf("Expected 2 PID files, got %d", len(files))
}
// Wait for first process to exit naturally (triggering defer)
if err := cmd1.Wait(); err != nil {
t.Fatalf("cmd1 failed: %v", err)
}
time.Sleep(500 * time.Millisecond)
files, err = os.ReadDir(pidsDir)
if err != nil {
t.Fatalf("Failed to read pids dir: %v", err)
}
if len(files) != 1 {
t.Errorf("Expected 1 PID file after first exit, got %d", len(files))
}
// Wait for second process to exit naturally
if err := cmd2.Wait(); err != nil {
t.Fatalf("cmd2 failed: %v", err)
}
time.Sleep(500 * time.Millisecond)
// Verify a clean state
files, err = os.ReadDir(pidsDir)
if err != nil && !os.IsNotExist(err) {
t.Fatalf("Failed to read pids dir: %v", err)
}
if err == nil && len(files) != 0 {
t.Errorf("Expected 0 PID files after all exits, got %d", len(files))
}
})
}
|
