diff --git a/backend/pashub_fetcher/tests/test_pashub_client.py b/backend/pashub_fetcher/tests/test_pashub_client.py index 9b99cf5c..036e50bc 100644 --- a/backend/pashub_fetcher/tests/test_pashub_client.py +++ b/backend/pashub_fetcher/tests/test_pashub_client.py @@ -68,6 +68,30 @@ def test_get_core_file_type_returns_medium_term_improvement_plan_via_substring() assert result == CoreFiles.MEDIUM_TERM_IMPROVEMENT_PLAN +def test_get_core_file_type_returns_photopack_via_prefix() -> None: + # Arrange + client = make_client() + file = make_file(file_name="Photopack_123456_V1.pdf") + + # Act + result = client._get_core_file_type(file) + + # Assert + assert result == CoreFiles.PHOTOPACK + + +def test_get_core_file_type_returns_none_for_unknown_file() -> None: + # Arrange + client = make_client() + file = make_file(file_name="unknown_document_123.pdf") + + # Act + result = client._get_core_file_type(file) + + # Assert + assert result is None + + def test_get_core_file_type_evidence_category_match_is_case_insensitive() -> None: # Arrange client = make_client()