1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE.txt in the project root for
# license information.
# -------------------------------------------------------------------------
from datetime import timedelta
from unittest import mock
from azure.monitor.query import MetricsQueryClient, MetricAggregationType, Metric
from azure.monitor.query._version import VERSION
from base_testcase import MetricsQueryClientTestCase
METRIC_NAME = "requests/count"
METRIC_RESOURCE_PROVIDER = "Microsoft.Insights/components"
class TestMetricsQueryClient(MetricsQueryClientTestCase):
def test_metrics_auth(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.query_resource(
monitor_info["metrics_resource_id"],
metric_names=[METRIC_NAME],
timespan=timedelta(days=1),
aggregations=[MetricAggregationType.COUNT],
)
assert response
assert response.metrics
def test_metrics_granularity(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.query_resource(
monitor_info["metrics_resource_id"],
metric_names=[METRIC_NAME],
timespan=timedelta(days=1),
granularity=timedelta(minutes=5),
aggregations=[MetricAggregationType.COUNT],
)
assert response
assert response.granularity == timedelta(minutes=5)
metric = response.metrics[METRIC_NAME]
assert metric.timeseries
for t in metric.timeseries:
assert t.metadata_values is not None
def test_metrics_filter(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.query_resource(
monitor_info["metrics_resource_id"],
metric_names=[METRIC_NAME],
timespan=timedelta(days=1),
granularity=timedelta(minutes=5),
filter="request/success eq '0'",
aggregations=[MetricAggregationType.COUNT],
)
assert response
metric = response.metrics[METRIC_NAME]
for t in metric.timeseries:
assert t.metadata_values is not None
def test_metrics_list(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.query_resource(
monitor_info["metrics_resource_id"],
metric_names=[METRIC_NAME],
timespan=timedelta(days=1),
granularity=timedelta(minutes=5),
aggregations=[MetricAggregationType.COUNT],
)
assert response
metrics = response.metrics
assert len(metrics) == 1
assert metrics[0].__class__ == Metric
assert metrics[METRIC_NAME].__class__ == Metric
assert metrics[METRIC_NAME] == metrics[0]
def test_metrics_list_with_commas(self):
"""Commas in metric names should be encoded as %2."""
with mock.patch("azure.monitor.query._generated.metrics.operations.MetricsOperations.list") as mock_list:
mock_list.return_value = {"foo": "bar"}
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
client.query_resource(
"resource",
metric_names=["metric1,metric2", "foo,test,test"],
timespan=timedelta(days=1),
granularity=timedelta(minutes=5),
aggregations=[MetricAggregationType.COUNT],
)
assert "metricnames" in mock_list.call_args[1]
assert mock_list.call_args[1]["metricnames"] == "metric1%2metric2,foo%2test%2test"
def test_metrics_namespaces(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.list_metric_namespaces(monitor_info["metrics_resource_id"])
assert response is not None
for item in response:
assert item
def test_metrics_definitions(self, recorded_test, monitor_info):
client = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
response = client.list_metric_definitions(
monitor_info["metrics_resource_id"], namespace=METRIC_RESOURCE_PROVIDER
)
assert response is not None
for item in response:
assert item
def test_client_different_endpoint(self):
credential = self.get_credential(MetricsQueryClient)
endpoint = "https://management.chinacloudapi.cn"
client = MetricsQueryClient(credential, endpoint=endpoint)
assert client._endpoint == endpoint
assert f"{endpoint}/.default" in client._client._config.authentication_policy._scopes
def test_client_user_agent(self):
client: MetricsQueryClient = self.get_client(MetricsQueryClient, self.get_credential(MetricsQueryClient))
assert f"monitor-query/{VERSION}" in client._client._config.user_agent_policy.user_agent
|
