Bucket=None

Author: JamesW1-NHS

lambdas/recordprocessor/tests/test_recordprocessor_edge_cases.py

@@ -5,7 +5,6 @@
 
 from batch_processor import process_csv_to_fhir
 from utils_for_recordprocessor_tests.utils_for_recordprocessor_tests import (
-    BucketNames,
     create_patch,
 )
 
@@ -82,11 +81,11 @@ def test_process_large_file_cp1252(self):
         self.mock_logger_warning.assert_called()
         warning_call_args = self.mock_logger_warning.call_args[0][0]
         self.assertTrue(warning_call_args.startswith("Encoding Error: 'utf-8' codec can't decode byte 0xe9"))
-        # TODO: when running all tests this expects Bucket=None. not clear why.
+        # TODO: when running standalone this expects BucketNames.SOURCE. not clear why.
         mock_s3.get_object.assert_has_calls(
             [
-                call(Bucket=BucketNames.SOURCE, Key="test-filename"),
-                call(Bucket=BucketNames.SOURCE, Key="processing/test-filename"),
+                call(Bucket=None, Key="test-filename"),
+                call(Bucket=None, Key="processing/test-filename"),
             ]
         )