def test_azure_init(self, azure_client):
result = AzureResult(container="bob", connection_string="conn")
assert result.value == None
assert result.connection_string == "conn"
assert result.connection_string_secret == None
assert azure_client.called is False
result.initialize_service()
assert azure_client.called is True
def test_azure_init_connection_string(self, azure_client):
result = AzureResult(container="bob", connection_string="con1")
result.initialize_service()
azure_client.assert_called_with(conn_str="con1")
with prefect.context({"secrets": {"test": "con2"}}):
result = AzureResult(container="bob", connection_string_secret="test")
result.initialize_service()
azure_client.assert_called_with(conn_str="con2")
def test_azure_init_without_connection_string(self, azure_client):
result = AzureResult(container="bob", connection_string=None)
with pytest.raises(ValueError):
result.initialize_service()
