def test_counter_to_prometheus(self):
meter = get_meter_provider().get_meter(__name__)
metric = meter.create_metric(
"test@name",
"testdesc",
"unit",
int,
metrics.Counter,
["environment@", "os"],
)
kvp = {"environment@": "staging", "os": "Windows"}
label_set = meter.get_label_set(kvp)
aggregator = CounterAggregator()
aggregator.update(123)
aggregator.take_checkpoint()
record = MetricRecord(aggregator, label_set, metric)
collector = CustomCollector("testprefix")
collector.add_metrics_data([record])
for prometheus_metric in collector.collect():
self.assertEqual(type(prometheus_metric), CounterMetricFamily)
self.assertEqual(prometheus_metric.name, "testprefix_test_name")
self.assertEqual(prometheus_metric.documentation, "testdesc")
self.assertTrue(len(prometheus_metric.samples) == 1)
self.assertEqual(prometheus_metric.samples[0].value, 123)
self.assertTrue(len(prometheus_metric.samples[0].labels) == 2)
self.assertEqual(
prometheus_metric.samples[0].labels["environment_"], "staging")
self.assertEqual(prometheus_metric.samples[0].labels["os"],
"Windows")
def test_ungrouped_batcher_process_exists(self):
meter = metrics.MeterProvider().get_meter(__name__)
batcher = UngroupedBatcher(True)
aggregator = CounterAggregator()
aggregator2 = CounterAggregator()
metric = metrics.Counter(
"available memory",
"available memory",
"bytes",
int,
meter,
("environment", ),
)
label_set = metrics.LabelSet()
_batch_map = {}
_batch_map[(metric, label_set)] = aggregator
aggregator2.update(1.0)
batcher._batch_map = _batch_map
record = metrics.Record(metric, label_set, aggregator2)
batcher.process(record)
self.assertEqual(len(batcher._batch_map), 1)
self.assertIsNotNone(batcher._batch_map.get((metric, label_set)))
self.assertEqual(
batcher._batch_map.get((metric, label_set)).current, 0)
self.assertEqual(
batcher._batch_map.get((metric, label_set)).checkpoint, 1.0)
def test_update(self, time_mock):
time_mock.return_value = 123
counter = CounterAggregator()
counter.update(1.0)
counter.update(2.0)
self.assertEqual(counter.current, 3.0)
self.assertEqual(counter.last_update_timestamp, 123)
def test_finished_collection_stateless(self):
meter = metrics.MeterProvider().get_meter(__name__)
batcher = UngroupedBatcher(False)
aggregator = CounterAggregator()
metric = metrics.Counter(
"available memory",
"available memory",
"bytes",
int,
meter,
("environment", ),
)
aggregator.update(1.0)
label_set = metrics.LabelSet()
_batch_map = {}
_batch_map[(metric, label_set)] = aggregator
batcher._batch_map = _batch_map
batcher.finished_collection()
self.assertEqual(len(batcher._batch_map), 0)
def test_metric_to_envelope(self):
aggregator = CounterAggregator()
aggregator.update(123)
aggregator.take_checkpoint()
record = MetricRecord(
aggregator, self._test_label_set, self._test_metric
)
exporter = AzureMonitorMetricsExporter()
envelope = exporter.metric_to_envelope(record)
self.assertIsInstance(envelope, Envelope)
self.assertEqual(envelope.ver, 1)
self.assertEqual(envelope.name, "Microsoft.ApplicationInsights.Metric")
self.assertEqual(
envelope.time,
ns_to_iso_str(
record.metric.get_handle(
record.label_set
).last_update_timestamp
),
)
self.assertEqual(envelope.sample_rate, None)
self.assertEqual(envelope.seq, None)
self.assertEqual(envelope.ikey, "1234abcd-5678-4efa-8abc-1234567890ab")
self.assertEqual(envelope.flags, None)
self.assertIsInstance(envelope.data, Data)
self.assertIsInstance(envelope.data.base_data, MetricData)
self.assertEqual(envelope.data.base_data.ver, 2)
self.assertEqual(len(envelope.data.base_data.metrics), 1)
self.assertEqual(envelope.data.base_data.metrics[0].ns, "testname")
self.assertEqual(envelope.data.base_data.metrics[0].name, "testdesc")
self.assertEqual(envelope.data.base_data.metrics[0].value, 123)
self.assertEqual(
envelope.data.base_data.properties["environment"], "staging"
)
self.assertIsNotNone(envelope.tags["ai.cloud.role"])
self.assertIsNotNone(envelope.tags["ai.cloud.roleInstance"])
self.assertIsNotNone(envelope.tags["ai.device.id"])
self.assertIsNotNone(envelope.tags["ai.device.locale"])
self.assertIsNotNone(envelope.tags["ai.device.osVersion"])
self.assertIsNotNone(envelope.tags["ai.device.type"])
self.assertIsNotNone(envelope.tags["ai.internal.sdkVersion"])
def test_checkpoint_set(self):
meter = metrics.MeterProvider().get_meter(__name__)
batcher = UngroupedBatcher(True)
aggregator = CounterAggregator()
metric = metrics.Counter(
"available memory",
"available memory",
"bytes",
int,
meter,
("environment", ),
)
aggregator.update(1.0)
label_set = metrics.LabelSet()
_batch_map = {}
_batch_map[(metric, label_set)] = aggregator
batcher._batch_map = _batch_map
records = batcher.checkpoint_set()
self.assertEqual(len(records), 1)
self.assertEqual(records[0].metric, metric)
self.assertEqual(records[0].label_set, label_set)
self.assertEqual(records[0].aggregator, aggregator)
def test_checkpoint(self):
counter = CounterAggregator()
counter.update(2.0)
counter.take_checkpoint()
self.assertEqual(counter.current, 0)
self.assertEqual(counter.checkpoint, 2.0)
def test_update(self):
counter = CounterAggregator()
counter.update(1.0)
counter.update(2.0)
self.assertEqual(counter.current, 3.0)