21
21
22
22
from opentelemetry import metrics
23
23
from opentelemetry .ext .otcollector import metrics_exporter
24
- from opentelemetry .sdk .metrics import Counter , Measure , MeterProvider
24
+ from opentelemetry .sdk .metrics import (
25
+ Counter ,
26
+ Measure ,
27
+ MeterProvider ,
28
+ get_labels_as_key ,
29
+ )
25
30
from opentelemetry .sdk .metrics .export import (
26
31
MetricRecord ,
27
32
MetricsExportResult ,
@@ -36,8 +41,8 @@ def setUpClass(cls):
36
41
# pylint: disable=protected-access
37
42
metrics .set_meter_provider (MeterProvider ())
38
43
cls ._meter = metrics .get_meter (__name__ )
39
- kvp = {"environment" : "staging" }
40
- cls ._test_label_set = cls ._meter . get_label_set ( kvp )
44
+ cls . _labels = {"environment" : "staging" }
45
+ cls ._key_labels = get_labels_as_key ( cls ._labels )
41
46
42
47
def test_constructor (self ):
43
48
mock_get_node = mock .Mock ()
@@ -77,7 +82,6 @@ def test_get_collector_metric_type(self):
77
82
78
83
def test_get_collector_point (self ):
79
84
aggregator = aggregate .CounterAggregator ()
80
- label_set = self ._meter .get_label_set ({"environment" : "staging" })
81
85
int_counter = self ._meter .create_metric (
82
86
"testName" , "testDescription" , "unit" , int , Counter
83
87
)
@@ -88,21 +92,21 @@ def test_get_collector_point(self):
88
92
"testName" , "testDescription" , "unit" , float , Measure
89
93
)
90
94
result = metrics_exporter .get_collector_point (
91
- MetricRecord (aggregator , label_set , int_counter )
95
+ MetricRecord (aggregator , self . _key_labels , int_counter )
92
96
)
93
97
self .assertIsInstance (result , metrics_pb2 .Point )
94
98
self .assertIsInstance (result .timestamp , Timestamp )
95
99
self .assertEqual (result .int64_value , 0 )
96
100
aggregator .update (123.5 )
97
101
aggregator .take_checkpoint ()
98
102
result = metrics_exporter .get_collector_point (
99
- MetricRecord (aggregator , label_set , float_counter )
103
+ MetricRecord (aggregator , self . _key_labels , float_counter )
100
104
)
101
105
self .assertEqual (result .double_value , 123.5 )
102
106
self .assertRaises (
103
107
TypeError ,
104
108
metrics_exporter .get_collector_point (
105
- MetricRecord (aggregator , label_set , measure )
109
+ MetricRecord (aggregator , self . _key_labels , measure )
106
110
),
107
111
)
108
112
@@ -118,7 +122,7 @@ def test_export(self):
118
122
"testname" , "testdesc" , "unit" , int , Counter , ["environment" ]
119
123
)
120
124
record = MetricRecord (
121
- aggregate .CounterAggregator (), self ._test_label_set , test_metric
125
+ aggregate .CounterAggregator (), self ._key_labels , test_metric
122
126
)
123
127
124
128
result = collector_exporter .export ([record ])
@@ -137,14 +141,13 @@ def test_export(self):
137
141
)
138
142
139
143
def test_translate_to_collector (self ):
140
-
141
144
test_metric = self ._meter .create_metric (
142
145
"testname" , "testdesc" , "unit" , int , Counter , ["environment" ]
143
146
)
144
147
aggregator = aggregate .CounterAggregator ()
145
148
aggregator .update (123 )
146
149
aggregator .take_checkpoint ()
147
- record = MetricRecord (aggregator , self ._test_label_set , test_metric )
150
+ record = MetricRecord (aggregator , self ._key_labels , test_metric )
148
151
output_metrics = metrics_exporter .translate_to_collector ([record ])
149
152
self .assertEqual (len (output_metrics ), 1 )
150
153
self .assertIsInstance (output_metrics [0 ], metrics_pb2 .Metric )
@@ -175,12 +178,12 @@ def test_translate_to_collector(self):
175
178
self .assertEqual (len (output_metrics [0 ].timeseries [0 ].points ), 1 )
176
179
self .assertEqual (
177
180
output_metrics [0 ].timeseries [0 ].points [0 ].timestamp .seconds ,
178
- record .metric .bind (record . label_set ).last_update_timestamp
181
+ record .metric .bind (self . _labels ).last_update_timestamp
179
182
// 1000000000 ,
180
183
)
181
184
self .assertEqual (
182
185
output_metrics [0 ].timeseries [0 ].points [0 ].timestamp .nanos ,
183
- record .metric .bind (record . label_set ).last_update_timestamp
186
+ record .metric .bind (self . _labels ).last_update_timestamp
184
187
% 1000000000 ,
185
188
)
186
189
self .assertEqual (
0 commit comments