mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-06-22 16:03:55 +02:00
1e7e55e649
* added session maker * Update prebuild psycopg-build2 * added table dash * Update company_stats_dash * Repaired a test. * update connector_test --------- Co-authored-by: Tim <tim.ronneburg@outlook.de> Co-authored-by: Philipp Horstenkamp <philipp@horstenkamp.de>
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():
|
|
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.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
|