def test_build_no_upload_if_file(monkeypatch):
storage = Azure(container="container", stored_as_script=True)
with pytest.raises(ValueError):
storage.build()
storage = Azure(container="container", stored_as_script=True, blob_name="flow.py")
assert storage == storage.build()
def test_upload_flow_to_azure_blob_name(monkeypatch):
client = MagicMock(upload_blob=MagicMock())
service = MagicMock(get_blob_client=MagicMock(return_value=client))
monkeypatch.setattr(
"prefect.environments.storage.Azure._azure_block_blob_service", service
)
storage = Azure(container="container", blob_name="name")
f = Flow("test")
assert storage.add_flow(f)
assert storage.build()
assert service.get_blob_client.call_args[1]["container"] == "container"
assert service.get_blob_client.call_args[1]["blob"] == "name"
def test_upload_flow_to_azure(monkeypatch):
client = MagicMock(upload_blob=MagicMock())
service = MagicMock(get_blob_client=MagicMock(return_value=client))
monkeypatch.setattr(
"prefect.environments.storage.Azure._azure_block_blob_service", service
)
storage = Azure(container="container")
f = Flow("test")
assert f.name not in storage
assert storage.add_flow(f)
assert storage.build()
assert client.upload_blob.called
assert f.name in storage
