mock exception log

Author: nhsdevws

delta_backend/src/delta.py

@@ -46,7 +46,9 @@ def handler(event, context):
     intrusion_check = True
     try:
         dynamodb = boto3.resource("dynamodb", region_name="eu-west-2")
+        logger.info("ref table...")
         delta_table = dynamodb.Table(delta_table_name)
+        logger.info("ref table done")
 
         # Converting ApproximateCreationDateTime directly to string will give Unix timestamp
         # I am converting it to isofformat for filtering purpose. This can be changed accordingly

delta_backend/tests/test_delta.py

@@ -19,6 +19,12 @@ class DeltaTestCase(unittest.TestCase):
     def setUp(self):
         # Common setup if needed
         self.context = {}
+        self.logger_exception_patch = patch("delta.logger.exception")
+        self.mock_logger_exception = self.logger_exception_patch.start()
+
+    def tearDown(self):
+        # Stop the patch after each test
+        self.logger_exception_patch.stop()
 
     @staticmethod
     def setup_mock_sqs(mock_boto_client, return_value={"ResponseMetadata": {"HTTPStatusCode": 200}}):
@@ -202,6 +208,7 @@ def test_handler_success_remove(self, mock_boto_resource, mock_send_log):
         # Assert
         self.assertEqual(result["statusCode"], 200)
 
+    # @TOD Check this
     @patch("delta.FirehoseLogger.send_log")  # Patch the method directly
     @patch("boto3.resource")
     @patch("boto3.client")
@@ -233,17 +240,21 @@ def test_handler_exception_intrusion(self, mock_boto_resource, mock_boto_client,
 
     @patch("delta.FirehoseLogger.send_log")
     @patch("boto3.resource")
-    @patch("delta.handler")
-    def test_handler_exception_intrusion_check_false(self, mock_boto_resource, mock_boto_client, mock_send_log):
+    @patch("boto3.client")
+    @patch("delta.boto3.resource")
+    def test_handler_exception_intrusion_check_false(self, mock_dynamodb, mock_boto_resource, mock_boto_client, mock_send_log):
         # Arrange
-        mock_send_log.return_value = None
         self.setUp_mock_resources(mock_boto_resource, mock_boto_client)
+    
+        # set up the event
         event = self.get_event()
         context = {}
 
         # Act & Assert
         with self.assertRaises(Exception):
             handler(event, context)
+            
+        self.mock_logger_exception.assert_called()
 
     @patch("delta.FirehoseLogger.send_log")  # Patch the method directly
     @patch("boto3.client")