reset counter - initial commit - testcases success

Justin Johnson · Justin Johnson · commit 51dccc06ac6b · 2024-01-26T14:18:37.000+05:30
diff --git a/.gitignore b/.gitignore
@@ -7,4 +7,5 @@ dist
 .coverage
 .tox
 .*cache
-htmlcov
+htmlcov
+*venv
diff --git a/prometheus_client/metrics.py b/prometheus_client/metrics.py
@@ -290,6 +290,11 @@ def _metric_init(self) -> None:
                                         self._labelvalues, self._documentation)
         self._created = time.time()
 
+    def reset(self) -> None:
+        """Reset the counter and the created time."""
+        self._value.set(value=0.0)
+        self._created = time.time()
+
     def inc(self, amount: float = 1, exemplar: Optional[Dict[str, str]] = None) -> None:
         """Increment counter by the given amount."""
         self._raise_if_not_observable()
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -42,6 +42,15 @@ def test_increment(self):
         self.assertEqual(1, self.registry.get_sample_value('c_total'))
         self.counter.inc(7)
         self.assertEqual(8, self.registry.get_sample_value('c_total'))
+    
+    def test_reset(self):
+        self.assertEqual(0, self.registry.get_sample_value('c_total'))
+        self.counter.inc()
+        self.assertEqual(1, self.registry.get_sample_value('c_total'))
+        self.counter.inc(5)
+        self.assertEqual(6, self.registry.get_sample_value('c_total'))
+        self.counter.reset()
+        self.assertEqual(0, self.registry.get_sample_value('c_total'))
 
     def test_repr(self):
         self.assertEqual(repr(self.counter), "prometheus_client.metrics.Counter(c)")

-Original file line number
+Diff line change
 .coverage
 .tox
 .*cache
 -htmlcov
 +htmlcov
 +*venv