log_firehose II

Author: JamesW1-NHS

lambdas/ack_backend/tests/test_logging_decorators.py

@@ -9,7 +9,7 @@ def setUp(self):
         # Patch logger and firehose_client
         self.logger_patcher = patch("common.log_decorator.logger")
         self.mock_logger = self.logger_patcher.start()
-        self.firehose_patcher = patch("common.log_decorator.firehose_client")
+        self.firehose_patcher = patch("common.log_firehose.firehose_client")
         self.mock_firehose = self.firehose_patcher.start()
 
     def tearDown(self):

lambdas/redis_sync/tests/test_handler_decorator.py

@@ -40,7 +40,7 @@ def setUp(self):
         self.mock_get_s3_records = self.get_s3_records_patcher.start()
         self.record_processor_patcher = patch("redis_sync.process_record")
         self.mock_record_processor = self.record_processor_patcher.start()
-        self.firehose_patcher = patch("common.log_decorator.firehose_client")
+        self.firehose_patcher = patch("common.log_firehose.firehose_client")
         self.mock_firehose_client = self.firehose_patcher.start()
         self.mock_firehose_client.put_record.return_value = True