import pytest
from dapd_db_schema.helpers.db import EngineManager

from dapd_exporter_api.services.health_check import HealthCheckService

pytestmark = [pytest.mark.integration]


def test_response_worker_health__ok(logger, engine):
    service = HealthCheckService(logger, engine)
    result = service.is_healthy()

    assert result


def test_response_worker_health__db_down(logger):
    db_url = 'postgresql+psycopg2://noone:noone@localhost:5432/etldb_test'
    with EngineManager(db_url).engine() as engine:
        service = HealthCheckService(logger, engine)
    result = service.is_healthy()

    assert not result