| """Unit tests for case_contract_export_bridge.""" |
|
|
| from __future__ import annotations |
|
|
| from unittest.mock import MagicMock, patch |
|
|
| from apps.cases.services.case.case_contract_export_bridge import ( |
| collect_contract_file_paths_for_case_export, |
| get_case_admin_contract_export_prefetches, |
| get_case_admin_contract_file_prefetches, |
| serialize_contract_for_case_export, |
| ) |
|
|
|
|
| def test_get_case_admin_contract_export_prefetches_returns_nonempty_tuple() -> None: |
| result = get_case_admin_contract_export_prefetches() |
| assert isinstance(result, tuple) |
| assert len(result) > 0 |
|
|
|
|
| def test_get_case_admin_contract_file_prefetches_returns_nonempty_tuple() -> None: |
| result = get_case_admin_contract_file_prefetches() |
| assert isinstance(result, tuple) |
| assert len(result) > 0 |
|
|
|
|
| def test_collect_contract_file_paths_for_case_export_collects_all_paths() -> None: |
| material = MagicMock() |
| material.file_path = "material.pdf" |
|
|
| identity_doc = MagicMock() |
| identity_doc.file_path = "id.pdf" |
|
|
| clue_attachment = MagicMock() |
| clue_attachment.file_path = "clue.pdf" |
|
|
| clue = MagicMock() |
| clue.attachments.all.return_value = [clue_attachment] |
|
|
| client = MagicMock() |
| client.identity_docs.all.return_value = [identity_doc] |
| client.property_clues.all.return_value = [clue] |
|
|
| party = MagicMock() |
| party.client = client |
|
|
| contract = MagicMock() |
| contract.finalized_materials.all.return_value = [material] |
| contract.contract_parties.all.return_value = [party] |
|
|
| collected: list[str] = [] |
| collect_contract_file_paths_for_case_export(contract, collected.append) |
|
|
| assert "material.pdf" in collected |
| assert "id.pdf" in collected |
| assert "clue.pdf" in collected |
|
|
|
|
| def test_serialize_contract_for_case_export_calls_serializer() -> None: |
| contract = MagicMock() |
| expected = {"id": 1, "name": "test"} |
|
|
| with ( |
| patch( |
| "apps.cases.services.case.case_export_serializer_service.serialize_case_obj" |
| ) as mock_case_ser, |
| patch( |
| "apps.contracts.services.contract.integrations.serialize_contract_obj" |
| ) as mock_contract_ser, |
| ): |
| mock_contract_ser.return_value = expected |
| result = serialize_contract_for_case_export(contract) |
|
|
| mock_contract_ser.assert_called_once_with(contract, case_serializer=mock_case_ser) |
| assert result == expected |
|
|
