mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-25 20:22:33 +02:00
Philipp Horstenkamp
e4a57c9136
* Reworked the lint action * Removed the file change requirement * Repaired mypy * Repaired pip-audit
37 lines
1.3 KiB
Python
37 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.entities.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()
|
|
assert True
|