integration: deflake TestIssue9103

fuweid · henry118 · commit 5dbc258a81c2 · 2023-11-08T23:23:31.000Z
Fixes: #9334 Signed-off-by: Wei Fu <fuweid89@gmail.com> (cherry picked from commit dac056f)
diff --git a/integration/client/container_linux_test.go b/integration/client/container_linux_test.go
@@ -1531,6 +1531,7 @@ func TestIssue9103(t *testing.T) {
 	for idx, tc := range []struct {
 		desc           string
 		cntrOpts       []NewContainerOpts
+		bakingFn       func(ctx context.Context, t *testing.T, task Task)
 		expectedStatus ProcessStatus
 	}{
 		{
@@ -1540,6 +1541,7 @@ func TestIssue9103(t *testing.T) {
 					withProcessArgs("sleep", "30"),
 				),
 			},
+			bakingFn:       func(context.Context, *testing.T, Task) {},
 			expectedStatus: Created,
 		},
 		{
@@ -1555,6 +1557,17 @@ func TestIssue9103(t *testing.T) {
 					BinaryName: "runc-fp",
 				}),
 			},
+			bakingFn: func(ctx context.Context, t *testing.T, task Task) {
+				waitCh, err := task.Wait(ctx)
+				require.NoError(t, err)
+
+				select {
+				case <-time.After(30 * time.Second):
+					t.Fatal("timeout")
+				case e := <-waitCh:
+					require.NoError(t, e.Error())
+				}
+			},
 			expectedStatus: Stopped,
 		},
 	} {
@@ -1574,6 +1587,8 @@ func TestIssue9103(t *testing.T) {
 
 			defer task.Delete(ctx, WithProcessKill)
 
+			tc.bakingFn(ctx, t, task)
+
 			status, err := task.Status(ctx)
 			require.NoError(t, err)
 			require.Equal(t, status.Status, tc.expectedStatus)
