@@ -30,35 +30,37 @@ class TestMiddlewareMetrics(PrometheusTestCaseMixin, SimpleTestCase):
30
30
"""
31
31
32
32
def test_request_counters (self ):
33
- r = self .saveRegistry ()
33
+ registry = self .saveRegistry ()
34
34
self .client .get ("/" )
35
35
self .client .get ("/" )
36
36
self .client .get ("/help" )
37
37
self .client .post ("/" , {"test" : "data" })
38
38
39
- self .assertMetricDiff (r , 4 , M ("requests_before_middlewares_total" ))
40
- self .assertMetricDiff (r , 4 , M ("responses_before_middlewares_total" ))
41
- self .assertMetricDiff (r , 3 , T ("requests_total_by_method" ), method = "GET" )
42
- self .assertMetricDiff (r , 1 , T ("requests_total_by_method" ), method = "POST" )
43
- self .assertMetricDiff (r , 4 , T ("requests_total_by_transport" ), transport = "http" )
39
+ self .assertMetricDiff (registry , 4 , M ("requests_before_middlewares_total" ))
40
+ self .assertMetricDiff (registry , 4 , M ("responses_before_middlewares_total" ))
41
+ self .assertMetricDiff (registry , 3 , T ("requests_total_by_method" ), method = "GET" )
42
+ self .assertMetricDiff (registry , 1 , T ("requests_total_by_method" ), method = "POST" )
44
43
self .assertMetricDiff (
45
- r ,
44
+ registry , 4 , T ("requests_total_by_transport" ), transport = "http"
45
+ )
46
+ self .assertMetricDiff (
47
+ registry ,
46
48
2 ,
47
49
T ("requests_total_by_view_transport_method" ),
48
50
view = "testapp.views.index" ,
49
51
transport = "http" ,
50
52
method = "GET" ,
51
53
)
52
54
self .assertMetricDiff (
53
- r ,
55
+ registry ,
54
56
1 ,
55
57
T ("requests_total_by_view_transport_method" ),
56
58
view = "testapp.views.help" ,
57
59
transport = "http" ,
58
60
method = "GET" ,
59
61
)
60
62
self .assertMetricDiff (
61
- r ,
63
+ registry ,
62
64
1 ,
63
65
T ("requests_total_by_view_transport_method" ),
64
66
view = "testapp.views.index" ,
@@ -68,42 +70,54 @@ def test_request_counters(self):
68
70
# We have 3 requests with no post body, and one with a few
69
71
# bytes, but buckets are cumulative so that is 4 requests with
70
72
# <=128 bytes bodies.
71
- self .assertMetricDiff (r , 3 , M ("requests_body_total_bytes_bucket" ), le = "0.0" )
72
- self .assertMetricDiff (r , 4 , M ("requests_body_total_bytes_bucket" ), le = "128.0" )
73
+ self .assertMetricDiff (
74
+ registry , 3 , M ("requests_body_total_bytes_bucket" ), le = "0.0"
75
+ )
76
+ self .assertMetricDiff (
77
+ registry , 4 , M ("requests_body_total_bytes_bucket" ), le = "128.0"
78
+ )
73
79
self .assertMetricEquals (
74
80
None , M ("responses_total_by_templatename" ), templatename = "help.html"
75
81
)
76
82
self .assertMetricDiff (
77
- r , 3 , T ("responses_total_by_templatename" ), templatename = "index.html"
83
+ registry , 3 , T ("responses_total_by_templatename" ), templatename = "index.html"
84
+ )
85
+ self .assertMetricDiff (registry , 4 , T ("responses_total_by_status" ), status = "200" )
86
+ self .assertMetricDiff (
87
+ registry , 0 , M ("responses_body_total_bytes_bucket" ), le = "0.0"
78
88
)
79
- self .assertMetricDiff (r , 4 , T ("responses_total_by_status" ), status = "200" )
80
- self .assertMetricDiff (r , 0 , M ("responses_body_total_bytes_bucket" ), le = "0.0" )
81
- self .assertMetricDiff (r , 3 , M ("responses_body_total_bytes_bucket" ), le = "128.0" )
82
- self .assertMetricDiff (r , 4 , M ("responses_body_total_bytes_bucket" ), le = "8192.0" )
83
- self .assertMetricDiff (r , 4 , T ("responses_total_by_charset" ), charset = "utf-8" )
84
- self .assertMetricDiff (r , 0 , M ("responses_streaming_total" ))
89
+ self .assertMetricDiff (
90
+ registry , 3 , M ("responses_body_total_bytes_bucket" ), le = "128.0"
91
+ )
92
+ self .assertMetricDiff (
93
+ registry , 4 , M ("responses_body_total_bytes_bucket" ), le = "8192.0"
94
+ )
95
+ self .assertMetricDiff (
96
+ registry , 4 , T ("responses_total_by_charset" ), charset = "utf-8"
97
+ )
98
+ self .assertMetricDiff (registry , 0 , M ("responses_streaming_total" ))
85
99
86
100
def test_latency_histograms (self ):
87
101
# Caution: this test is timing-based. This is not ideal. It
88
102
# runs slowly (each request to /slow takes at least .1 seconds
89
103
# to complete) and it may be flaky when run on very slow
90
104
# systems.
91
105
92
- r = self .saveRegistry ()
106
+ registry = self .saveRegistry ()
93
107
94
108
# This always takes more than .1 second, so checking the lower
95
109
# buckets is fine.
96
110
self .client .get ("/slow" )
97
111
self .assertMetricDiff (
98
- r ,
112
+ registry ,
99
113
0 ,
100
114
M ("requests_latency_seconds_by_view_method_bucket" ),
101
115
le = "0.05" ,
102
116
view = "slow" ,
103
117
method = "GET" ,
104
118
)
105
119
self .assertMetricDiff (
106
- r ,
120
+ registry ,
107
121
1 ,
108
122
M ("requests_latency_seconds_by_view_method_bucket" ),
109
123
le = "5.0" ,
@@ -112,14 +126,14 @@ def test_latency_histograms(self):
112
126
)
113
127
114
128
def test_exception_latency_histograms (self ):
115
- r = self .saveRegistry ()
129
+ registry = self .saveRegistry ()
116
130
117
131
try :
118
132
self .client .get ("/objection" )
119
133
except ObjectionException :
120
134
pass
121
135
self .assertMetricDiff (
122
- r ,
136
+ registry ,
123
137
2 ,
124
138
M ("requests_latency_seconds_by_view_method_bucket" ),
125
139
le = "0.05" ,
@@ -128,8 +142,10 @@ def test_exception_latency_histograms(self):
128
142
)
129
143
130
144
def test_streaming_responses (self ):
131
- r = self .saveRegistry ()
145
+ registry = self .saveRegistry ()
132
146
self .client .get ("/" )
133
147
self .client .get ("/file" )
134
- self .assertMetricDiff (r , 1 , M ("responses_streaming_total" ))
135
- self .assertMetricDiff (r , 1 , M ("responses_body_total_bytes_bucket" ), le = "+Inf" )
148
+ self .assertMetricDiff (registry , 1 , M ("responses_streaming_total" ))
149
+ self .assertMetricDiff (
150
+ registry , 1 , M ("responses_body_total_bytes_bucket" ), le = "+Inf"
151
+ )
0 commit comments