From 199639fd81e12389c7bbd63771d242287d3a63eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Ma=C5=9Blanka?= <piotr.maslanka@henrietta.com.pl>
Date: Sat, 25 Apr 2020 15:31:46 +0200
Subject: [PATCH] extra unit test for AggregateMetric

---
 .../test_metrics/test_metrics.py                 | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/test_instrumentation/test_metrics/test_metrics.py b/tests/test_instrumentation/test_metrics/test_metrics.py
index a82e7e7b..b71d828e 100644
--- a/tests/test_instrumentation/test_metrics/test_metrics.py
+++ b/tests/test_instrumentation/test_metrics/test_metrics.py
@@ -123,6 +123,22 @@ class TestMetric(unittest.TestCase):
         self.assertTrue(inspect.isgeneratorfunction(generator))
         self.assertGreaterEqual(next(iter(metric.to_metric_data().values)).value, 1)
 
+    def test_aggregate_metric_measure_generator(self):
+        my_metric = getMetric('my_metric', 'summary', quantiles=0.5)
+        metric = AggregateMetric(my_metric)
+
+        @metric.measure()
+        def generator():
+            yield 2
+            time.sleep(1)
+            yield 3
+
+        for _ in generator():
+            pass
+        self.assertTrue(inspect.isgeneratorfunction(generator))
+        self.assertGreaterEqual(next(iter(my_metric.to_metric_data().values)).value, 1)
+
+
     def test_quantile_context_manager(self):
         metric = getMetric('test_metric', 'summary', quantiles=[0.5])
         with metric.measure():
-- 
GitLab