Prometheus system tests (#299)

Author: vcasadei

README.rst

@@ -600,4 +600,4 @@ Apache 2.0 - See `LICENSE <LICENSE>`__ for more information.
 Disclaimer
 ----------
 
-This is not an official Google product.
+This is not an official Google product.

opencensus/stats/aggregation_data.py

@@ -230,7 +230,6 @@ def increment_bucket_count(self, value):
             return i
 
         self._counts_per_bucket[(len(self._bounds))-1] += 1
-
         return i
 
 

opencensus/stats/exporters/prometheus_exporter.py

@@ -237,13 +237,13 @@ def describe(self):
         registered = {}
         for sign in self.registered_views:
             registered[sign] = self.registered_views[sign]
-
         for v_data in list(self.view_data):  # pragma: NO COVER
-            signature = view_signature(self.options.namespace, v_data.view)
-            desc = self.registered_views[signature]
-            metric = self.to_metric(desc,
-                                    self.view_data[v_data].view)
-            yield metric
+            if not isinstance(v_data, str):
+                signature = view_signature(self.options.namespace, v_data.view)
+                desc = self.registered_views[signature]
+                metric = self.to_metric(desc,
+                                        self.view_data[v_data].view)
+                yield metric
 
 
 class PrometheusStatsExporter(base.StatsExporter):

tests/system/stats/prometheus/prometheus_stats_test.py

@@ -0,0 +1,72 @@
+# Copyright 2018, OpenCensus Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+
+class TestPrometheusStats(unittest.TestCase):
+
+    def test_prometheus_stats(self):
+        import random
+        import time
+        import sys
+
+        from opencensus.stats import aggregation as aggregation_module
+        from opencensus.stats.exporters import prometheus_exporter as prometheus
+        from opencensus.stats import measure as measure_module
+        from opencensus.stats import stats as stats_module
+        from opencensus.stats import view as view_module
+        from opencensus.tags import tag_key as tag_key_module
+        from opencensus.tags import tag_map as tag_map_module
+        from opencensus.tags import tag_value as tag_value_module
+
+        MiB = 1 << 20
+        FRONTEND_KEY = tag_key_module.TagKey("my.org/keys/frontend")
+        VIDEO_SIZE_MEASURE = measure_module.MeasureInt(
+            "my.org/measures/video_size", "size of processed videos", "By")
+        VIDEO_SIZE_VIEW_NAME = "my.org/views/video_size"
+        VIDEO_SIZE_DISTRIBUTION = aggregation_module.CountAggregation(
+            256.0 * MiB)
+        VIDEO_SIZE_VIEW = view_module.View(VIDEO_SIZE_VIEW_NAME,
+                                           "processed video size over time",
+                                           [FRONTEND_KEY],
+                                           VIDEO_SIZE_MEASURE,
+                                           VIDEO_SIZE_DISTRIBUTION)
+        stats = stats_module.Stats()
+        view_manager = stats.view_manager
+        stats_recorder = stats.stats_recorder
+
+        exporter = prometheus.new_stats_exporter(prometheus.Options(namespace="opencensus", port=9303))
+        view_manager.register_exporter(exporter)
+
+        view_manager.register_view(VIDEO_SIZE_VIEW)
+
+        time.sleep(random.randint(1, 10) / 1000.0)
+
+        tag_value = tag_value_module.TagValue(str(random.randint(1, 10000)))
+        tag_map = tag_map_module.TagMap()
+        tag_map.insert(FRONTEND_KEY, tag_value)
+        measure_map = stats_recorder.new_measurement_map()
+        measure_map.measure_int_put(VIDEO_SIZE_MEASURE, 25 * MiB)
+        measure_map.record(tag_map)
+
+        if sys.version_info > (3, 0):
+            import urllib.request
+            contents = urllib.request.urlopen("http://localhost:9303/metrics").read()
+        else:
+            import urllib2
+            contents = urllib2.urlopen("http://localhost:9303/metrics").read()
+
+        self.assertIn(b'# TYPE opencensus_my.org/views/video_size counter', contents)
+        self.assertIn(b'opencensus_my.org/views/video_size 268435456.0', contents)