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