sdk/platform-tools/systrace/catapult/telemetry/telemetry/internal/platform/power_monitor/android_power_monitor_controller_unittest.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import unittest
   6
   7 from telemetry.internal.platform import power_monitor as power_monitor
   8 from telemetry.internal.platform.power_monitor import (
   9   android_power_monitor_controller)
  10 import mock
  11 from devil.android import battery_utils
  12
  13 # pylint: disable=import-error, unused-argument
  14
  15
  16 class AndroidPowerMonitorControllerTest(unittest.TestCase):
  17   @mock.patch.object(battery_utils, 'BatteryUtils')
  18   def testComposition(self, _):
  19
  20     class P1(power_monitor.PowerMonitor):
  21       def StartMonitoringPower(self, browser):
  22         raise NotImplementedError()
  23       def StopMonitoringPower(self):
  24         raise NotImplementedError()
  25
  26     class P2(power_monitor.PowerMonitor):
  27       def __init__(self, value):
  28         super(P2, self).__init__()
  29         self._value = {'P2': value}
  30       def CanMonitorPower(self):
  31         return True
  32       def StartMonitoringPower(self, browser):
  33         pass
  34       def StopMonitoringPower(self):
  35         return self._value
  36
  37     class P3(power_monitor.PowerMonitor):
  38       def __init__(self, value):
  39         super(P3, self).__init__()
  40         self._value = {'P3': value}
  41       def CanMonitorPower(self):
  42         return True
  43       def StartMonitoringPower(self, browser):
  44         pass
  45       def StopMonitoringPower(self):
  46         return self._value
  47
  48     battery = battery_utils.BatteryUtils(None)
  49     controller = android_power_monitor_controller.AndroidPowerMonitorController(
  50         [P1(), P2(1), P3(2)], battery)
  51     self.assertEqual(controller.CanMonitorPower(), True)
  52     controller.StartMonitoringPower(None)
  53     controller_returns = controller.StopMonitoringPower()
  54     self.assertEqual(controller_returns['P2'], 1)
  55     self.assertEqual(controller_returns['P3'], 2)
  56
  57   @mock.patch.object(battery_utils, 'BatteryUtils')
  58   def testReenableChargingIfNeeded(self, mock_battery):
  59     battery = battery_utils.BatteryUtils(None)
  60     battery.GetCharging.return_value = False
  61     android_power_monitor_controller._ReenableChargingIfNeeded(battery)
  62
  63   def testMergePowerResultsOneEmpty(self):
  64     dict_one = {'platform_info': {}, 'component_utilization': {}}
  65     dict_two = {'test': 1, 'component_utilization': {'test': 2}}
  66     results = {
  67         'platform_info': {},
  68         'component_utilization': {'test': 2},
  69         'test': 1
  70     }
  71     (android_power_monitor_controller.AndroidPowerMonitorController.
  72      _MergePowerResults(dict_one, dict_two))
  73     self.assertDictEqual(dict_one, results)
  74
  75   def testMergePowerResultsSameEntry(self):
  76     dict_one = {
  77         'test': 1,
  78         'component_utilization': {'test': 2},
  79         'platform_info': {'test2': 'a'}
  80     }
  81     dict_two = {'test': 3, 'platform_info': {'test': 4}}
  82     results = {
  83         'test' : 3,
  84         'component_utilization': {'test': 2},
  85         'platform_info': {'test': 4, 'test2': 'a'}
  86     }
  87     (android_power_monitor_controller.AndroidPowerMonitorController.
  88      _MergePowerResults(dict_one, dict_two))
  89     self.assertDictEqual(dict_one, results)