def test_performance_collection_success(self, mock_collect_perf_metrics,
mock_dispatch):
perf_task = telemetry.PerformanceCollectionTask()
storage_id = fake_storage['id']
mock_collect_perf_metrics.return_value = []
perf_task.collect(context, storage_id, [], 100800, 100900)
self.assertEqual(mock_collect_perf_metrics.call_count, 1)
self.assertEqual(mock_dispatch.call_count, 1)
def test_successful_remove(self, mock_task_del, mock_failed_task_del):
telemetry_obj = telemetry.PerformanceCollectionTask()
telemetry_obj.remove_telemetry(context,
'c5c91c98-91aa-40e6-85ac-37a1d3b32bda')
mock_task_del.assert_called_with(
context, 'c5c91c98-91aa-40e6-85ac-37a1d3b32bda')
mock_failed_task_del.assert_called_with(
context, 'c5c91c98-91aa-40e6-85ac-37a1d3b32bda')
def test_performance_collection_failure(self, mock_collect_perf_metrics,
mock_dispatch, mock_log_error):
perf_task = telemetry.PerformanceCollectionTask()
storage_id = fake_storage['id']
# No alert
mock_collect_perf_metrics.return_value = []
mock_collect_perf_metrics.side_effect = \
exception.Invalid('Fake exception')
perf_task.collect(context, storage_id, [], 100800, 100900)
# Verify that dispatch is not done and error is logged
# when collect metric fails
self.assertEqual(mock_dispatch.call_count, 0)
self.assertEqual(mock_log_error.call_count, 1)