Function TestWaitRestartedContainer

integration/container/wait_test.go:182–247

(t *testing.T)

}

func TestWaitRestartedContainer(t *testing.T) {
	ctx := setupTest(t)
	cli := request.NewAPIClient(t)

	tests := []struct {
		doc      string
		waitCond containertypes.WaitCondition
	}{
		{
			doc: "default",
		},
		{
			doc:      "not-running",
			waitCond: containertypes.WaitConditionNotRunning,
		},
		{
			doc:      "next-exit",
			waitCond: containertypes.WaitConditionNextExit,
		},
	}

	// We can't catch the SIGTERM in the Windows based busybox image
	isWindowDaemon := testEnv.DaemonInfo.OSType == "windows"

	for _, tc := range tests {
		t.Run(tc.doc, func(t *testing.T) {
			// TODO(vvoland): Verify why this helps for flakiness
			// t.Parallel()
			ctx := testutil.StartSpan(ctx, t)
			containerID := container.Run(ctx, t, cli,
				container.WithCmd("sh", "-c", "trap 'exit 5' SIGTERM; echo ready; while true; do sleep 0.1; done"),
			)
			defer cli.ContainerRemove(ctx, containerID, client.ContainerRemoveOptions{Force: true})
			poll.WaitOn(t, logsContains(ctx, cli, containerID, "ready"))

			// Container is running now, wait for exit
			wait := cli.ContainerWait(ctx, containerID, client.ContainerWaitOptions{Condition: tc.waitCond})

			timeout := 10
			// On Windows it will always timeout, because our process won't receive SIGTERM
			// Skip to force killing immediately
			if isWindowDaemon {
				timeout = 0
			}

			_, err := cli.ContainerRestart(ctx, containerID, client.ContainerRestartOptions{
				Timeout: &timeout,
				Signal:  "SIGTERM",
			})
			assert.NilError(t, err)

			select {
			case err := <-wait.Error:
				t.Fatalf("Unexpected error: %v", err)
			case <-time.After(time.Second * 3):
				t.Fatalf("Wait should end after restart")
			case waitRes := <-wait.Result:
				expectedCode := int64(5)