Eerste versie

tests/test_prometheus_additions.py

@@ -0,0 +1,72 @@
+"""
+Pytest for prometheus client enhancements
+"""
+import os
+import logging
+import time
+import pytest
+from utils.prometheus_additions import CounterAbsolute
+import prometheus_client as prometheus
+
+logging.basicConfig(level=logging.DEBUG)
+
+TMP_DIR=os.path.join(
+    os.path.dirname(__file__),
+    'tmp_data'
+)
+DATA_DIR=os.path.join(
+    os.path.dirname(__file__),
+    'test_data'
+)
+
+@pytest.fixture(scope="class")
+def get_registry():
+    yield prometheus.REGISTRY
+    # reset prometheus registry between tests
+    collectors = list(prometheus.REGISTRY._collector_to_names.keys())
+    for collector in collectors:
+        prometheus.REGISTRY.unregister(collector)
+
+old_creation_time = 0.0
+
+class TestCounterWithReset:
+    a_counter_absolute = CounterAbsolute('Absolute_Counter', 'Test metric' )
+    old_creation_time = 0.0
+    param_test_data_sets = [
+        (10, False),
+        (10, True),
+        (11, True),
+        (110, True),
+        (110, True),
+        (210, True),
+        (310.7, True),
+        (110, False),
+        (210, True),
+        (310.7, True),
+    ]
+
+    @pytest.mark.parametrize("value, same_creation_time", param_test_data_sets)
+    def test_counter_absolute(self, request, get_registry, value, same_creation_time):
+        global old_creation_time
+        self.a_counter_absolute.set(value)
+        creation_time = self.a_counter_absolute._created
+        logging.info(f"Creation time: {creation_time:e}")
+        registry = get_registry
+        registry.collect()
+        prometheus.write_to_textfile(os.path.join(TMP_DIR, f"absolute_counter_{request.node.callspec.id}_{value:05}.txt"), prometheus.REGISTRY)
+
+        assert self.a_counter_absolute._value.get() == value
+        assert (creation_time == old_creation_time ) == same_creation_time
+        old_creation_time = creation_time
+        time.sleep(0.005)
+
+
+class TestCounterRestForbidden:
+    a_counter_absolute = CounterAbsolute('Strict_Absolute_Counter', "This Counters doesn't allow reset")
+
+    def test_counter_reset(self):
+        val_first = 0.3324234
+        val_second = 0.3324233
+        self.a_counter_absolute.set(val_first, fail_on_decrease=True)
+        with pytest.raises(ValueError, match=rf"Counter must increase {val_second} lower {val_first}"):
+            self.a_counter_absolute.set(val_second, fail_on_decrease=True)