Skip to content

Commit 1fed689

Browse files
committed
Follow API changes from prometheus client
1 parent db1f998 commit 1fed689

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

django_prometheus/testutils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ def getMetricFromFrozenRegistry(self, metric_name, frozen_registry,
4646
"""Gets a single metric from a frozen registry."""
4747
for metric in frozen_registry:
4848
for sample in metric.samples:
49-
if sample.name == metric_name and sample.labels == labels:
50-
return sample.value
49+
if sample[0] == metric_name and sample[1] == labels:
50+
return sample[2]
5151

5252
def getMetric(self, metric_name, registry=REGISTRY, **labels):
5353
"""Gets a single metric."""
@@ -59,8 +59,8 @@ def getMetricVectorFromFrozenRegistry(self, metric_name, frozen_registry):
5959
output = []
6060
for metric in frozen_registry:
6161
for sample in metric.samples:
62-
if sample.name == metric_name:
63-
output.append((sample.labels, sample.value))
62+
if sample[0] == metric_name:
63+
output.append((sample[1], sample[2]))
6464
return output
6565

6666
def getMetricVector(self, metric_name, registry=REGISTRY):

0 commit comments

Comments
 (0)