def __query_request_from(query, experiment_configuration: Configuration): configuration = load_configuration(experiment_configuration) arg_query_options = arg.models.QueryRequestOptions(result_format="table") result = QueryRequest(query=query, subscriptions=[configuration.get('subscription_id')], options=arg_query_options) return result
def test_load_subscription_from_credential_file(monkeypatch): # arrange experiment_configuration = None monkeypatch.setenv("AZURE_AUTH_LOCATION", os.path.join(settings_dir, 'credentials.json')) # act configuration = config.load_configuration(experiment_configuration) # assert assert configuration.get('subscription_id') == "AZURE_SUBSCRIPTION_ID"
def test_load_subscription_from_experiment_dict(): # arrange experiment_configuration = { "azure_subscription_id": "AZURE_SUBSCRIPTION_ID", "some_other_settings": "OTHER_SETTING" } # act configuration = config.load_configuration(experiment_configuration) # assert assert configuration.get('subscription_id') == "AZURE_SUBSCRIPTION_ID"
def test_load_legacy_subscription_from_experiment_dict(): # arrange experiment_configuration = { "azure": { "subscription_id": "AZURE_SUBSCRIPTION_ID" } } # act configuration = config.load_configuration(experiment_configuration) # assert assert configuration.get('subscription_id') == "AZURE_SUBSCRIPTION_ID"
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 __query_request_from(query, experiment_configuration: Configuration): configuration = load_configuration(experiment_configuration) result = QueryRequest(query=query, subscriptions=[configuration.get('subscription_id')]) return result