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