def test_fail_on_missing_auth(self):
""" Raise an exception when loading settings for GoogleCloudStorage
and no authentication information is found
"""
settings = {"storage.bucket": "new_bucket", "storage.region_name": "us-east-1"}
with self.assertRaises(Exception):
GoogleCloudStorage.configure(settings)
def test_fail_on_missing_auth(self):
""" Raise an exception when loading settings for GoogleCloudStorage
and no authentication information is found
"""
settings = {
"storage.bucket": "new_bucket",
"storage.region_name": "us-east-1"
}
with self.assertRaises(Exception):
GoogleCloudStorage.configure(settings)
def test_client_without_credentials(self):
"""Can create a client without passing in application credentials"""
kwargs = GoogleCloudStorage.configure({
"storage.bucket":
"new_bucket",
"storage.region_name":
"us-east-1"
})
GoogleCloudStorage(MagicMock(), **kwargs)
def test_storage_class(self):
"""Can specify a storage class for GCS objects"""
settings = dict(self.settings)
settings["storage.storage_class"] = "COLDLINE"
kwargs = GoogleCloudStorage.configure(settings)
storage = GoogleCloudStorage(MagicMock(), **kwargs)
package = make_package()
storage.upload(package, BytesIO())
blob = self.bucket.list_blobs()[0]
blob.update_storage_class.assert_called_with("COLDLINE")
def test_storage_class(self):
""" Can specify a storage class for GCS objects """
settings = dict(self.settings)
settings["storage.storage_class"] = "COLDLINE"
kwargs = GoogleCloudStorage.configure(settings)
storage = GoogleCloudStorage(MagicMock(), **kwargs)
package = make_package()
storage.upload(package, BytesIO())
blob = self.bucket.list_blobs()[0]
blob.update_storage_class.assert_called_with("COLDLINE")
def test_create_bucket(self):
""" If GCS bucket doesn't exist, create it """
settings = {
"storage.bucket": "new_bucket",
"storage.region_name": "us-east-1",
"storage.gcp_service_account_json_filename": "my-filename.json",
}
storage = GoogleCloudStorage.configure(settings)
self.gcs.bucket.assert_called_with("new_bucket")
bucket = self.gcs.bucket("new_bucket")
bucket.create.assert_called_once_with()
def test_create_bucket(self):
""" If GCS bucket doesn't exist, create it """
settings = {
"storage.bucket": "new_bucket",
"storage.region_name": "us-east-1",
"storage.gcp_service_account_json_filename": "my-filename.json",
}
storage = GoogleCloudStorage.configure(settings)
self.gcs.bucket.assert_called_with("new_bucket")
bucket = self.gcs.bucket("new_bucket")
bucket.create.assert_called_once_with()
def test_object_acl(self):
"""Can specify an object ACL for GCS objects. Just test to make
sure that the configured ACL is forwarded to the API client
"""
settings = dict(self.settings)
settings["storage.object_acl"] = "authenticated-read"
kwargs = GoogleCloudStorage.configure(settings)
storage = GoogleCloudStorage(MagicMock(), **kwargs)
package = make_package()
storage.upload(package, BytesIO())
blob = self.bucket.list_blobs()[0]
self.assertEqual(blob._acl, "authenticated-read")
def test_create_bucket(self):
"""If GCS bucket doesn't exist, create it"""
settings = {
"storage.bucket": "new_bucket",
"storage.region_name": "us-east-1",
"storage.gcp_service_account_json_filename": self._config_file,
}
arguments = GoogleCloudStorage.configure(settings)
arguments["bucket_factory"]()
self.gcs.bucket.assert_called_with("new_bucket")
bucket = self.gcs.bucket("new_bucket")
bucket.create.assert_called_once_with()
def setUp(self):
super(TestGoogleCloudStorage, self).setUp()
self.gcs = MockGCSClient()
patch("google.cloud.storage.Client", self.gcs).start()
self.settings = {
"storage.bucket": "mybucket",
"storage.gcp_service_account_json_filename": "my-filename.json",
}
self.bucket = self.gcs.bucket("mybucket")
self.bucket._created = True
patch.object(GoogleCloudStorage, "test", True).start()
kwargs = GoogleCloudStorage.configure(self.settings)
self.storage = GoogleCloudStorage(MagicMock(), **kwargs)
def test_object_acl(self):
""" Can specify an object ACL for GCS objects. Just test to make
sure that the configured ACL is forwarded to the API client
"""
settings = dict(self.settings)
settings["storage.object_acl"] = "authenticated-read"
kwargs = GoogleCloudStorage.configure(settings)
storage = GoogleCloudStorage(MagicMock(), **kwargs)
package = make_package()
storage.upload(package, BytesIO())
blob = self.bucket.list_blobs()[0]
self.assertEqual(blob._acl, "authenticated-read")
def setUp(self):
super(TestGoogleCloudStorage, self).setUp()
self.gcs = MockGCSClient()
patch("google.cloud.storage.Client", self.gcs).start()
self.settings = {
"storage.bucket": "mybucket",
"storage.gcp_service_account_json_filename": "my-filename.json",
}
self.bucket = self.gcs.bucket("mybucket")
self.bucket._created = True
patch.object(GoogleCloudStorage, "test", True).start()
kwargs = GoogleCloudStorage.configure(self.settings)
self.storage = GoogleCloudStorage(MagicMock(), **kwargs)
def setUp(self):
super(TestGoogleCloudStorage, self).setUp()
self.gcs = MockGCSClient()
self._config_file = tempfile.mktemp()
with open(self._config_file, "w") as ofile:
json.dump({}, ofile)
patch("google.cloud.storage.Client", self.gcs).start()
self.settings = {
"storage.bucket": "mybucket",
"storage.gcp_service_account_json_filename": self._config_file,
}
self.bucket = self.gcs.bucket("mybucket")
self.bucket._created = True
patch.object(GoogleCloudStorage, "test", True).start()
kwargs = GoogleCloudStorage.configure(self.settings)
self.storage = GoogleCloudStorage(MagicMock(), **kwargs)