mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
99 lines
3.3 KiB
Go
99 lines
3.3 KiB
Go
![]() |
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")
|
||
|
}
|
||
|
}
|