def test_blob_name_can_be_overwritten_at_runtime_by_download(self, monkeypatch):
task = GCSDownload(bucket="test", blob="blobber")
blob = MagicMock()
client = MagicMock()
client.return_value = MagicMock(
get_bucket=MagicMock(return_value=MagicMock(blob=blob))
)
monkeypatch.setattr("prefect.tasks.gcp.storage.get_storage_client", client)
task.run(blob="run-time", credentials={})
assert blob.call_args[0] == ("run-time",)
def test_download_initializes_attr_from_kwargs(self, attr):
task = GCSDownload(bucket="bucket", **{attr: "my-value"})
assert task.bucket == "bucket"
assert getattr(task, attr) == "my-value"