Intermediate changes

commit_hash:acdbcb5ab09c7f6e8d5bf8a01ff1954c04d7a80f

1 files changed, 48 insertions, 0 deletions

diff --git a/library/python/monlib/ut/py3/test.py b/library/python/monlib/ut/py3/test.py
index bbb2b42bca..75cd0494cd 100644
--- a/library/python/monlib/ut/py3/test.py
+++ b/library/python/monlib/ut/py3/test.py
@@ -410,3 +410,51 @@ def test_reset_and_clear_registry():
     out = dumps(registry, format="json")
     j = json.loads(out)
     assert j == {}
+
+
+def test_mem_only_metrics():
+    registry = MetricRegistry()
+
+    registry.gauge({"some": "gauge"}, mem_only=True)
+    with pytest.raises(Exception):
+        registry.gauge({"some": "gauge"})
+
+    registry.int_gauge({"some": "int_gauge"}, mem_only=True)
+    with pytest.raises(Exception):
+        registry.int_gauge({"some": "int_gauge"})
+
+    registry.counter({"some": "counter"}, mem_only=True)
+    with pytest.raises(Exception):
+        registry.counter({"some": "counter"})
+
+    registry.rate({"some": "rate"}, mem_only=True)
+    with pytest.raises(Exception):
+        registry.rate({"some": "rate"})
+
+    registry.histogram_counter(
+        {"some": "histogram_counter"},
+        HistogramType.Explicit,
+        mem_only=True,
+        buckets=[1, 5, 15, 20, 25]
+    )
+    with pytest.raises(Exception):
+        registry.histogram_counter(
+            {"some": "histogram_counter"},
+            HistogramType.Explicit,
+            buckets=[1, 5, 15, 20, 25],
+        )
+
+    registry.histogram_rate(
+        {"some": "histogram_rate"},
+        HistogramType.Exponential,
+        mem_only=True,
+        bucket_count=5,
+        base=2
+    )
+    with pytest.raises(Exception):
+        registry.histogram_rate(
+            {"some": "histogram_rate"},
+            HistogramType.Exponential,
+            bucket_count=5,
+            base=2
+        )