mock log info

Author: nhsdevws

delta_backend/tests/test_delta.py

@@ -227,7 +227,6 @@ def test_dps_record_skipped(self, mock_logger_info):
         context = {}
 
         response = handler(event, context)
-        print(f"final response1: {response}")
 
         self.assertEqual(response["statusCode"], 200)
         self.assertEqual(response["body"], "Record from DPS skipped for 12345")

delta_backend/tests/test_log_firehose.py

@@ -6,6 +6,15 @@
 
 class TestFirehoseLogger(unittest.TestCase):
 
+    def setUp(self):
+        # Common setup if needed
+        self.context = {}
+        self.logger_info_patcher = patch("logging.Logger.info")
+        self.mock_logger_info = self.logger_info_patcher.start()
+
+    def tearDown(self):
+        self.logger_info_patcher.stop()
+
     @patch("boto3.client")
     def test_send_log(self, mock_boto_client):
         """it should send log message to Firehose"""