Skip to content

Commit 1cd32fa

Browse files
dnephindnephin
committed
Cleanup extra returns in tests
Signed-off-by: Daniel Nephin <dnephin@gmail.com>
1 parent cee56a1 commit 1cd32fa

File tree

6 files changed

+273
-546
lines changed

6 files changed

+273
-546
lines changed

‎client_test.go‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,6 @@ func TestImagePull(t *testing.T) {
184184
defer cancel()
185185
_, err = client.Pull(ctx, testImage)
186186
if err != nil {
187-
t.Error(err)
188-
return
187+
t.Fatal(err)
189188
}
190189
}

‎container_checkpoint_test.go‎

Lines changed: 33 additions & 66 deletions
Original file line numberDiff line numberDiff line change
@@ -27,66 +27,55 @@ func TestCheckpointRestore(t *testing.T) {
2727

2828
image, err := client.GetImage(ctx, testImage)
2929
if err != nil {
30-
t.Error(err)
31-
return
30+
t.Fatal(err)
3231
}
3332
container, err := client.NewContainer(ctx, id, WithNewSpec(oci.WithImageConfig(image), oci.WithProcessArgs("sleep", "100")), WithNewSnapshot(id, image))
3433
if err != nil {
35-
t.Error(err)
36-
return
34+
t.Fatal(err)
3735
}
3836
defer container.Delete(ctx, WithSnapshotCleanup)
3937

4038
task, err := container.NewTask(ctx, empty())
4139
if err != nil {
42-
t.Error(err)
43-
return
40+
t.Fatal(err)
4441
}
4542
defer task.Delete(ctx)
4643

4744
statusC, err := task.Wait(ctx)
4845
if err != nil {
49-
t.Error(err)
50-
return
46+
t.Fatal(err)
5147
}
5248

5349
if err := task.Start(ctx); err != nil {
54-
t.Error(err)
55-
return
50+
t.Fatal(err)
5651
}
5752

5853
checkpoint, err := task.Checkpoint(ctx, WithExit)
5954
if err != nil {
60-
t.Error(err)
61-
return
55+
t.Fatal(err)
6256
}
6357

6458
<-statusC
6559

6660
if _, err := task.Delete(ctx); err != nil {
67-
t.Error(err)
68-
return
61+
t.Fatal(err)
6962
}
7063
if task, err = container.NewTask(ctx, empty(), WithTaskCheckpoint(checkpoint)); err != nil {
71-
t.Error(err)
72-
return
64+
t.Fatal(err)
7365
}
7466
defer task.Delete(ctx)
7567

7668
statusC, err = task.Wait(ctx)
7769
if err != nil {
78-
t.Error(err)
79-
return
70+
t.Fatal(err)
8071
}
8172

8273
if err := task.Start(ctx); err != nil {
83-
t.Error(err)
84-
return
74+
t.Fatal(err)
8575
}
8676

8777
if err := task.Kill(ctx, syscall.SIGKILL); err != nil {
88-
t.Error(err)
89-
return
78+
t.Fatal(err)
9079
}
9180
<-statusC
9281
}
@@ -107,74 +96,61 @@ func TestCheckpointRestoreNewContainer(t *testing.T) {
10796

10897
image, err := client.GetImage(ctx, testImage)
10998
if err != nil {
110-
t.Error(err)
111-
return
99+
t.Fatal(err)
112100
}
113101
container, err := client.NewContainer(ctx, id, WithNewSpec(oci.WithImageConfig(image), oci.WithProcessArgs("sleep", "100")), WithNewSnapshot(id, image))
114102
if err != nil {
115-
t.Error(err)
116-
return
103+
t.Fatal(err)
117104
}
118105
defer container.Delete(ctx, WithSnapshotCleanup)
119106

120107
task, err := container.NewTask(ctx, empty())
121108
if err != nil {
122-
t.Error(err)
123-
return
109+
t.Fatal(err)
124110
}
125111
defer task.Delete(ctx)
126112

127113
statusC, err := task.Wait(ctx)
128114
if err != nil {
129-
t.Error(err)
130-
return
115+
t.Fatal(err)
131116
}
132117

133118
if err := task.Start(ctx); err != nil {
134-
t.Error(err)
135-
return
119+
t.Fatal(err)
136120
}
137121

138122
checkpoint, err := task.Checkpoint(ctx, WithExit)
139123
if err != nil {
140-
t.Error(err)
141-
return
124+
t.Fatal(err)
142125
}
143126

144127
<-statusC
145128

146129
if _, err := task.Delete(ctx); err != nil {
147-
t.Error(err)
148-
return
130+
t.Fatal(err)
149131
}
150132
if err := container.Delete(ctx, WithSnapshotCleanup); err != nil {
151-
t.Error(err)
152-
return
133+
t.Fatal(err)
153134
}
154135
if container, err = client.NewContainer(ctx, id, WithCheckpoint(checkpoint, id)); err != nil {
155-
t.Error(err)
156-
return
136+
t.Fatal(err)
157137
}
158138
if task, err = container.NewTask(ctx, empty(), WithTaskCheckpoint(checkpoint)); err != nil {
159-
t.Error(err)
160-
return
139+
t.Fatal(err)
161140
}
162141
defer task.Delete(ctx)
163142

164143
statusC, err = task.Wait(ctx)
165144
if err != nil {
166-
t.Error(err)
167-
return
145+
t.Fatal(err)
168146
}
169147

170148
if err := task.Start(ctx); err != nil {
171-
t.Error(err)
172-
return
149+
t.Fatal(err)
173150
}
174151

175152
if err := task.Kill(ctx, syscall.SIGKILL); err != nil {
176-
t.Error(err)
177-
return
153+
t.Fatal(err)
178154
}
179155
<-statusC
180156
}
@@ -200,52 +176,43 @@ func TestCheckpointLeaveRunning(t *testing.T) {
200176

201177
image, err := client.GetImage(ctx, testImage)
202178
if err != nil {
203-
t.Error(err)
204-
return
179+
t.Fatal(err)
205180
}
206181
container, err := client.NewContainer(ctx, id, WithNewSpec(oci.WithImageConfig(image), oci.WithProcessArgs("sleep", "100")), WithNewSnapshot(id, image))
207182
if err != nil {
208-
t.Error(err)
209-
return
183+
t.Fatal(err)
210184
}
211185
defer container.Delete(ctx, WithSnapshotCleanup)
212186

213187
task, err := container.NewTask(ctx, empty())
214188
if err != nil {
215-
t.Error(err)
216-
return
189+
t.Fatal(err)
217190
}
218191
defer task.Delete(ctx)
219192

220193
statusC, err := task.Wait(ctx)
221194
if err != nil {
222-
t.Error(err)
223-
return
195+
t.Fatal(err)
224196
}
225197

226198
if err := task.Start(ctx); err != nil {
227-
t.Error(err)
228-
return
199+
t.Fatal(err)
229200
}
230201

231202
if _, err := task.Checkpoint(ctx); err != nil {
232-
t.Error(err)
233-
return
203+
t.Fatal(err)
234204
}
235205

236206
status, err := task.Status(ctx)
237207
if err != nil {
238-
t.Error(err)
239-
return
208+
t.Fatal(err)
240209
}
241210
if status.Status != Running {
242-
t.Errorf("expected status %q but received %q", Running, status)
243-
return
211+
t.Fatalf("expected status %q but received %q", Running, status)
244212
}
245213

246214
if err := task.Kill(ctx, syscall.SIGKILL); err != nil {
247-
t.Error(err)
248-
return
215+
t.Fatal(err)
249216
}
250217

251218
<-statusC

0 commit comments

Comments
 (0)