1 from autotest_lib.client.common_lib.cros.cfm.metrics import ( 2 media_metrics_collector) 3 4 import mock 5 import unittest 6 7 # pylint: disable=missing-docstring 8 class MediaMetricsCollectorTest(unittest.TestCase): 9 10 def test_data_point_collector_same_timestamp_one_entry(self): 11 cfm_facade = mock.MagicMock() 12 cfm_facade.get_media_info_data_points = mock.Mock( 13 return_value = [{'timestamp': 123.1}]) 14 data_point_collector = media_metrics_collector.DataPointCollector( 15 cfm_facade) 16 data_point_collector.collect_snapshot() 17 data_point_collector.collect_snapshot() 18 self.assertEqual(1, len(data_point_collector.get_data_points())) 19 20 def test_data_point_collector_different_timestamps_many_entries(self): 21 cfm_facade = mock.MagicMock() 22 cfm_facade.get_media_info_data_points = mock.Mock( 23 side_effect=[[{'timestamp': 123.1}], 24 [{'timestamp': 124.1}], 25 [{'timestamp': 125.1}] 26 ]) 27 data_point_collector = media_metrics_collector.DataPointCollector( 28 cfm_facade) 29 data_point_collector.collect_snapshot() 30 data_point_collector.collect_snapshot() 31 data_point_collector.collect_snapshot() 32 self.assertEqual(3, len(data_point_collector.get_data_points())) 33 34 35 36 37 38 39