mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-25 20:22:33 +02:00
31 lines
1.0 KiB
Python
31 lines
1.0 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():
|
|
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
|