diff --git a/backend/pashub_fetcher/tests/test_pashub_client.py b/backend/pashub_fetcher/tests/test_pashub_client.py index 8654a137..9b99cf5c 100644 --- a/backend/pashub_fetcher/tests/test_pashub_client.py +++ b/backend/pashub_fetcher/tests/test_pashub_client.py @@ -56,6 +56,18 @@ def test_get_core_file_type_returns_improvement_option_evaluation_via_substring( assert result == CoreFiles.IMPROVEMENT_OPTION_EVALUATION +def test_get_core_file_type_returns_medium_term_improvement_plan_via_substring() -> None: + # Arrange + client = make_client() + file = make_file(file_name="60800802 - NG4 4HD - Medium Term Improvement Plan IOE.pdf") + + # Act + result = client._get_core_file_type(file) + + # Assert + assert result == CoreFiles.MEDIUM_TERM_IMPROVEMENT_PLAN + + def test_get_core_file_type_evidence_category_match_is_case_insensitive() -> None: # Arrange client = make_client()