S3 upload tests should not depend on identity of bucket

nationalarchives · Sep 4, 2023 · 9192e25 · 9192e25
1 parent f72f7a4
commit 9192e25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ds-caselaw-ingester/tests.py b/ds-caselaw-ingester/tests.py
@@ -269,23 +269,23 @@ def test_store_file_success(self, mock_print):
         session.upload_fileobj = MagicMock()
         lambda_function.store_file(None, "folder", "filename.ext", session)
         mock_print.assert_called_with("Upload Successful folder/filename.ext")
-        session.upload_fileobj.assert_called_with(None, None, "folder/filename.ext")
+        session.upload_fileobj.assert_called_with(None, ANY, "folder/filename.ext")
 
     @patch("builtins.print")
     def test_store_file_file_not_found(self, mock_print):
         session = boto3.Session
         session.upload_fileobj = MagicMock(side_effect=FileNotFoundError)
         lambda_function.store_file(None, "folder", "filename.ext", session)
         mock_print.assert_called_with("The file folder/filename.ext was not found")
-        session.upload_fileobj.assert_called_with(None, None, "folder/filename.ext")
+        session.upload_fileobj.assert_called_with(None, ANY, "folder/filename.ext")
 
     @patch("builtins.print")
     def test_store_file_file_no_credentials(self, mock_print):
         session = boto3.Session
         session.upload_fileobj = MagicMock(side_effect=NoCredentialsError)
         lambda_function.store_file(None, "folder", "filename.ext", session)
         mock_print.assert_called_with("Credentials not available")
-        session.upload_fileobj.assert_called_with(None, None, "folder/filename.ext")
+        session.upload_fileobj.assert_called_with(None, ANY, "folder/filename.ext")
 
     @patch.dict(
         os.environ,