diff --git a/integration-cli/docker_cli_wait_test.go b/integration-cli/docker_cli_wait_test.go new file mode 100644 index 0000000000..aece88e33e --- /dev/null +++ b/integration-cli/docker_cli_wait_test.go @@ -0,0 +1,121 @@ +package main + +import ( + "os/exec" + "testing" + "time" +) + +// non-blocking wait with 0 exit code +func TestWaitNonBlockedExitZero(t *testing.T) { + defer deleteAllContainers() + + runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "true") + out, _, err := runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(out, err) + } + containerID := stripTrailingCharacters(out) + + status := "true" + for i := 0; status != "false"; i++ { + runCmd = exec.Command(dockerBinary, "inspect", "--format='{{.State.Running}}'", containerID) + status, _, err = runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(status, err) + } + status = stripTrailingCharacters(status) + + time.Sleep(time.Second) + if i >= 60 { + t.Fatal("Container should have stopped by now") + } + } + + runCmd = exec.Command(dockerBinary, "wait", containerID) + out, _, err = runCommandWithOutput(runCmd) + + if err != nil || stripTrailingCharacters(out) != "0" { + t.Fatal("failed to set up container", out, err) + } + + logDone("wait - non-blocking wait with 0 exit code") +} + +// blocking wait with 0 exit code +func TestWaitBlockedExitZero(t *testing.T) { + defer deleteAllContainers() + + runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "sleep 10") + out, _, err := runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(out, err) + } + containerID := stripTrailingCharacters(out) + + runCmd = exec.Command(dockerBinary, "wait", containerID) + out, _, err = runCommandWithOutput(runCmd) + + if err != nil || stripTrailingCharacters(out) != "0" { + t.Fatal("failed to set up container", out, err) + } + + logDone("wait - blocking wait with 0 exit code") +} + +// non-blocking wait with random exit code +func TestWaitNonBlockedExitRandom(t *testing.T) { + defer deleteAllContainers() + + runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "exit 99") + out, _, err := runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(out, err) + } + containerID := stripTrailingCharacters(out) + + status := "true" + for i := 0; status != "false"; i++ { + runCmd = exec.Command(dockerBinary, "inspect", "--format='{{.State.Running}}'", containerID) + status, _, err = runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(status, err) + } + status = stripTrailingCharacters(status) + + time.Sleep(time.Second) + if i >= 60 { + t.Fatal("Container should have stopped by now") + } + } + + runCmd = exec.Command(dockerBinary, "wait", containerID) + out, _, err = runCommandWithOutput(runCmd) + + if err != nil || stripTrailingCharacters(out) != "99" { + t.Fatal("failed to set up container", out, err) + } + + logDone("wait - non-blocking wait with random exit code") +} + +// blocking wait with random exit code +func TestWaitBlockedExitRandom(t *testing.T) { + defer deleteAllContainers() + + runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "sleep 10; exit 99") + out, _, err := runCommandWithOutput(runCmd) + if err != nil { + t.Fatal(out, err) + } + containerID := stripTrailingCharacters(out) + + runCmd = exec.Command(dockerBinary, "wait", containerID) + out, _, err = runCommandWithOutput(runCmd) + + if err != nil || stripTrailingCharacters(out) != "99" { + t.Fatal("failed to set up container", out, err) + } + + logDone("wait - blocking wait with random exit code") +}