Skip to content

Commit 2319282

Browse files
authored
Merge pull request #149 from asherf/pytest
Avoid single letter variables.
2 parents 5ec9aba + ea73624 commit 2319282

File tree

4 files changed

+66
-46
lines changed

4 files changed

+66
-46
lines changed

django_prometheus/tests/end2end/testapp/test_db.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -64,14 +64,18 @@ def testCounters(self):
6464
)
6565

6666
def testExecuteMany(self):
67-
r = self.saveRegistry()
67+
registry = self.saveRegistry()
6868
cursor_db1 = connections["test_db_1"].cursor()
6969
cursor_db1.executemany(
7070
"INSERT INTO testapp_lawn(location) VALUES (?)",
7171
[("Paris",), ("New York",), ("Berlin",), ("San Francisco",)],
7272
)
7373
self.assertMetricDiff(
74-
r, 4, "django_db_execute_many_total", alias="test_db_1", vendor="sqlite"
74+
registry,
75+
4,
76+
"django_db_execute_many_total",
77+
alias="test_db_1",
78+
vendor="sqlite",
7579
)
7680

7781

@@ -89,14 +93,14 @@ class TestPostgresDbMetrics(BaseDbMetricTest):
8993
"""
9094

9195
def testCounters(self):
92-
r = self.saveRegistry()
96+
registry = self.saveRegistry()
9397
cursor = connections["postgresql"].cursor()
9498

9599
for _ in range(20):
96100
cursor.execute("SELECT 1")
97101

98102
self.assertMetricCompare(
99-
r,
103+
registry,
100104
lambda a, b: a + 20 <= b < a + 25,
101105
"django_db_execute_total",
102106
alias="postgresql",
@@ -116,14 +120,14 @@ class TestMysDbMetrics(BaseDbMetricTest):
116120
"""
117121

118122
def testCounters(self):
119-
r = self.saveRegistry()
123+
registry = self.saveRegistry()
120124
cursor = connections["mysql"].cursor()
121125

122126
for _ in range(20):
123127
cursor.execute("SELECT 1")
124128

125129
self.assertMetricCompare(
126-
r,
130+
registry,
127131
lambda a, b: a + 20 <= b < a + 25,
128132
"django_db_execute_total",
129133
alias="mysql",

django_prometheus/tests/end2end/testapp/test_middleware.py

Lines changed: 42 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -30,35 +30,37 @@ class TestMiddlewareMetrics(PrometheusTestCaseMixin, SimpleTestCase):
3030
"""
3131

3232
def test_request_counters(self):
33-
r = self.saveRegistry()
33+
registry = self.saveRegistry()
3434
self.client.get("/")
3535
self.client.get("/")
3636
self.client.get("/help")
3737
self.client.post("/", {"test": "data"})
3838

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")
4443
self.assertMetricDiff(
45-
r,
44+
registry, 4, T("requests_total_by_transport"), transport="http"
45+
)
46+
self.assertMetricDiff(
47+
registry,
4648
2,
4749
T("requests_total_by_view_transport_method"),
4850
view="testapp.views.index",
4951
transport="http",
5052
method="GET",
5153
)
5254
self.assertMetricDiff(
53-
r,
55+
registry,
5456
1,
5557
T("requests_total_by_view_transport_method"),
5658
view="testapp.views.help",
5759
transport="http",
5860
method="GET",
5961
)
6062
self.assertMetricDiff(
61-
r,
63+
registry,
6264
1,
6365
T("requests_total_by_view_transport_method"),
6466
view="testapp.views.index",
@@ -68,42 +70,54 @@ def test_request_counters(self):
6870
# We have 3 requests with no post body, and one with a few
6971
# bytes, but buckets are cumulative so that is 4 requests with
7072
# <=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+
)
7379
self.assertMetricEquals(
7480
None, M("responses_total_by_templatename"), templatename="help.html"
7581
)
7682
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"
7888
)
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"))
8599

86100
def test_latency_histograms(self):
87101
# Caution: this test is timing-based. This is not ideal. It
88102
# runs slowly (each request to /slow takes at least .1 seconds
89103
# to complete) and it may be flaky when run on very slow
90104
# systems.
91105

92-
r = self.saveRegistry()
106+
registry = self.saveRegistry()
93107

94108
# This always takes more than .1 second, so checking the lower
95109
# buckets is fine.
96110
self.client.get("/slow")
97111
self.assertMetricDiff(
98-
r,
112+
registry,
99113
0,
100114
M("requests_latency_seconds_by_view_method_bucket"),
101115
le="0.05",
102116
view="slow",
103117
method="GET",
104118
)
105119
self.assertMetricDiff(
106-
r,
120+
registry,
107121
1,
108122
M("requests_latency_seconds_by_view_method_bucket"),
109123
le="5.0",
@@ -112,14 +126,14 @@ def test_latency_histograms(self):
112126
)
113127

114128
def test_exception_latency_histograms(self):
115-
r = self.saveRegistry()
129+
registry = self.saveRegistry()
116130

117131
try:
118132
self.client.get("/objection")
119133
except ObjectionException:
120134
pass
121135
self.assertMetricDiff(
122-
r,
136+
registry,
123137
2,
124138
M("requests_latency_seconds_by_view_method_bucket"),
125139
le="0.05",
@@ -128,8 +142,10 @@ def test_exception_latency_histograms(self):
128142
)
129143

130144
def test_streaming_responses(self):
131-
r = self.saveRegistry()
145+
registry = self.saveRegistry()
132146
self.client.get("/")
133147
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+
)

django_prometheus/tests/end2end/testapp/test_models.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -18,35 +18,35 @@ class TestModelMetrics(PrometheusTestCaseMixin, TestCase):
1818
"""Test django_prometheus.models."""
1919

2020
def test_counters(self):
21-
r = self.saveRegistry()
21+
registry = self.saveRegistry()
2222
cool = Dog()
2323
cool.name = "Cool"
2424
cool.save()
25-
self.assertMetricDiff(r, 1, M("inserts_total"), model="dog")
25+
self.assertMetricDiff(registry, 1, M("inserts_total"), model="dog")
2626

2727
elysees = Lawn()
2828
elysees.location = "Champs Elysees, Paris"
2929
elysees.save()
30-
self.assertMetricDiff(r, 1, M("inserts_total"), model="lawn")
31-
self.assertMetricDiff(r, 1, M("inserts_total"), model="dog")
30+
self.assertMetricDiff(registry, 1, M("inserts_total"), model="lawn")
31+
self.assertMetricDiff(registry, 1, M("inserts_total"), model="dog")
3232

3333
galli = Dog()
3434
galli.name = "Galli"
3535
galli.save()
36-
self.assertMetricDiff(r, 2, M("inserts_total"), model="dog")
36+
self.assertMetricDiff(registry, 2, M("inserts_total"), model="dog")
3737

3838
cool.breed = "Wolfhound"
39-
self.assertMetricDiff(r, 2, M("inserts_total"), model="dog")
39+
self.assertMetricDiff(registry, 2, M("inserts_total"), model="dog")
4040

4141
cool.save()
42-
self.assertMetricDiff(r, 2, M("inserts_total"), model="dog")
43-
self.assertMetricDiff(r, 1, M("updates_total"), model="dog")
42+
self.assertMetricDiff(registry, 2, M("inserts_total"), model="dog")
43+
self.assertMetricDiff(registry, 1, M("updates_total"), model="dog")
4444

4545
cool.age = 9
4646
cool.save()
47-
self.assertMetricDiff(r, 2, M("updates_total"), model="dog")
47+
self.assertMetricDiff(registry, 2, M("updates_total"), model="dog")
4848

4949
cool.delete() # :(
50-
self.assertMetricDiff(r, 2, M("inserts_total"), model="dog")
51-
self.assertMetricDiff(r, 2, M("updates_total"), model="dog")
52-
self.assertMetricDiff(r, 1, M("deletes_total"), model="dog")
50+
self.assertMetricDiff(registry, 2, M("inserts_total"), model="dog")
51+
self.assertMetricDiff(registry, 2, M("updates_total"), model="dog")
52+
self.assertMetricDiff(registry, 1, M("deletes_total"), model="dog")

django_prometheus/testutils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ def saveRegistry(self, registry=REGISTRY):
3535
"""Freezes a registry. This lets a user test changes to a metric
3636
instead of testing the absolute value. A typical use case looks like:
3737
38-
r = self.saveRegistry()
38+
registry = self.saveRegistry()
3939
doStuff()
40-
self.assertMetricDiff(r, 1, 'stuff_done_total')
40+
self.assertMetricDiff(registry, 1, 'stuff_done_total')
4141
"""
4242
return copy.deepcopy(list(registry.collect()))
4343

0 commit comments

Comments
 (0)