diff --git a/tests/test_gcloud.py b/tests/test_gcloud.py index 14241efd3..42c83d337 100644 --- a/tests/test_gcloud.py +++ b/tests/test_gcloud.py @@ -11,6 +11,7 @@ from django.utils import timezone from google.cloud.exceptions import NotFound from google.cloud.storage.blob import Blob +from google.cloud.storage.retry import DEFAULT_RETRY from storages.backends import gcloud from tests.utils import NonSeekableContentFile @@ -97,6 +98,7 @@ def test_open_write(self, MockBlob): MockBlob().upload_from_file.assert_called_with( tmpfile, rewind=True, content_type=mimetypes.guess_type(self.filename)[0], + retry=DEFAULT_RETRY, predefined_acl='projectPrivate') def test_save(self): @@ -107,8 +109,13 @@ def test_save(self): self.storage._client.bucket.assert_called_with(self.bucket_name) self.storage._bucket.get_blob().upload_from_file.assert_called_with( - content, rewind=True, size=len(data), content_type=mimetypes.guess_type(self.filename)[0], - predefined_acl=None) + content, + rewind=True, + retry=DEFAULT_RETRY, + size=len(data), + content_type=mimetypes.guess_type(self.filename)[0], + predefined_acl=None + ) def test_save2(self): data = 'This is some test ủⓝï℅ⅆℇ content.' @@ -119,7 +126,7 @@ def test_save2(self): self.storage._client.bucket.assert_called_with(self.bucket_name) self.storage._bucket.get_blob().upload_from_file.assert_called_with( - content, rewind=True, size=len(data), content_type=mimetypes.guess_type(filename)[0], + content, rewind=True, retry=DEFAULT_RETRY, size=len(data), content_type=mimetypes.guess_type(filename)[0], predefined_acl=None) def test_save_with_default_acl(self): @@ -136,7 +143,7 @@ def test_save_with_default_acl(self): self.storage._client.bucket.assert_called_with(self.bucket_name) self.storage._bucket.get_blob().upload_from_file.assert_called_with( - content, rewind=True, size=len(data), content_type=mimetypes.guess_type(filename)[0], + content, rewind=True, retry=DEFAULT_RETRY, size=len(data), content_type=mimetypes.guess_type(filename)[0], predefined_acl='publicRead') def test_delete(self): @@ -420,6 +427,7 @@ def test_storage_save_gzipped(self): obj.upload_from_file.assert_called_with( mock.ANY, rewind=True, + retry=DEFAULT_RETRY, size=11, predefined_acl=None, content_type=None @@ -436,6 +444,7 @@ def test_storage_save_gzipped_non_seekable(self): obj.upload_from_file.assert_called_with( mock.ANY, rewind=True, + retry=DEFAULT_RETRY, size=11, predefined_acl=None, content_type=None @@ -455,6 +464,7 @@ def test_storage_save_gzip(self): obj.upload_from_file.assert_called_with( mock.ANY, rewind=True, + retry=DEFAULT_RETRY, size=None, predefined_acl=None, content_type='text/css', @@ -484,6 +494,7 @@ def test_storage_save_gzip_twice(self): obj.upload_from_file.assert_called_with( mock.ANY, rewind=True, + retry=DEFAULT_RETRY, size=None, predefined_acl=None, content_type='text/css',