Skip to content

Commit ac1f786

Browse files
authored
Add assertMetricNotEquals (#385)
1 parent b299188 commit ac1f786

File tree

2 files changed

+16
-9
lines changed

2 files changed

+16
-9
lines changed

django_prometheus/tests/test_testutils.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -67,14 +67,10 @@ def testAssertMetricEquals(self):
6767
# First we test that a scalar metric can be tested.
6868
self.test_case.assertMetricEquals(42, "some_gauge", registry=self.registry)
6969
assert self.test_case.passes is True
70-
self.test_case.assertMetricEquals(43, "some_gauge", registry=self.registry)
71-
assert self.test_case.passes is False
72-
self.test_case.passes = True
70+
self.test_case.assertMetricNotEquals(43, "some_gauge", registry=self.registry)
7371

7472
# Here we test that assertMetricEquals fails on nonexistent gauges.
75-
self.test_case.assertMetricEquals(42, "some_nonexistent_gauge", registry=self.registry)
76-
assert not self.test_case.passes
77-
self.test_case.passes = True
73+
self.test_case.assertMetricNotEquals(42, "some_nonexistent_gauge", registry=self.registry)
7874

7975
# Here we test that labelled metrics can be tested.
8076
self.test_case.assertMetricEquals(
@@ -85,15 +81,13 @@ def testAssertMetricEquals(self):
8581
labelblue="indigo",
8682
)
8783
assert self.test_case.passes is True
88-
self.test_case.assertMetricEquals(
84+
self.test_case.assertMetricNotEquals(
8985
1,
9086
"some_labelled_gauge",
9187
registry=self.registry,
9288
labelred="tomato",
9389
labelblue="sky",
9490
)
95-
assert self.test_case.passes is False
96-
self.test_case.passes = True
9791

9892
def testRegistrySaving(self):
9993
"""Tests saveRegistry and frozen registries operations."""

django_prometheus/testutils.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,19 @@ def assertMetricEquals(self, expected_value, metric_name, registry=REGISTRY, **l
103103
)
104104
self.assertEqual(expected_value, value, assert_err)
105105

106+
def assertMetricNotEquals(self, expected_value, metric_name, registry=REGISTRY, **labels):
107+
"""Asserts that metric_name{**labels} == expected_value."""
108+
value = self.getMetric(metric_name, registry=registry, **labels)
109+
assert_err = METRIC_EQUALS_ERR_EXPLANATION % (
110+
metric_name,
111+
self.formatLabels(labels),
112+
value,
113+
expected_value,
114+
metric_name,
115+
self.formatVector(self.getMetricVector(metric_name)),
116+
)
117+
assert expected_value != value, assert_err
118+
106119
def assertMetricDiff(self, frozen_registry, expected_diff, metric_name, registry=REGISTRY, **labels):
107120
"""Asserts that metric_name{**labels} changed by expected_diff between
108121
the frozen registry and now. A frozen registry can be obtained

0 commit comments

Comments
 (0)