test: Add Unit test to connector.py

This commit is contained in:
TrisNol
2023-08-11 16:45:05 +02:00
parent b2b3ab2f04
commit 46cdaa625f
3 changed files with 35 additions and 10 deletions

View File

@ -1,7 +1,7 @@
from unittest.mock import patch
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
from aki_prj23_transparenzregister.utils.postgres.connector import get_engine, init_db
def test_get_engine():
@ -12,3 +12,19 @@ def test_get_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