def test_create_token_auth(mocked_auth_create):
mocked_auth_create.return_value = MagicMock()
secrets = secrets_provider.provide_secrets_via_token()
with auth(secrets) as _:
pass
mocked_auth_create.assert_called_once_with(secrets)
def test_create_service_principal_with_auth_error(mocked_auth_create):
secrets = secrets_provider.provide_secrets_via_service_principal()
inner_exception = MagicMock()
mocked_auth_create.side_effect = \
AuthenticationError("Auth error", inner_exception)
with pytest.raises(InterruptExecution) as _:
with auth(secrets) as _:
pass
def init_resource_graph_client(
experiment_secrets: Secrets) -> ResourceGraphClient:
secrets = load_secrets(experiment_secrets)
with auth(secrets) as authentication:
base_url = secrets.get('cloud').endpoints.resource_manager
client = ResourceGraphClient(credentials=authentication,
base_url=base_url)
return client
def init_client(
experiment_secrets: Secrets,
experiment_configuration: Configuration) -> ComputeManagementClient:
secrets = load_secrets(experiment_secrets)
configuration = load_configuration(experiment_configuration)
with auth(secrets) as authentication:
base_url = secrets.get('cloud').endpoints.resource_manager
client = ComputeManagementClient(
credentials=authentication,
subscription_id=configuration.get('subscription_id'),
base_url=base_url)
return client
def init_website_management_client(
experiment_secrets: Secrets,
experiment_configuration: Configuration) -> WebSiteManagementClient:
"""
Initializes Website management client for webapp resource under Azure
Resource manager.
"""
secrets = load_secrets(experiment_secrets)
configuration = load_configuration(experiment_configuration)
with auth(secrets) as authentication:
base_url = secrets.get('cloud').endpoints.resource_manager
client = WebSiteManagementClient(
credentials=authentication,
subscription_id=configuration.get('subscription_id'),
base_url=base_url)
return client
def init_compute_management_client(
experiment_secrets: Secrets,
experiment_configuration: Configuration) -> ComputeManagementClient:
"""
Initializes Compute management client for virtual machine,
and virtual machine scale sets resources under Azure Resource manager.
"""
secrets = load_secrets(experiment_secrets)
configuration = load_configuration(experiment_configuration)
with auth(secrets) as authentication:
base_url = secrets.get('cloud').endpoints.resource_manager
client = ComputeManagementClient(
credentials=authentication,
subscription_id=configuration.get('subscription_id'),
base_url=base_url)
return client
def test_violate_authentication_type():
secrets = secrets_provider.provide_violating_secrets()
with pytest.raises(InterruptExecution) as _:
with auth(secrets) as _:
pass
