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")