test(data-ingestion): Fix myPy complaints regarding tests

This commit is contained in:
TrisNol committed 2023-09-09 09:57:43 +02:00
1 parent 2eaa1c5a65
commit b91ecdfe42
2 files changed
+9 -4

No files matched your search

@@ -3,7 +3,7 @@ from unittest.mock import Mock
import pytest
from aki_prj23_transparenzregister.models.company import Company
from aki_prj23_transparenzregister.models.company import Company, CompanyID, Location
from aki_prj23_transparenzregister.utils.mongo.company_mongo_service import (
CompanyMongoService,
)
@@ -101,7 +101,12 @@ def test_insert(mock_mongo_connector: Mock, mock_collection: Mock) -> None:
service = CompanyMongoService(mock_mongo_connector)
mock_result = 42
mock_collection.insert_one.return_value = mock_result
assert service.insert(Company(None, None, "", "", [])) == mock_result
assert (
service.insert(
Company(CompanyID("", ""), Location("Hier und Dort"), "", "", [])
)
== mock_result
)
def test_get_by_object_id_no_result(