Raise exception when logging to non-observable Enum (prometheus#501) (prometheus#520)

jxpp · web-flow · commit 72317c15d681 · 2020-02-29T15:12:36.000Z
Signed-off-by: jxpp &lt;jxpp@chigui.re&gt;
diff --git a/prometheus_client/metrics.py b/prometheus_client/metrics.py
@@ -656,6 +656,7 @@ def _metric_init(self):
 
     def state(self, state):
         """Set enum metric state."""
+        self._raise_if_not_observable()
         with self._lock:
             self._value = self._states.index(state)
 
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -428,6 +428,7 @@ def test_labels(self):
         self.assertEqual(0, self.registry.get_sample_value('el', {'l': 'a', 'el': 'a'}))
         self.assertEqual(0, self.registry.get_sample_value('el', {'l': 'a', 'el': 'b'}))
         self.assertEqual(1, self.registry.get_sample_value('el', {'l': 'a', 'el': 'c'}))
+        self.assertRaises(ValueError, self.labels.state, 'a')
 
     def test_overlapping_labels(self):
         with pytest.raises(ValueError):
