[Android] Add option to wait for battery temperature to device provisioning

build/android/pylib/device/battery_utils_test.py

 #!/usr/bin/env python
 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """
 Unit tests for the contents of battery_utils.py
 """
 
 # pylint: disable=W0613
@@ skipping 187 matching lines @@
 
   @mock.patch('time.sleep', mock.Mock())
   def testChargeDeviceToLevel(self):
     with self.assertCalls(
         (self.call.battery.SetCharging(True)),
         (self.call.battery.GetBatteryInfo(), {'level': '50'}),
         (self.call.battery.GetBatteryInfo(), {'level': '100'})):
       self.battery.ChargeDeviceToLevel(95)
 
 
-class DeviceUtilsGetBatteryInfoTest(BatteryUtilsTest):
+class BatteryUtilsGetBatteryInfoTest(BatteryUtilsTest):
 
   def testGetBatteryInfo_normal(self):
     with self.assertCall(
         self.call.device.RunShellCommand(
             ['dumpsys', 'battery'], check_return=True),
         [
           'Current Battery Service state:',
           '  AC powered: false',
           '  USB powered: true',
           '  level: 100',
           '  temperature: 321',
         ]):
       self.assertEquals(
           {
             'AC powered': 'false',
             'USB powered': 'true',
             'level': '100',
             'temperature': '321',
           },
           self.battery.GetBatteryInfo())
 
   def testGetBatteryInfo_nothing(self):
     with self.assertCall(
         self.call.device.RunShellCommand(
             ['dumpsys', 'battery'], check_return=True), []):
       self.assertEquals({}, self.battery.GetBatteryInfo())
 
 
-class DeviceUtilsGetChargingTest(BatteryUtilsTest):
+class BatteryUtilsGetChargingTest(BatteryUtilsTest):
 
   def testGetCharging_usb(self):
     with self.assertCall(
         self.call.battery.GetBatteryInfo(), {'USB powered': 'true'}):
       self.assertTrue(self.battery.GetCharging())
 
   def testGetCharging_usbFalse(self):
     with self.assertCall(
         self.call.battery.GetBatteryInfo(), {'USB powered': 'false'}):
       self.assertFalse(self.battery.GetCharging())
 
   def testGetCharging_ac(self):
     with self.assertCall(
         self.call.battery.GetBatteryInfo(), {'AC powered': 'true'}):
       self.assertTrue(self.battery.GetCharging())
 
   def testGetCharging_wireless(self):
     with self.assertCall(
         self.call.battery.GetBatteryInfo(), {'Wireless powered': 'true'}):
       self.assertTrue(self.battery.GetCharging())
 
   def testGetCharging_unknown(self):
     with self.assertCall(
         self.call.battery.GetBatteryInfo(), {'level': '42'}):
       self.assertFalse(self.battery.GetCharging())
 
 
-class DeviceUtilsGetNetworkDataTest(BatteryUtilsTest):
+class BatteryUtilsGetNetworkDataTest(BatteryUtilsTest):
 
   def testGetNetworkData_noDataUsage(self):
     with self.assertCalls(
         (self.call.device.RunShellCommand(
             ['dumpsys', 'batterystats', '-c'], check_return=True),
             _DUMPSYS_OUTPUT),
         (self.call.device.ReadFile('/proc/uid_stat/1000/tcp_snd'),
             self.ShellError()),
         (self.call.device.ReadFile('/proc/uid_stat/1000/tcp_rcv'),
             self.ShellError())):
@@ skipping 25 matching lines @@
   def testGetNetworkData_clearedCache(self):
     with self.assertCalls(
         (self.call.device.RunShellCommand(
             ['dumpsys', 'batterystats', '-c'], check_return=True),
             _DUMPSYS_OUTPUT),
         (self.call.device.ReadFile('/proc/uid_stat/1000/tcp_snd'), 1),
         (self.call.device.ReadFile('/proc/uid_stat/1000/tcp_rcv'), 2)):
       self.battery._cache.clear()
       self.assertEqual(self.battery.GetNetworkData('test_package1'), (1,2))
 
+class BatteryUtilsLetBatteryCoolToTemperatureTest(BatteryUtilsTest):
+
+  @mock.patch('time.sleep', mock.Mock())
+  def testLetBatteryCoolToTemperature_startUnder(self):
+    with self.assertCalls(
+        (self.call.battery.GetBatteryInfo(), {'temperature': '500'})):
+      self.battery.LetBatteryCoolToTemperature(600)
+
+  @mock.patch('time.sleep', mock.Mock())
+  def testLetBatteryCoolToTemperature_startOver(self):
+    with self.assertCalls(
+        (self.call.battery.GetBatteryInfo(), {'temperature': '500'}),
+        (self.call.battery.GetBatteryInfo(), {'temperature': '400'})):
+      self.battery.LetBatteryCoolToTemperature(400)
 
 if __name__ == '__main__':
   logging.getLogger().setLevel(logging.DEBUG)
   unittest.main(verbosity=2)