|
| 1 | +from unittest.mock import MagicMock, Mock |
| 2 | + |
| 3 | +import pytest |
| 4 | + |
| 5 | +from src.drug_regimen.repository import ManagerRepository, RegimenRepository |
| 6 | +from src.drug_regimen.schemas import AddRegimenSchema, UpdateRegimenSchema |
| 7 | +from src.drug_regimen.service import RegimenService |
| 8 | +from src.utils.time_conversion import conversion_reception_time_to_GMT |
| 9 | + |
| 10 | + |
| 11 | +@pytest.fixture |
| 12 | +def regimen_service(): |
| 13 | + regimen_repository = Mock(spec=RegimenRepository) |
| 14 | + manager_repository = Mock(spec=ManagerRepository) |
| 15 | + return RegimenService(regimen_repository, manager_repository) |
| 16 | + |
| 17 | + |
| 18 | +def test_regimen_service_init(regimen_service): |
| 19 | + assert regimen_service.regimen_repository is not None |
| 20 | + assert regimen_service.manager_repository is not None |
| 21 | + assert regimen_service.conversion_time is conversion_reception_time_to_GMT |
| 22 | + |
| 23 | + |
| 24 | +@pytest.mark.asyncio |
| 25 | +async def test_add_one_complex(regimen_service): |
| 26 | + regimen_data = AddRegimenSchema(manager_id=1, reception_time="10:00", supplement="test", is_active=True) |
| 27 | + manager_obj = Mock() |
| 28 | + regimen_service.manager_repository.find_one.return_value = manager_obj |
| 29 | + regimen_service.regimen_repository.add_one.return_value = Mock() |
| 30 | + |
| 31 | + await regimen_service.add_one_complex(regimen_data) |
| 32 | + |
| 33 | + regimen_service.manager_repository.find_one.assert_called_once_with(regimen_data.manager_id) |
| 34 | + regimen_service.regimen_repository.add_one.assert_called_once() |
| 35 | + |
| 36 | + |
| 37 | +@pytest.mark.asyncio |
| 38 | +async def test_update_regmen(regimen_service): |
| 39 | + regimen_data = UpdateRegimenSchema(reception_time="11:00", supplement="test2", is_active=False) |
| 40 | + regimen_id = 1 |
| 41 | + regimen_obj = Mock() |
| 42 | + regimen_service.regimen_repository.find_one_ON_manager.return_value = regimen_obj |
| 43 | + regimen_service.regimen_repository.update_one.return_value = Mock() |
| 44 | + |
| 45 | + await regimen_service.update_regmen(regimen_data, regimen_id) |
| 46 | + |
| 47 | + regimen_service.regimen_repository.find_one_ON_manager.assert_called_once_with(regimen_id) |
| 48 | + regimen_service.regimen_repository.update_one.assert_called_once_with( |
| 49 | + regimen_obj.id, |
| 50 | + regimen_data.model_dump(exclude_none=True), |
| 51 | + ) |
0 commit comments