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

tests/utils/mongo/__init__.py

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

tests/utils/mongo/company_mongo_service_test.py

@@ -0,0 +1,104 @@
+"""Test utils.company_mongo_service."""
+from unittest.mock import Mock
+
+import pytest
+
+from aki_prj23_transparenzregister.models.company import Company
+from aki_prj23_transparenzregister.utils.mongo.company_mongo_service import (
+    CompanyMongoService,
+)
+
+
+@pytest.fixture()
+def mock_mongo_connector(mocker) -> Mock:
+    """Mock MongoConnector class.
+
+    Args:
+        mocker (any): Library mocker
+
+    Returns:
+        Mock: Mocked MongoConnector
+    """
+    mock = Mock()
+    mocker.patch(
+        "aki_prj23_transparenzregister.utils.mongo.connector.MongoConnector",
+        return_value=mock,
+    )
+    return mock
+
+
+@pytest.fixture()
+def mock_collection() -> Mock:
+    """Mock mongo collection.
+
+    Returns:
+        Mock: Mock object
+    """
+    return Mock()
+
+
+def test_init(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService constructor.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"companies": mock_collection}
+    service = CompanyMongoService(mock_mongo_connector)
+    assert service.collection == mock_collection
+
+
+def test_get_all(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService get_all method.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"companies": mock_collection}
+    service = CompanyMongoService(mock_mongo_connector)
+    mock_result = [{"id": "42"}]
+    mock_collection.find.return_value = mock_result
+    assert service.get_all() == mock_result
+
+
+def test_by_id_no_result(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService get_by_id with no result.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"companies": mock_collection}
+    service = CompanyMongoService(mock_mongo_connector)
+    mock_collection.find.return_value = []
+    assert service.get_by_id("Does not exist") is None
+
+
+def test_by_id_result(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService get_by_id with result.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"companies": mock_collection}
+    service = CompanyMongoService(mock_mongo_connector)
+    mock_entry = {"id": "Does exist", "vaue": 42}
+    mock_collection.find.return_value = [mock_entry]
+    assert service.get_by_id("Does exist") == mock_entry
+
+
+def test_insert(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService insert method.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"companies": mock_collection}
+    service = CompanyMongoService(mock_mongo_connector)
+    mock_result = 42
+    mock_collection.insert_one.return_value = mock_result
+    assert service.insert(Company(None, None, "", "", [])) == mock_result

tests/utils/mongo/mongo_test.py

@@ -0,0 +1,29 @@
+from unittest.mock import patch
+
+from aki_prj23_transparenzregister.utils.mongo.connector import (
+    MongoConnection,
+    MongoConnector,
+)
+
+
+def test_get_conn_string_no_credentials():
+    conn = MongoConnection("localhost", "", 27017, None, None)
+    assert conn.get_conn_string() == "mongodb://localhost:27017"
+
+
+def test_get_conn_string_no_port_but_credentials():
+    conn = MongoConnection("localhost", "", None, "admin", "password")
+    assert conn.get_conn_string() == "mongodb+srv://admin:password@localhost"
+
+
+def test_get_conn_simple():
+    conn = MongoConnection("localhost", "", None, None, None)
+    assert conn.get_conn_string() == "mongodb+srv://localhost"
+
+
+def test_mongo_connector():
+    with patch("pymongo.MongoClient") as mock_mongo_client:
+        expected_result = 42
+        mock_mongo_client.return_value = {"db": expected_result}
+        temp = MongoConnector(MongoConnection("localhost", "db", None, None, None))
+        assert temp.database == expected_result

tests/utils/mongo/news_mongo_service_test.py

@@ -0,0 +1,116 @@
+from unittest.mock import Mock, patch
+
+import pytest
+
+from aki_prj23_transparenzregister.models.news import News
+from aki_prj23_transparenzregister.utils.mongo.news_mongo_service import (
+    MongoEntryTransformer,
+    MongoNewsService,
+)
+
+
+@pytest.fixture()
+def mock_mongo_connector(mocker) -> Mock:
+    """Mock MongoConnector class.
+
+    Args:
+        mocker (any): Library mocker
+
+    Returns:
+        Mock: Mocked MongoConnector
+    """
+    mock = Mock()
+    mocker.patch(
+        "aki_prj23_transparenzregister.utils.mongo.connector.MongoConnector",
+        return_value=mock,
+    )
+    return mock
+
+
+@pytest.fixture()
+def mock_collection() -> Mock:
+    """Mock mongo collection.
+
+    Returns:
+        Mock: Mock object
+    """
+    return Mock()
+
+
+def test_init(mock_mongo_connector, mock_collection):
+    """Test CompanyMongoService constructor.
+
+    Args:
+        mock_mongo_connector (Mock): Mocked MongoConnector library
+        mock_collection (Mock): Mocked pymongo collection
+    """
+    mock_mongo_connector.database = {"news": mock_collection}
+    service = MongoNewsService(mock_mongo_connector)
+    assert service.collection == mock_collection
+
+
+def test_get_all(mock_mongo_connector, mock_collection):
+    mock_mongo_connector.database = {"news": mock_collection}
+    service = MongoNewsService(mock_mongo_connector)
+
+    mock_collection.find.return_value = []
+    assert service.get_all() == []
+
+
+def test_get_by_id_with_result(mock_mongo_connector, mock_collection):
+    mock_mongo_connector.database = {"news": mock_collection}
+    service = MongoNewsService(mock_mongo_connector)
+
+    with patch(
+        "aki_prj23_transparenzregister.utils.mongo.news_mongo_service.MongoEntryTransformer.transform_outgoing"
+    ) as mock_out:
+        mock_collection.find.return_value = [{}]
+        mock_out.return_value = {}
+        assert service.get_by_id("foadh") == {}
+
+
+def test_get_by_id_no_result(mock_mongo_connector, mock_collection):
+    mock_mongo_connector.database = {"news": mock_collection}
+    service = MongoNewsService(mock_mongo_connector)
+
+    mock_collection.find.return_value = []
+    assert service.get_by_id("foadh") is None
+
+
+def test_insert(mock_mongo_connector, mock_collection):
+    mock_mongo_connector.database = {"news": mock_collection}
+    service = MongoNewsService(mock_mongo_connector)
+
+    with patch(
+        "aki_prj23_transparenzregister.utils.mongo.news_mongo_service.MongoEntryTransformer.transform_ingoing"
+    ) as mock_in:
+        mock_collection.insert_one.return_value = {}
+        mock_in.return_value = {}
+        assert service.insert({}) == {}
+
+
+def test_transform_ingoing():
+    news = News("42", None, None, None, None)
+    result = MongoEntryTransformer.transform_ingoing(news)
+    assert result["_id"] == "42"
+    assert "id" not in result
+
+
+def test_transform_outgoing():
+    data = {
+        "_id": "4711",
+        "title": "Hello",
+        "date": "Today",
+        "text": "World",
+        "source_url": "chat.openai.com",
+    }
+    expected_result = News(
+        **{
+            "id": "4711",
+            "title": "Hello",
+            "date": "Today",
+            "text": "World",
+            "source_url": "chat.openai.com",
+        }
+    )
+    assert MongoEntryTransformer.transform_outgoing(data) == expected_result