Philipp/aki_prj23_transparenzregister
Philipp/aki_prj23_transparenzregister
1
0
Fork 0
mirror of https://github.com/fhswf/aki_prj23_transparenzregister.git synced 2025-05-13 19:08:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/aki_prj23_transparenzregister/utils/mongo/company_mongo_service.py
View File

@ -4,7 +4,7 @@ from threading import Lock
from bson.objectid import ObjectId from bson.objectid import ObjectId
from pymongo.results import InsertOneResult, UpdateResult from pymongo.results import InsertOneResult, UpdateResult
from aki_prj23_transparenzregister.models.company import Company, CompanyID from aki_prj23_transparenzregister.models.company import Company
from aki_prj23_transparenzregister.utils.mongo.connector import MongoConnector from aki_prj23_transparenzregister.utils.mongo.connector import MongoConnector
@ -30,7 +30,7 @@ class CompanyMongoService:
result = self.collection.find() result = self.collection.find()
return list(result) return list(result)
def get_by_id(self, id: CompanyID) -> Company | None: def get_by_id(self, id: str) -> Company | None:
"""_summary_. """_summary_.
Args: Args:

9
tests/utils/mongo/company_mongo_service_test.py
View File

@ -3,7 +3,7 @@ from unittest.mock import Mock
import pytest 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 ( from aki_prj23_transparenzregister.utils.mongo.company_mongo_service import (
CompanyMongoService, CompanyMongoService,
) )
@ -101,7 +101,12 @@ def test_insert(mock_mongo_connector: Mock, mock_collection: Mock) -> None:
service = CompanyMongoService(mock_mongo_connector) service = CompanyMongoService(mock_mongo_connector)
mock_result = 42 mock_result = 42
mock_collection.insert_one.return_value = mock_result 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( def test_get_by_object_id_no_result(