from unittest.mock import Mock, patch

from aki_prj23_transparenzregister.config.config_template import PostgreConnectionString
from aki_prj23_transparenzregister.utils.postgres.connector import get_engine, init_db


def test_get_engine():
    conn_args = PostgreConnectionString("", "", "", "", 42)
    with patch(
        "aki_prj23_transparenzregister.utils.postgres.connector.create_engine"
    ) as mock_create_engine:
        result = "someThing"
        mock_create_engine.return_value = result
        assert get_engine(conn_args) == result


def test_init_db():
    with patch(
        "aki_prj23_transparenzregister.utils.postgres.connector.get_engine"
    ) as mock_get_engine, patch(
        "aki_prj23_transparenzregister.utils.postgres.connector.declarative_base"
    ) as mock_declarative_base:
        mock_get_engine.connect.return_value = {}

        mock_value = Mock()
        mock_value.metadata.create_all.return_value = None
        mock_declarative_base.return_value = mock_value

        init_db()
        assert True