tidy

Author: nhsdevws

recordprocessor/tests/test_batch_processor.py

@@ -2,7 +2,6 @@
 import os
 from io import BytesIO
 from unittest.mock import patch
-from moto import mock_kinesis, mock_firehose
 from batch_processor import process_csv_to_fhir
 
 
@@ -11,8 +10,6 @@ def create_patch(target: str):
     return patcher.start()
 
 
-@mock_kinesis
-@mock_firehose
 class TestProcessCsvToFhir(unittest.TestCase):
 
     def setUp(self):
@@ -27,7 +24,6 @@ def setUp(self):
         self.mock_move_file = create_patch("file_level_validation.move_file")
         self.mock_get_permitted_operations = create_patch("file_level_validation.get_permitted_operations")
         self.mock_firehose_client = create_patch("logging_decorator.firehose_client")
-        self.mock_firehose_client.put_record.return_value = True
 
     def tearDown(self):
         patch.stopall()