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