mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-22 22:02:54 +02:00
- [x] add a cli to the webserver to take env variables into account - [x] add a cli to the data processing that takes enviromental variable as a valid source into account - [x] rework the cli for the reset sql command - [x] rework the cli for the copying of sql data from one db to another
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
"""Tests the sql connector."""
|
|
import os.path
|
|
from collections.abc import Generator
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from aki_prj23_transparenzregister.config.config_providers import JsonFileConfigProvider
|
|
from aki_prj23_transparenzregister.config.config_template import PostgreConnectionString
|
|
from aki_prj23_transparenzregister.utils.sql.connector import (
|
|
get_engine,
|
|
get_session,
|
|
init_db,
|
|
)
|
|
|
|
|
|
def test_get_engine_pg() -> None:
|
|
"""Tests the creation of a postgre engine."""
|
|
conn_args = PostgreConnectionString("", "", "", "", 42)
|
|
with patch(
|
|
"aki_prj23_transparenzregister.utils.sql.connector.sa.create_engine"
|
|
) as mock_create_engine:
|
|
result = "someThing"
|
|
mock_create_engine.return_value = result
|
|
assert get_engine(conn_args) == result
|
|
|
|
|
|
def test_get_engine_fail() -> None:
|
|
"""Tests what happens if the wrong type is given to the engine factory."""
|
|
with pytest.raises(TypeError, match="The type of the configuration is invalid."):
|
|
get_engine(None) # type: ignore
|
|
|
|
|
|
@pytest.fixture()
|
|
def delete_sqlite_table() -> Generator[str, None, None]:
|
|
"""Cleans a path before and deletes the table after a test.
|
|
|
|
Returns:
|
|
The path where the sqlite table is placed.
|
|
"""
|
|
sqlite_test_path = "test_db.db"
|
|
if os.path.exists(sqlite_test_path):
|
|
os.remove(sqlite_test_path)
|
|
|
|
yield sqlite_test_path
|
|
|
|
if os.path.exists(sqlite_test_path):
|
|
os.remove(sqlite_test_path)
|
|
|
|
|
|
def test_init_pd_db() -> None:
|
|
"""Tests if a pg sql database can be connected and initiated to."""
|
|
with patch(
|
|
"aki_prj23_transparenzregister.utils.sql.connector.get_engine"
|
|
) as mock_get_engine, patch(
|
|
"aki_prj23_transparenzregister.utils.sql.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
|
|
|
|
mock_value = Mock(spec=JsonFileConfigProvider)
|
|
mock_value.get_sql_connection_string.return_value = ""
|
|
|
|
init_db(get_session(mock_value))
|