def test_enable_heartbeat_metrics_exists(self, transport_mock): # pylint: disable=protected-access producer = heartbeat_metrics.AzureHeartbeatMetricsProducer() heartbeat_metrics._HEARTBEAT_METRICS = producer heartbeat_metrics.enable_heartbeat_metrics(None, None) self.assertEqual(heartbeat_metrics._HEARTBEAT_METRICS, producer) transport_mock.assert_not_called()
def test_producer_get_metrics(self): producer = heartbeat_metrics.AzureHeartbeatMetricsProducer() metrics = producer.get_metrics() self.assertEqual(len(metrics), 1)
def test_producer_ctor(self): producer = heartbeat_metrics.AzureHeartbeatMetricsProducer() # pylint: disable=protected-access metric = producer._heartbeat self.assertTrue( isinstance(metric, heartbeat_metrics.heartbeat.HeartbeatMetric))