def test_azure_service_init_uses_custom_secrets(self, azure_service): handler = AzureResultHandler(container="bob", azure_credentials_secret="MY_FOO") with prefect.context(secrets=dict( MY_FOO=dict(ACCOUNT_NAME=1, ACCOUNT_KEY=999))): with set_temporary_config({"cloud.use_local_secrets": True}): handler.initialize_service() assert handler.container == "bob" assert azure_service.call_args[1] == { "account_name": 1, "account_key": 999 }
def test_azure_service_init_uses_secrets(self, azure_service): handler = AzureResultHandler(container="bob") assert handler.container == "bob" assert azure_service.called is False with prefect.context(secrets=dict( AZ_CREDENTIALS=dict(ACCOUNT_NAME="1", ACCOUNT_KEY="42"))): with set_temporary_config({"cloud.use_local_secrets": True}): handler.initialize_service() assert azure_service.call_args[1] == { "account_name": "1", "account_key": "42" }
def test_azure_service_init_uses_connection_string_over_secret( self, azure_service): handler = AzureResultHandler(container="bob", azure_credentials_secret="MY_FOO", connection_string="TEST") with prefect.context(secrets=dict( MY_FOO=dict(ACCOUNT_NAME=1, ACCOUNT_KEY=999))): with set_temporary_config({"cloud.use_local_secrets": True}): handler.initialize_service() assert handler.container == "bob" assert azure_service.call_args[1] == { "connection_string": "TEST", }