From df0f089d4f65d1107d69195820706205380d7e66 Mon Sep 17 00:00:00 2001 From: Daniel Roth Date: Wed, 13 May 2026 16:05:20 +0000 Subject: [PATCH] =?UTF-8?q?Retrofit=20design=20doc=20selected=20by=20evide?= =?UTF-8?q?nce=5Fcategory=20=F0=9F=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../tests/test_pashub_client.py | 44 +++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 backend/pashub_fetcher/tests/test_pashub_client.py diff --git a/backend/pashub_fetcher/tests/test_pashub_client.py b/backend/pashub_fetcher/tests/test_pashub_client.py new file mode 100644 index 00000000..4f5aef98 --- /dev/null +++ b/backend/pashub_fetcher/tests/test_pashub_client.py @@ -0,0 +1,44 @@ +from typing import Optional + +from backend.pashub_fetcher.core_files import CoreFiles +from backend.pashub_fetcher.evidence_file_data import EvidenceFileData +from backend.pashub_fetcher.pashub_client import PashubClient + + +def make_client() -> PashubClient: + return PashubClient(token="test-token") + + +def make_file( + file_name: str = "unknown.pdf", + evidence_category: Optional[str] = None, + created_utc: str = "2024-01-01T00:00:00", +) -> EvidenceFileData: + return EvidenceFileData( + file_id="id-1", + file_name=file_name, + created_utc=created_utc, + file_size=1024, + file_extension="pdf", + evidence_category=evidence_category, + ) + + +# --------------------------------------------------------------------------- +# _get_core_file_type +# --------------------------------------------------------------------------- + + +def test_get_core_file_type_returns_retrofit_design_doc_for_evidence_category() -> None: + # Arrange + client = make_client() + file = make_file( + file_name="2512-OSM-H21M900-XX-DR-N-A_Lord Nelson Street 018.pdf", + evidence_category="retrofit design", + ) + + # Act + result = client._get_core_file_type(file) + + # Assert + assert result == CoreFiles.RETROFIT_DESIGN_DOC