@@ -41,7 +41,7 @@ def setUpClass(cls):
41
41
# pylint: disable=protected-access
42
42
metrics .set_meter_provider (MeterProvider ())
43
43
cls ._meter = metrics .get_meter (__name__ )
44
- cls ._labels = {"environment" : "staging" }
44
+ cls ._labels = {"environment" : "staging" , "number" : 321 }
45
45
cls ._key_labels = get_labels_as_key (cls ._labels )
46
46
47
47
def test_constructor (self ):
@@ -119,7 +119,7 @@ def test_export(self):
119
119
client = mock_client , host_name = host_name
120
120
)
121
121
test_metric = self ._meter .create_metric (
122
- "testname" , "testdesc" , "unit" , int , Counter , [ "environment" ]
122
+ "testname" , "testdesc" , "unit" , int , Counter , self . _labels . keys (),
123
123
)
124
124
record = MetricRecord (
125
125
test_metric , self ._key_labels , aggregate .SumAggregator (),
@@ -142,13 +142,21 @@ def test_export(self):
142
142
143
143
def test_translate_to_collector (self ):
144
144
test_metric = self ._meter .create_metric (
145
- "testname" , "testdesc" , "unit" , int , Counter , ["environment" ]
145
+ "testname" ,
146
+ "testdesc" ,
147
+ "unit" ,
148
+ int ,
149
+ Counter ,
150
+ ["environment" , "number" ],
146
151
)
147
152
aggregator = aggregate .SumAggregator ()
148
153
aggregator .update (123 )
149
154
aggregator .take_checkpoint ()
150
155
record = MetricRecord (test_metric , self ._key_labels , aggregator ,)
151
- output_metrics = metrics_exporter .translate_to_collector ([record ])
156
+ start_timestamp = Timestamp ()
157
+ output_metrics = metrics_exporter .translate_to_collector (
158
+ [record ], start_timestamp ,
159
+ )
152
160
self .assertEqual (len (output_metrics ), 1 )
153
161
self .assertIsInstance (output_metrics [0 ], metrics_pb2 .Metric )
154
162
self .assertEqual (output_metrics [0 ].metric_descriptor .name , "testname" )
@@ -161,14 +169,20 @@ def test_translate_to_collector(self):
161
169
metrics_pb2 .MetricDescriptor .CUMULATIVE_INT64 ,
162
170
)
163
171
self .assertEqual (
164
- len (output_metrics [0 ].metric_descriptor .label_keys ), 1
172
+ len (output_metrics [0 ].metric_descriptor .label_keys ), 2
165
173
)
166
174
self .assertEqual (
167
175
output_metrics [0 ].metric_descriptor .label_keys [0 ].key ,
168
176
"environment" ,
169
177
)
178
+ self .assertEqual (
179
+ output_metrics [0 ].metric_descriptor .label_keys [1 ].key , "number" ,
180
+ )
170
181
self .assertEqual (len (output_metrics [0 ].timeseries ), 1 )
171
- self .assertEqual (len (output_metrics [0 ].timeseries [0 ].label_values ), 1 )
182
+ self .assertEqual (len (output_metrics [0 ].timeseries [0 ].label_values ), 2 )
183
+ self .assertEqual (
184
+ output_metrics [0 ].timeseries [0 ].start_timestamp , start_timestamp
185
+ )
172
186
self .assertEqual (
173
187
output_metrics [0 ].timeseries [0 ].label_values [0 ].has_value , True
174
188
)
0 commit comments