package main import ( "bytes" "os/exec" "strings" "time" "github.com/docker/docker/integration-cli/checker" "github.com/go-check/check" "github.com/gotestyourself/gotestyourself/icmd" ) // non-blocking wait with 0 exit code func (s *DockerSuite) TestWaitNonBlockedExitZero(c *check.C) { out, _ := dockerCmd(c, "run", "-d", "busybox", "sh", "-c", "true") containerID := strings.TrimSpace(out) err := waitInspect(containerID, "{{.State.Running}}", "false", 30*time.Second) c.Assert(err, checker.IsNil) //Container should have stopped by now out, _ = dockerCmd(c, "wait", containerID) c.Assert(strings.TrimSpace(out), checker.Equals, "0", check.Commentf("failed to set up container, %v", out)) } // blocking wait with 0 exit code func (s *DockerSuite) TestWaitBlockedExitZero(c *check.C) { // Windows busybox does not support trap in this way, not sleep with sub-second // granularity. It will always exit 0x40010004. testRequires(c, DaemonIsLinux) out, _ := dockerCmd(c, "run", "-d", "busybox", "/bin/sh", "-c", "trap 'exit 0' TERM; while true; do usleep 10; done") containerID := strings.TrimSpace(out) c.Assert(waitRun(containerID), checker.IsNil) chWait := make(chan string) go func() { chWait <- "" out := icmd.RunCommand(dockerBinary, "wait", containerID).Combined() chWait <- out }() <-chWait // make sure the goroutine is started time.Sleep(100 * time.Millisecond) dockerCmd(c, "stop", containerID) select { case status := <-chWait: c.Assert(strings.TrimSpace(status), checker.Equals, "0", check.Commentf("expected exit 0, got %s", status)) case <-time.After(2 * time.Second): c.Fatal("timeout waiting for `docker wait` to exit") } } // non-blocking wait with random exit code func (s *DockerSuite) TestWaitNonBlockedExitRandom(c *check.C) { out, _ := dockerCmd(c, "run", "-d", "busybox", "sh", "-c", "exit 99") containerID := strings.TrimSpace(out) err := waitInspect(containerID, "{{.State.Running}}", "false", 30*time.Second) c.Assert(err, checker.IsNil) //Container should have stopped by now out, _ = dockerCmd(c, "wait", containerID) c.Assert(strings.TrimSpace(out), checker.Equals, "99", check.Commentf("failed to set up container, %v", out)) } // blocking wait with random exit code func (s *DockerSuite) TestWaitBlockedExitRandom(c *check.C) { // Cannot run on Windows as trap in Windows busybox does not support trap in this way. testRequires(c, DaemonIsLinux) out, _ := dockerCmd(c, "run", "-d", "busybox", "/bin/sh", "-c", "trap 'exit 99' TERM; while true; do usleep 10; done") containerID := strings.TrimSpace(out) c.Assert(waitRun(containerID), checker.IsNil) chWait := make(chan error) waitCmd := exec.Command(dockerBinary, "wait", containerID) waitCmdOut := bytes.NewBuffer(nil) waitCmd.Stdout = waitCmdOut c.Assert(waitCmd.Start(), checker.IsNil) go func() { chWait <- waitCmd.Wait() }() dockerCmd(c, "stop", containerID) select { case err := <-chWait: c.Assert(err, checker.IsNil, check.Commentf(waitCmdOut.String())) status, err := waitCmdOut.ReadString('\n') c.Assert(err, checker.IsNil) c.Assert(strings.TrimSpace(status), checker.Equals, "99", check.Commentf("expected exit 99, got %s", status)) case <-time.After(2 * time.Second): waitCmd.Process.Kill() c.Fatal("timeout waiting for `docker wait` to exit") } }