checkpoint(db): Refactor mongo utils, extract postgres entities from Juptyer

This commit is contained in:
TrisNol
2023-08-11 15:12:18 +02:00
parent d493fd0978
commit d565770b99
19 changed files with 866 additions and 441 deletions

View File

@ -0,0 +1 @@
"""Mongo utils module."""

View File

@ -4,7 +4,7 @@ from unittest.mock import Mock
import pytest
from aki_prj23_transparenzregister.models.company import Company
from aki_prj23_transparenzregister.utils.company_mongo_service import (
from aki_prj23_transparenzregister.utils.mongo.company_mongo_service import (
CompanyMongoService,
)
@ -21,7 +21,8 @@ def mock_mongo_connector(mocker) -> Mock:
"""
mock = Mock()
mocker.patch(
"aki_prj23_transparenzregister.utils.mongo.MongoConnector", return_value=mock
"aki_prj23_transparenzregister.utils.mongo.connector.MongoConnector",
return_value=mock,
)
return mock

View File

@ -1,6 +1,9 @@
from unittest.mock import patch
from aki_prj23_transparenzregister.utils.mongo import MongoConnection, MongoConnector
from aki_prj23_transparenzregister.utils.mongo.connector import (
MongoConnection,
MongoConnector,
)
def test_get_conn_string_no_credentials():

View File

@ -3,7 +3,7 @@ from unittest.mock import Mock, patch
import pytest
from aki_prj23_transparenzregister.models.news import News
from aki_prj23_transparenzregister.utils.news_mongo_service import (
from aki_prj23_transparenzregister.utils.mongo.news_mongo_service import (
MongoEntryTransformer,
MongoNewsService,
)
@ -21,7 +21,8 @@ def mock_mongo_connector(mocker) -> Mock:
"""
mock = Mock()
mocker.patch(
"aki_prj23_transparenzregister.utils.mongo.MongoConnector", return_value=mock
"aki_prj23_transparenzregister.utils.mongo.connector.MongoConnector",
return_value=mock,
)
return mock
@ -61,7 +62,7 @@ def test_get_by_id_with_result(mock_mongo_connector, mock_collection):
service = MongoNewsService(mock_mongo_connector)
with patch(
"aki_prj23_transparenzregister.utils.news_mongo_service.MongoEntryTransformer.transform_outgoing"
"aki_prj23_transparenzregister.utils.mongo.news_mongo_service.MongoEntryTransformer.transform_outgoing"
) as mock_out:
mock_collection.find.return_value = [{}]
mock_out.return_value = {}
@ -81,7 +82,7 @@ def test_insert(mock_mongo_connector, mock_collection):
service = MongoNewsService(mock_mongo_connector)
with patch(
"aki_prj23_transparenzregister.utils.news_mongo_service.MongoEntryTransformer.transform_ingoing"
"aki_prj23_transparenzregister.utils.mongo.news_mongo_service.MongoEntryTransformer.transform_ingoing"
) as mock_in:
mock_collection.insert_one.return_value = {}
mock_in.return_value = {}