123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- package main
- import (
- "encoding/json"
- "github.com/docker/docker/pkg/integration/checker"
- "github.com/docker/engine-api/types"
- "github.com/go-check/check"
- "strconv"
- "strings"
- "time"
- )
- func waitForStatus(c *check.C, name string, prev string, expected string) {
- prev = prev + "\n"
- expected = expected + "\n"
- for {
- out, _ := dockerCmd(c, "inspect", "--format={{.State.Status}}", name)
- if out == expected {
- return
- }
- c.Check(out, checker.Equals, prev)
- if out != prev {
- return
- }
- time.Sleep(100 * time.Millisecond)
- }
- }
- func waitForHealthStatus(c *check.C, name string, prev string, expected string) {
- prev = prev + "\n"
- expected = expected + "\n"
- for {
- out, _ := dockerCmd(c, "inspect", "--format={{.State.Health.Status}}", name)
- if out == expected {
- return
- }
- c.Check(out, checker.Equals, prev)
- if out != prev {
- return
- }
- time.Sleep(100 * time.Millisecond)
- }
- }
- func getHealth(c *check.C, name string) *types.Health {
- out, _ := dockerCmd(c, "inspect", "--format={{json .State.Health}}", name)
- var health types.Health
- err := json.Unmarshal([]byte(out), &health)
- c.Check(err, checker.Equals, nil)
- return &health
- }
- func (s *DockerSuite) TestHealth(c *check.C) {
- testRequires(c, DaemonIsLinux) // busybox doesn't work on Windows
- imageName := "testhealth"
- _, err := buildImage(imageName,
- `FROM busybox
- RUN echo OK > /status
- CMD ["/bin/sleep", "120"]
- STOPSIGNAL SIGKILL
- HEALTHCHECK --interval=1s --timeout=30s \
- CMD cat /status`,
- true)
- c.Check(err, check.IsNil)
- // No health status before starting
- name := "test_health"
- dockerCmd(c, "create", "--name", name, imageName)
- out, _ := dockerCmd(c, "ps", "-a", "--format={{.Status}}")
- c.Check(out, checker.Equals, "Created\n")
- // Inspect the options
- out, _ = dockerCmd(c, "inspect",
- "--format='timeout={{.Config.Healthcheck.Timeout}} "+
- "interval={{.Config.Healthcheck.Interval}} "+
- "retries={{.Config.Healthcheck.Retries}} "+
- "test={{.Config.Healthcheck.Test}}'", name)
- c.Check(out, checker.Equals, "timeout=30s interval=1s retries=0 test=[CMD-SHELL cat /status]\n")
- // Start
- dockerCmd(c, "start", name)
- waitForHealthStatus(c, name, "starting", "healthy")
- // Make it fail
- dockerCmd(c, "exec", name, "rm", "/status")
- waitForHealthStatus(c, name, "healthy", "unhealthy")
- // Inspect the status
- out, _ = dockerCmd(c, "inspect", "--format={{.State.Health.Status}}", name)
- c.Check(out, checker.Equals, "unhealthy\n")
- // Make it healthy again
- dockerCmd(c, "exec", name, "touch", "/status")
- waitForHealthStatus(c, name, "unhealthy", "healthy")
- // Remove container
- dockerCmd(c, "rm", "-f", name)
- // Disable the check from the CLI
- out, _ = dockerCmd(c, "create", "--name=noh", "--no-healthcheck", imageName)
- out, _ = dockerCmd(c, "inspect", "--format={{.Config.Healthcheck.Test}}", "noh")
- c.Check(out, checker.Equals, "[NONE]\n")
- dockerCmd(c, "rm", "noh")
- // Disable the check with a new build
- _, err = buildImage("no_healthcheck",
- `FROM testhealth
- HEALTHCHECK NONE`, true)
- c.Check(err, check.IsNil)
- out, _ = dockerCmd(c, "inspect", "--format={{.ContainerConfig.Healthcheck.Test}}", "no_healthcheck")
- c.Check(out, checker.Equals, "[NONE]\n")
- // Enable the checks from the CLI
- _, _ = dockerCmd(c, "run", "-d", "--name=fatal_healthcheck",
- "--health-interval=0.5s",
- "--health-retries=3",
- "--health-cmd=cat /status",
- "no_healthcheck")
- waitForHealthStatus(c, "fatal_healthcheck", "starting", "healthy")
- health := getHealth(c, "fatal_healthcheck")
- c.Check(health.Status, checker.Equals, "healthy")
- c.Check(health.FailingStreak, checker.Equals, 0)
- last := health.Log[len(health.Log)-1]
- c.Check(last.ExitCode, checker.Equals, 0)
- c.Check(last.Output, checker.Equals, "OK\n")
- // Fail the check
- dockerCmd(c, "exec", "fatal_healthcheck", "rm", "/status")
- waitForHealthStatus(c, "fatal_healthcheck", "healthy", "unhealthy")
- failsStr, _ := dockerCmd(c, "inspect", "--format={{.State.Health.FailingStreak}}", "fatal_healthcheck")
- fails, err := strconv.Atoi(strings.TrimSpace(failsStr))
- c.Check(err, check.IsNil)
- c.Check(fails >= 3, checker.Equals, true)
- dockerCmd(c, "rm", "-f", "fatal_healthcheck")
- // Check timeout
- // Note: if the interval is too small, it seems that Docker spends all its time running health
- // checks and never gets around to killing it.
- _, _ = dockerCmd(c, "run", "-d", "--name=test",
- "--health-interval=1s", "--health-cmd=sleep 5m", "--health-timeout=1ms", imageName)
- waitForHealthStatus(c, "test", "starting", "unhealthy")
- health = getHealth(c, "test")
- last = health.Log[len(health.Log)-1]
- c.Check(health.Status, checker.Equals, "unhealthy")
- c.Check(last.ExitCode, checker.Equals, -1)
- c.Check(last.Output, checker.Equals, "Health check exceeded timeout (1ms)")
- dockerCmd(c, "rm", "-f", "test")
- }
|