VED-457: improve coverage test

Author: Akol125

backend/tests/test_fhir_controller.py

@@ -1149,6 +1149,7 @@ def test_update_immunization_UnauthorizedVaxError(self, mock_get_supplier_permis
         response = self.controller.update_immunization(aws_event)
         mock_get_supplier_permissions.assert_called_once_with("Test")
         self.assertEqual(response["statusCode"], 403)
+
     @patch("fhir_controller.get_supplier_permissions")
     def test_update_immunization_UnauthorizedVaxError_check_for_non_batch(self, mock_get_supplier_permissions):
         """it should not update the Immunization record"""
@@ -1479,6 +1480,44 @@ def test_update_immunization_for_batch_with_invalid_json(self, mock_get_supplier
         self.assertEqual(response["statusCode"], 400)
         outcome = json.loads(response["body"])
         self.assertEqual(outcome["resourceType"], "OperationOutcome")
+    
+    @patch("fhir_controller.get_supplier_permissions")
+    def test_update_immunization_when_reinstated_true(self, mock_get_permissions):
+        """it should update reinstated Immunization"""
+        mock_get_permissions.return_value = ["COVID19.CRUD"]
+        imms_id = "valid-id"
+        imms = '{"id": "valid-id"}'
+        aws_event = {
+            "headers": {"E-Tag": 1, "SupplierSystem": "Test"},
+            "body": imms,
+            "pathParameters": {"id": imms_id},
+        }
+        self.service.update_reinstated_immunization.return_value = UpdateOutcome.UPDATE, {}, 3
+        self.service.get_immunization_by_id_all.return_value = {
+            "resource": "existing",
+            "Version": 1,
+            "DeletedAt": False,
+            "Reinstated": True,
+            "VaccineType": "COVID19",
+        }
+
+        response = self.controller.update_immunization(aws_event)
+
+        self.service.update_reinstated_immunization.assert_called_once_with(
+            imms_id, json.loads(imms), 1, ["COVID19.CRUD"], "Test"
+        )
+        self.assertEqual(response["statusCode"], 200)
+        self.assertEqual(json.loads(response["body"]), {"E-Tag": 3})
+
+    def test_update_immunization_missing_id(self):
+        """it should raise KeyError if pathParameters['id'] is missing"""
+        aws_event = {
+            "headers": {"E-Tag": 1, "SupplierSystem": "Test"},
+            "body": '{"id": "valid-id"}',
+            "pathParameters": {},  # 'id' is missing
+        }
+        with self.assertRaises(KeyError):
+            self.controller.update_immunization(aws_event)
 
 
 class TestDeleteImmunization(unittest.TestCase):