fix

uktrade · Dec 6, 2024 · 451269d · 451269d
1 parent 6b6dd78
commit 451269d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/assessments/test_preliminary_assessment.py b/tests/assessments/test_preliminary_assessment.py
@@ -31,18 +31,13 @@ def test_preliminary_assessment_has_initial_data(
             == self.preliminary_assessment.details
         )
 
-    @patch(
-        "utils.api.resources.PreliminaryAssessmentResource.patch_preliminary_assessment"
-    )
     @patch(
         "utils.api.resources.PreliminaryAssessmentResource.get_preliminary_assessment"
     )
     def test_edit_preliminary_assessment_calls_api(
         self,
-        mock_patch_preliminary_assessment: Mock,
         mock_get_preliminary_assessment: Mock,
     ):
-        mock_patch_preliminary_assessment.return_value = True
         barrier_id = self.barrier["id"]
 
         url = reverse(
@@ -59,4 +54,5 @@ def test_edit_preliminary_assessment_calls_api(
             },
         )
 
+        mock_get_preliminary_assessment.assert_called_once_with(barrier_id=barrier_id)
         assert response.status_code == 200