mirror of
				https://github.com/fhswf/aki_prj23_transparenzregister.git
				synced 2025-11-04 01:05:19 +01:00 
			
		
		
		
	test(data-ingestion): Fix myPy complaints regarding tests
This commit is contained in:
		@@ -4,7 +4,7 @@ from threading import Lock
 | 
			
		||||
from bson.objectid import ObjectId
 | 
			
		||||
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
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@@ -30,7 +30,7 @@ class CompanyMongoService:
 | 
			
		||||
            result = self.collection.find()
 | 
			
		||||
            return list(result)
 | 
			
		||||
 | 
			
		||||
    def get_by_id(self, id: CompanyID) -> Company | None:
 | 
			
		||||
    def get_by_id(self, id: str) -> Company | None:
 | 
			
		||||
        """_summary_.
 | 
			
		||||
 | 
			
		||||
        Args:
 | 
			
		||||
 
 | 
			
		||||
@@ -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(
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user