carlos / platform Public

Download this file

1	package host
2
3	import (
4	"context"
5	"os"
6	"path/filepath"
7	"testing"
8	)
9
10	func TestPlan14ParticipationFlagOffUpdate(t *testing.T) {
11	t.Setenv("CARLOS_HOST_CLAIM", "")
12	for _, region := range []bool{false, true} {
13	for _, mirror := range []string{"absent", "empty", "pending", "symlink"} {
14	t.Run(map[bool]string{false: "region-off", true: "region-on"}[region]+"/"+mirror, func(t *testing.T) {
15	f, _ := applyHappy(t)
16	f.a.regionHA = region
17	if err := f.a.participation.State.Remove("census.json"); err != nil {
18	t.Fatal(err)
19	}
20	f.a.participation.VerifyLaunches = nil
21	switch mirror {
22	case "empty":
23	hostexec(t, f)
24	case "pending":
25	if err := os.WriteFile(filepath.Join(hostexec(t, f), "pending.json"), []byte(`{}`), 0600); err != nil {
26	t.Fatal(err)
27	}
28	case "symlink":
29	if err := os.Symlink(t.TempDir(), filepath.Join(f.a.dataDir, hostexecDir)); err != nil {
30	t.Fatal(err)
31	}
32	}
33	if err := f.a.Run(t.Context(), false); err != nil {
34	t.Fatal(err)
35	}
36	want := systemOutcomeApplied
37	if region && (mirror == "pending" || mirror == "symlink") {
38	want = systemOutcomeDeferred
39	}
40	if got := f.result(); got == nil || got.Outcome != want {
41	t.Fatalf("result=%+v want=%s", got, want)
42	}
43	if _, err := f.a.participation.UpdateID(); !os.IsNotExist(err) {
44	t.Fatalf("legacy update created root reservation: %v", err)
45	}
46	})
47	}
48	}
49	}
50
51	func TestPlan14ParticipationFlagOffRecovery(t *testing.T) {
52	t.Setenv("CARLOS_HOST_CLAIM", "")
53	for _, journal := range []bool{false, true} {
54	t.Run(map[bool]string{false: "marker", true: "journal"}[journal], func(t *testing.T) {
55	f := newApplyFixture(t, "stable", "v1")
56	if err := f.a.participation.State.Remove("census.json"); err != nil {
57	t.Fatal(err)
58	}
59	f.a.participation.VerifyLaunches = nil
60	if err := os.MkdirAll(systemGuardDir(f.a.optDir), 0755); err != nil {
61	t.Fatal(err)
62	}
63	if err := os.WriteFile(updateBegunPath(f.a.optDir), []byte(`{"target":"v2"}`), 0644); err != nil {
64	t.Fatal(err)
65	}
66	if journal {
67	if err := f.a.journal(systemJournal{Phase: systemPhaseBackingUp, Target: "v2", PrevBuild: "v1", StartedAt: f.now}); err != nil {
68	t.Fatal(err)
69	}
70	}
71	old := recoverPendingControllers
72	t.Cleanup(func() { recoverPendingControllers = old })
73	recoverPendingControllers = func(context.Context, string, string) error {
74	t.Fatal("legacy recovery called root participation recovery")
75	return nil
76	}
77	if err := f.a.Run(t.Context(), true); err != nil {
78	t.Fatal(err)
79	}
80	if _, err := os.Stat(updateBegunPath(f.a.optDir)); !os.IsNotExist(err) {
81	t.Fatalf("legacy marker retained: %v", err)
82	}
83	if j, err := readSystemJournal(f.a.optDir); err != nil || j != nil {
84	t.Fatalf("journal=%+v err=%v", j, err)
85	}
86	})
87	}
88	}
89
90	func TestPlan14ParticipationFlagAndCensusExclusion(t *testing.T) {
91	for _, flag := range []string{"", "1"} {
92	for _, census := range []bool{false, true} {
93	t.Run(flag+map[bool]string{false: "/missing", true: "/present"}[census], func(t *testing.T) {
94	t.Setenv("CARLOS_HOST_CLAIM", flag)
95	f := newApplyFixture(t, "stable", "old")
96	if !census {
97	if err := f.a.participation.State.Remove("census.json"); err != nil {
98	t.Fatal(err)
99	}
100	}
101	f.a.participation.VerifyLaunches = nil
102	err := f.a.beginExecutableChange(t.Context(), "new", f.now)
103	if (err != nil) != (flag != "" || census) {
104	t.Fatalf("admission error=%v flag=%q census=%t", err, flag, census)
105	}
106	})
107	}
108	}
109	}
110
111	func TestPlan14ParticipationFlagOffControllersRecovered(t *testing.T) {
112	t.Setenv("CARLOS_HOST_CLAIM", "")
113	f := newApplyFixture(t, "stable", "v1")
114	if err := f.a.participation.State.Remove("census.json"); err != nil {
115	t.Fatal(err)
116	}
117	old := recoverPendingControllers
118	t.Cleanup(func() { recoverPendingControllers = old })
119	recoverPendingControllers = func(context.Context, string, string) error {
120	t.Fatal("legacy controller invoked root recovery")
121	return nil
122	}
123	dir := hostexec(t, f)
124	if err := f.a.controllersRecovered(t.Context()); err != nil {
125	t.Fatal(err)
126	}
127	if err := os.WriteFile(filepath.Join(dir, "pending.json"), []byte(`{}`), 0600); err != nil {
128	t.Fatal(err)
129	}
130	if err := f.a.controllersRecovered(t.Context()); err == nil {
131	t.Fatal("malformed legacy controller admitted")
132	}
133	}
134
135	func TestPlan14ParticipationFlagOffGuardRecovery(t *testing.T) {
136	t.Setenv("CARLOS_HOST_CLAIM", "")
137	f, id := sharedGuardFixture(t)
138	a := f.a
139	if err := a.participation.State.Remove("census.json"); err != nil {
140	t.Fatal(err)
141	}
142	a.participation.VerifyLaunches = nil
143	if err := a.floor.Begin(t.Context(), id, a.dataDir); err != nil {
144	t.Fatal(err)
145	}
146	if err := a.beginExecutableChange(t.Context(), "new-build", f.now); err != nil {
147	t.Fatal(err)
148	}
149	if err := a.journal(systemJournal{Phase: systemPhaseBackingUp, Target: "new-build", PrevBuild: "old-build", StartedAt: f.now}); err != nil {
150	t.Fatal(err)
151	}
152	if err := a.Run(t.Context(), true); err != nil {
153	t.Fatal(err)
154	}
155	completed, err := os.ReadFile(filepath.Join(a.floor.Dir, "completed.json"))
156	if err != nil || string(completed) != id {
157	t.Fatalf("completion=%q err=%v", completed, err)
158	}
159	if _, err := os.Stat(updateBegunPath(a.optDir)); !os.IsNotExist(err) {
160	t.Fatalf("legacy floor marker retained: %v", err)
161	}
162	if _, err := a.participation.UpdateID(); !os.IsNotExist(err) {
163	t.Fatalf("legacy recovery created root reservation: %v", err)
164	}
165	}
166