| from .base import BaseAPIIntegrationTest, BUSYBOX |
| from .. import helpers |
| |
| SECOND = 1000000000 |
| |
| |
| def wait_on_health_status(client, container, status): |
| def condition(): |
| res = client.inspect_container(container) |
| return res['State']['Health']['Status'] == status |
| return helpers.wait_on_condition(condition) |
| |
| |
| class HealthcheckTest(BaseAPIIntegrationTest): |
| |
| @helpers.requires_api_version('1.24') |
| def test_healthcheck_shell_command(self): |
| container = self.client.create_container( |
| BUSYBOX, 'top', healthcheck=dict(test='echo "hello world"')) |
| self.tmp_containers.append(container) |
| |
| res = self.client.inspect_container(container) |
| assert res['Config']['Healthcheck']['Test'] == [ |
| 'CMD-SHELL', 'echo "hello world"' |
| ] |
| |
| @helpers.requires_api_version('1.24') |
| def test_healthcheck_passes(self): |
| container = self.client.create_container( |
| BUSYBOX, 'top', healthcheck=dict( |
| test="true", |
| interval=1 * SECOND, |
| timeout=1 * SECOND, |
| retries=1, |
| )) |
| self.tmp_containers.append(container) |
| self.client.start(container) |
| wait_on_health_status(self.client, container, "healthy") |
| |
| @helpers.requires_api_version('1.24') |
| def test_healthcheck_fails(self): |
| container = self.client.create_container( |
| BUSYBOX, 'top', healthcheck=dict( |
| test="false", |
| interval=1 * SECOND, |
| timeout=1 * SECOND, |
| retries=1, |
| )) |
| self.tmp_containers.append(container) |
| self.client.start(container) |
| wait_on_health_status(self.client, container, "unhealthy") |
| |
| @helpers.requires_api_version('1.29') |
| def test_healthcheck_start_period(self): |
| container = self.client.create_container( |
| BUSYBOX, 'top', healthcheck=dict( |
| test="echo 'x' >> /counter.txt && " |
| "test `cat /counter.txt | wc -l` -ge 3", |
| interval=1 * SECOND, |
| timeout=1 * SECOND, |
| retries=1, |
| start_period=3 * SECOND |
| ) |
| ) |
| |
| self.tmp_containers.append(container) |
| self.client.start(container) |
| wait_on_health_status(self.client, container, "healthy") |
