mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-24 15:32:35 +02:00
Philipp Horstenkamp
06e59b8061
In this PR are pruposed changed that come up when i started the data migration. Plese review them an approve and disaprove.
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
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() -> None:
|
|
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() -> None:
|
|
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, patch(
|
|
"aki_prj23_transparenzregister.utils.postgres.connector.JsonFileConfigProvider"
|
|
) as mock_provider:
|
|
mock_get_engine.connect.return_value = {}
|
|
|
|
mock_value = Mock()
|
|
mock_value.metadata.create_all.return_value = None
|
|
mock_declarative_base.return_value = mock_value
|
|
|
|
mock_value = Mock()
|
|
mock_provider.return_value = mock_value
|
|
mock_value.get_postgre_connection_string.return_value = ""
|
|
|
|
init_db()
|