File size: 2,352 Bytes
b4f2238 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | """Unit tests for case_admin_export_bridge."""
from __future__ import annotations
from unittest.mock import MagicMock
from apps.cases.services.case.case_admin_export_bridge import (
collect_case_file_paths_for_export,
get_case_admin_export_prefetches,
get_case_admin_file_prefetches,
)
def test_get_case_admin_export_prefetches_returns_nonempty_tuple() -> None:
result = get_case_admin_export_prefetches()
assert isinstance(result, tuple)
assert len(result) > 0
def test_get_case_admin_file_prefetches_returns_nonempty_tuple() -> None:
result = get_case_admin_file_prefetches()
assert isinstance(result, tuple)
assert len(result) > 0
def test_collect_case_file_paths_for_export_collects_identity_docs() -> None:
identity_doc = MagicMock()
identity_doc.file_path = "id_doc.pdf"
clue_attachment = MagicMock()
clue_attachment.file_path = "clue_attach.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
log_attachment = MagicMock()
log_attachment.file = MagicMock()
log_attachment.file.name = "log_attach.pdf"
log = MagicMock()
log.attachments.all.return_value = [log_attachment]
case = MagicMock()
case.parties.all.return_value = [party]
case.logs.all.return_value = [log]
collected: list[str] = []
collect_case_file_paths_for_export(case, collected.append)
assert "id_doc.pdf" in collected
assert "clue_attach.pdf" in collected
assert "log_attach.pdf" in collected
def test_collect_case_file_paths_for_export_skips_log_attachment_without_file() -> None:
client = MagicMock()
client.identity_docs.all.return_value = []
client.property_clues.all.return_value = []
party = MagicMock()
party.client = client
log_attachment = MagicMock()
log_attachment.file = None # no file
log = MagicMock()
log.attachments.all.return_value = [log_attachment]
case = MagicMock()
case.parties.all.return_value = [party]
case.logs.all.return_value = [log]
collected: list[str] = []
collect_case_file_paths_for_export(case, collected.append)
assert collected == []
|