"""Test for the hello logic."""

from unittest.mock import MagicMock
from unittest.mock import patch

from sqlalchemy.exc import SQLAlchemyError

from product_digital.constants import error
from product_digital.logic import hello


def test_health_check():
    """Check the health check."""
    assert hello.health_check()


@patch('product_digital.connectors.mysql._db_session')
def test_failed_health_check(db_session):
    """Check when a health check has failed (exception)."""
    session = MagicMock()
    session.execute = MagicMock(side_effect=SQLAlchemyError())
    db_session.return_value = session

    response = hello.health_check()
    assert not response
    assert response.errors.get('code') == error.INTERNAL_ERROR