def test_constructor_live_metrics(self):
mock_meter = mock.Mock()
performance_metrics_collector = PerformanceMetrics(
meter=mock_meter,
labels=self._test_labels,
collection_type=AutoCollectionType.LIVE_METRICS,
)
self.assertEqual(performance_metrics_collector._meter, mock_meter)
self.assertEqual(performance_metrics_collector._labels,
self._test_labels)
self.assertEqual(mock_meter.register_observer.call_count, 2)
reg_obs_calls = mock_meter.register_observer.call_args_list
reg_obs_calls[0].assert_called_with(
callback=performance_metrics_collector._track_cpu,
name="\\Processor(_Total)\\% Processor Time",
description="Processor time as a percentage",
unit="percentage",
value_type=float,
)
reg_obs_calls[1].assert_called_with(
callback=performance_metrics_collector._track_commited_memory,
name="\\Memory\\Committed Bytes",
description="Amount of commited memory in bytes",
unit="byte",
value_type=int,
)
def test_track_memory(self, psutil_mock):
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
memory = collections.namedtuple("memory", "available")
vmem = memory(available=100)
psutil_mock.return_value = vmem
obs = Observer(
callback=performance_metrics_collector._track_memory,
name="\\Memory\\Available Bytes",
description="Amount of available memory in bytes",
unit="byte",
value_type=int,
meter=self._meter,
)
performance_metrics_collector._track_memory(obs)
self.assertEqual(
obs.aggregators[tuple(self._test_labels.items())].current, 100)
def test_track_process_memory_exception(self, logger_mock):
with mock.patch(
"azure_monitor.sdk.auto_collection.performance_metrics.PROCESS",
throw(Exception),
):
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
obs = Observer(
callback=performance_metrics_collector._track_process_memory,
name="\\Process(??APP_WIN32_PROC??)\\Private Bytes",
description="Amount of memory process has used in bytes",
unit="byte",
value_type=int,
meter=self._meter,
)
performance_metrics_collector._track_process_memory(obs)
self.assertEqual(logger_mock.exception.called, True)
def test_track_process_cpu_exception(self, logger_mock):
with mock.patch(
"azure_monitor.sdk.auto_collection.performance_metrics.psutil"
) as psutil_mock:
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
psutil_mock.cpu_count.return_value = None
obs = Observer(
callback=performance_metrics_collector._track_process_cpu,
name="\\Process(??APP_WIN32_PROC??)\\% Processor Time",
description="Process CPU usage as a percentage",
unit="percentage",
value_type=float,
meter=self._meter,
)
performance_metrics_collector._track_process_cpu(obs)
self.assertEqual(logger_mock.exception.called, True)
def test_track_cpu(self):
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
with mock.patch("psutil.cpu_times_percent") as processor_mock:
cpu = collections.namedtuple("cpu", "idle")
cpu_times = cpu(idle=94.5)
processor_mock.return_value = cpu_times
obs = Observer(
callback=performance_metrics_collector._track_cpu,
name="\\Processor(_Total)\\% Processor Time",
description="Processor time as a percentage",
unit="percentage",
value_type=float,
meter=self._meter,
)
performance_metrics_collector._track_cpu(obs)
self.assertEqual(
obs.aggregators[tuple(self._test_labels.items())].current, 5.5)
def test_track_process_memory(self):
with mock.patch(
"azure_monitor.sdk.auto_collection.performance_metrics.PROCESS"
) as process_mock:
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
memory = collections.namedtuple("memory", "rss")
pmem = memory(rss=100)
process_mock.memory_info.return_value = pmem
obs = Observer(
callback=performance_metrics_collector._track_process_memory,
name="\\Process(??APP_WIN32_PROC??)\\Private Bytes",
description="Amount of memory process has used in bytes",
unit="byte",
value_type=int,
meter=self._meter,
)
performance_metrics_collector._track_process_memory(obs)
self.assertEqual(
obs.aggregators[tuple(self._test_labels.items())].current, 100)
def test_track_process_cpu(self, psutil_mock):
with mock.patch(
"azure_monitor.sdk.auto_collection.performance_metrics.PROCESS"
) as process_mock:
performance_metrics_collector = PerformanceMetrics(
meter=self._meter, labels=self._test_labels)
process_mock.cpu_percent.return_value = 44.4
psutil_mock.cpu_count.return_value = 2
obs = Observer(
callback=performance_metrics_collector._track_process_cpu,
name="\\Process(??APP_WIN32_PROC??)\\% Processor Time",
description="Process CPU usage as a percentage",
unit="percentage",
value_type=float,
meter=self._meter,
)
performance_metrics_collector._track_process_cpu(obs)
self.assertEqual(
obs.aggregators[tuple(self._test_labels.items())].current,
22.2)
def test_constructor_perf_counters(self):
mock_meter = mock.Mock()
performance_metrics_collector = PerformanceMetrics(
meter=mock_meter,
labels=self._test_labels,
collection_type=AutoCollectionType.PERF_COUNTER,
)
self.assertEqual(performance_metrics_collector._meter, mock_meter)
self.assertEqual(
performance_metrics_collector._labels, self._test_labels
)
self.assertEqual(mock_meter.register_observer.call_count, 4)
reg_obs_calls = mock_meter.register_observer.call_args_list
reg_obs_calls[0].assert_called_with(
callback=performance_metrics_collector._track_cpu,
name="\\Processor(_Total)\\% Processor Time",
description="Processor time as a percentage",
unit="percentage",
value_type=float,
)
reg_obs_calls[1].assert_called_with(
callback=performance_metrics_collector._track_memory,
name="\\Memory\\Available Bytes",
description="Amount of available memory in bytes",
unit="byte",
value_type=int,
)
reg_obs_calls[2].assert_called_with(
callback=performance_metrics_collector._track_process_cpu,
name="\\Process(??APP_WIN32_PROC??)\\% Processor Time",
description="Process CPU usage as a percentage",
unit="percentage",
value_type=float,
)
reg_obs_calls[3].assert_called_with(
callback=performance_metrics_collector._track_process_memory,
name="\\Process(??APP_WIN32_PROC??)\\Private Bytes",
description="Amount of memory process has used in bytes",
unit="byte",
value_type=int,
)
