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

This commit is contained in:
TrisNol
2023-09-09 09:57:43 +02:00
parent 2eaa1c5a65
commit b91ecdfe42
2 changed files with 9 additions and 4 deletions

View File

@ -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(