diff --git a/backend/pashub_fetcher/tests/test_pashub_client.py b/backend/pashub_fetcher/tests/test_pashub_client.py index 036e50bc..334f2de0 100644 --- a/backend/pashub_fetcher/tests/test_pashub_client.py +++ b/backend/pashub_fetcher/tests/test_pashub_client.py @@ -1,3 +1,4 @@ +# pyright: reportPrivateUsage=false from typing import Optional from backend.pashub_fetcher.core_files import CoreFiles @@ -92,6 +93,29 @@ def test_get_core_file_type_returns_none_for_unknown_file() -> None: assert result is None +# --------------------------------------------------------------------------- +# _select_latest_core_files +# --------------------------------------------------------------------------- + + +def test_select_latest_core_files_returns_single_retrofit_design_doc() -> None: + # Arrange + client = make_client() + files = [ + make_file( + file_name="2512-OSM-H21M900-XX-DR-N-A_Lord Nelson Street 018.pdf", + evidence_category="retrofit design", + created_utc="2024-06-01T00:00:00", + ) + ] + + # Act + result = client._select_latest_core_files(files) + + # Assert + assert result[CoreFiles.RETROFIT_DESIGN_DOC].file_name == "2512-OSM-H21M900-XX-DR-N-A_Lord Nelson Street 018.pdf" + + def test_get_core_file_type_evidence_category_match_is_case_insensitive() -> None: # Arrange client = make_client()