18
18
import sys
19
19
import unittest
20
20
21
+
21
22
class TestMetrics (unittest .TestCase ):
22
23
def test_metrics (self ):
23
24
"""Test that metrics example produces expected values"""
@@ -39,36 +40,64 @@ def test_metrics(self):
39
40
output_data = json .loads (result .stdout )
40
41
41
42
# Get the metrics from the JSON structure
42
- metrics = output_data ["resource_metrics" ][0 ]["scope_metrics" ][0 ]["metrics" ]
43
+ metrics = output_data ["resource_metrics" ][0 ]["scope_metrics" ][0 ][
44
+ "metrics"
45
+ ]
43
46
44
47
# Create a lookup dict for easier testing
45
48
metrics_by_name = {metric ["name" ]: metric for metric in metrics }
46
49
47
50
# Test Counter: should be 1 (called counter.add(1))
48
- counter_value = metrics_by_name ["counter" ]["data" ]["data_points" ][0 ]["value" ]
51
+ counter_value = metrics_by_name ["counter" ]["data" ]["data_points" ][0 ][
52
+ "value"
53
+ ]
49
54
self .assertEqual (counter_value , 1 , "Counter should have value 1" )
50
55
51
56
# Test UpDownCounter: should be -4 (1 + (-5) = -4)
52
- updown_value = metrics_by_name ["updown_counter" ]["data" ]["data_points" ][0 ]["value" ]
53
- self .assertEqual (updown_value , - 4 , "UpDownCounter should have value -4" )
57
+ updown_value = metrics_by_name ["updown_counter" ]["data" ][
58
+ "data_points"
59
+ ][0 ]["value" ]
60
+ self .assertEqual (
61
+ updown_value , - 4 , "UpDownCounter should have value -4"
62
+ )
54
63
55
64
# Test Histogram: should have count=1, sum=99.9
56
65
histogram_data = metrics_by_name ["histogram" ]["data" ]["data_points" ][0 ]
57
- self .assertEqual (histogram_data ["count" ], 1 , "Histogram should have count 1" )
58
- self .assertEqual (histogram_data ["sum" ], 99.9 , "Histogram should have sum 99.9" )
66
+ self .assertEqual (
67
+ histogram_data ["count" ], 1 , "Histogram should have count 1"
68
+ )
69
+ self .assertEqual (
70
+ histogram_data ["sum" ], 99.9 , "Histogram should have sum 99.9"
71
+ )
59
72
60
73
# Test Gauge: should be 1 (last value set)
61
- gauge_value = metrics_by_name ["gauge" ]["data" ]["data_points" ][0 ]["value" ]
74
+ gauge_value = metrics_by_name ["gauge" ]["data" ]["data_points" ][0 ][
75
+ "value"
76
+ ]
62
77
self .assertEqual (gauge_value , 1 , "Gauge should have value 1" )
63
78
64
79
# Test Observable Counter: should be 1 (from callback)
65
- obs_counter_value = metrics_by_name ["observable_counter" ]["data" ]["data_points" ][0 ]["value" ]
66
- self .assertEqual (obs_counter_value , 1 , "Observable counter should have value 1" )
80
+ obs_counter_value = metrics_by_name ["observable_counter" ]["data" ][
81
+ "data_points"
82
+ ][0 ]["value" ]
83
+ self .assertEqual (
84
+ obs_counter_value , 1 , "Observable counter should have value 1"
85
+ )
67
86
68
87
# Test Observable UpDownCounter: should be -10 (from callback)
69
- obs_updown_value = metrics_by_name ["observable_updown_counter" ]["data" ]["data_points" ][0 ]["value" ]
70
- self .assertEqual (obs_updown_value , - 10 , "Observable updown counter should have value -10" )
88
+ obs_updown_value = metrics_by_name ["observable_updown_counter" ][
89
+ "data"
90
+ ]["data_points" ][0 ]["value" ]
91
+ self .assertEqual (
92
+ obs_updown_value ,
93
+ - 10 ,
94
+ "Observable updown counter should have value -10" ,
95
+ )
71
96
72
97
# Test Observable Gauge: should be 9 (from callback)
73
- obs_gauge_value = metrics_by_name ["observable_gauge" ]["data" ]["data_points" ][0 ]["value" ]
74
- self .assertEqual (obs_gauge_value , 9 , "Observable gauge should have value 9" )
98
+ obs_gauge_value = metrics_by_name ["observable_gauge" ]["data" ][
99
+ "data_points"
100
+ ][0 ]["value" ]
101
+ self .assertEqual (
102
+ obs_gauge_value , 9 , "Observable gauge should have value 9"
103
+ )
0 commit comments