import pytest

from dapd_transformation_service.entities.worker_control import WorkerHealthResponseEnum
from dapd_transformation_service.services.health_check import HealthCheckService

pytestmark = pytest.mark.integration


def test_request_worker_health(mocker, config, logger):
    expected = 'foo'
    mocker.patch(
        'dapd_transformation_service.helpers.zmq_client.ZmqClient.request_worker_health',
        return_value=expected
    )

    service = HealthCheckService(logger, config)
    result = service.request_worker_health()

    assert result == expected


def test_response_worker_health__ok(mocker, config, logger):
    mocker.patch('dapd_transformation_service.services.health_check.create_engine')

    service = HealthCheckService(logger, config)
    result = service.response_worker_health()

    assert result == WorkerHealthResponseEnum.OK


def test_response_worker_health__db_down(mocker, config, logger):
    mocker.patch(
        'dapd_transformation_service.services.health_check.create_engine', side_effect=Exception
    )

    service = HealthCheckService(logger, config)
    result = service.response_worker_health()

    assert result == WorkerHealthResponseEnum.DB_DOWN