def test_count_metric(self):
metric = CountMetric()
for _ in range(10):
metric.incr()
self.assertEqual(metric.get_value_and_reset(), 10)
for _ in range(10):
metric.incr(to_add=10)
self.assertEqual(metric.get_value_and_reset(), 100)
self.assertEqual(metric.get_value_and_reset(), 0)
def test_gather_metrics(self):
name = "metric"
metric = CountMetric()
metric.incr(to_add=10)
self.metrics_collector.register_metric(name, metric, 60)
self.assertIn(60, self.metrics_collector.time_bucket_in_sec_to_metrics_name)
self.metrics_collector._gather_metrics(60)
message = self.metrics_collector.out_metrics.poll()
self.assertIsNotNone(message)
self.assertIsInstance(message, metrics_pb2.MetricPublisherPublishMessage)
self.assertEqual(message.metrics[0].name, name)
self.assertEqual(message.metrics[0].value, str(10))
self.assertEqual(metric.get_value_and_reset(), 0)
def test_gather_metrics(self):
name = "metric"
metric = CountMetric()
metric.incr(to_add=10)
self.metrics_collector.register_metric(name, metric, 60)
self.assertIn(
60, self.metrics_collector.time_bucket_in_sec_to_metrics_name)
self.metrics_collector._gather_metrics(60)
message = self.metrics_collector.out_metrics.poll()
self.assertIsNotNone(message)
self.assertIsInstance(message,
metrics_pb2.MetricPublisherPublishMessage)
self.assertEqual(message.metrics[0].name, name)
self.assertEqual(message.metrics[0].value, str(10))
self.assertEqual(metric.get_value_and_reset(), 0)