Fix GCS tests after adding chunk_size and timeout

rafsaf · Jun 18, 2023 · 1d276cc · 1d276cc
1 parent cdb025a
commit 1d276cc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_storage_provider_gcs.py b/tests/test_storage_provider_gcs.py
@@ -82,9 +82,12 @@ def test_gcs_post_save_with_google_bucket_upload_path(
         == "test123/fake_env_name/fake_backup.zip"
     )
     assert fake_backup_file_zip_path.exists()
-    bucket_mock.blob.assert_called_once_with("test123/fake_env_name/fake_backup.zip")
+    bucket_mock.blob.assert_called_once_with(
+        "test123/fake_env_name/fake_backup.zip",
+        chunk_size=GoogleCloudStorage.CHUNK_SIZE,
+    )
     single_blob_mock.upload_from_filename.assert_called_once_with(
-        fake_backup_file_zip_path
+        fake_backup_file_zip_path, timeout=120
     )