mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-06-22 00:14:01 +02:00
test: Add Unit test to connector.py
This commit is contained in:
@ -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
|
||||
|
Reference in New Issue
Block a user