src/chrome/test/functional/chromeos_battery.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2011 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import os
   7
   8 import pyauto_functional  # Must be imported before pyauto
   9 import pyauto
  10
  11 from chromeos.power_strip import PowerStrip
  12
  13 class ChromeosBattery(pyauto.PyUITest):
  14   """Tests ChromeOS Battery Status.
  15
  16   Preconditions:
  17      1) Device under test (DUT) is connected to the LAN via an
  18      Ethernet-to-USB adapter plugged into one of its USB ports.
  19      2) AC power cable is connected to the DUT, and plugged into
  20      the IP controlled Power Switch, outlet #4, located in the lab.
  21      3) Battery is installed in the DUT, and battery is not fully
  22      discharged.
  23      4) Tester should have physical access to the power switch.
  24
  25   Note about time calculation:
  26      When AC power is turned off or on, the battery will take from 2
  27      to 60 seconds to calculate the time remaining to empty or full.
  28      While calculating, the keys 'battery_time_to_full' and
  29      'battery_time_to_empty' are absent.
  30   """
  31
  32   _BATTERY_CONFIG_FILE = os.path.join(pyauto.PyUITest.DataDir(),
  33                                       'pyauto_private', 'chromeos', 'power',
  34                                       'battery_testbed_config')
  35
  36   def setUp(self):
  37     pyauto.PyUITest.setUp(self)
  38     self.InitPowerStrip()
  39
  40   def tearDown(self):
  41     # Leave AC power ON so battery does not discharge between tests
  42     self._power_strip.PowerOn(self._outlet_battery_full)
  43     pyauto.PyUITest.tearDown(self)
  44
  45   def InitPowerStrip(self):
  46     assert os.path.exists(ChromeosBattery._BATTERY_CONFIG_FILE), \
  47         'Power Strip configuration file does not exist.'
  48     power_config = pyauto.PyUITest.EvalDataFrom(
  49         ChromeosBattery._BATTERY_CONFIG_FILE)
  50     self._power_strip = PowerStrip(power_config['strip_ip'])
  51     self._outlet_battery_full = (power_config['configs']
  52                                              ['battery_full']
  53                                              ['outlet_id'])
  54
  55   def WaitUntilBatteryState(self, ac_power_on, time_key):
  56     assert self.WaitUntil(lambda: self.BatteryPowerAndChargeStateAgree(
  57         ac_power_on, time_key), timeout=60, retry_sleep=1), \
  58         'Battery charge/discharge time was not calculated.'
  59     return
  60
  61   def BatteryPowerAndChargeStateAgree(self, ac_power_on, time_key):
  62     battery_status = self.GetBatteryInfo()
  63     return (battery_status.get('line_power_on') == ac_power_on and
  64         time_key in battery_status)
  65
  66   def testBatteryChargesWhenACisOn(self):
  67     """Calculate battery time to full when AC is ON"""
  68     # Apply AC power to chromebook with battery.
  69     self._power_strip.PowerOn(self._outlet_battery_full)
  70
  71     # Get info about charging battery
  72     self.WaitUntilBatteryState(True,'battery_time_to_full')
  73     battery_status = self.GetBatteryInfo()
  74     self.assertTrue(battery_status.get('battery_is_present'),
  75         msg='Battery is not present.')
  76     self.assertTrue(battery_status.get('line_power_on'),
  77         msg='Line power is off.')
  78     self.assertTrue(battery_status.get('battery_time_to_full') >= 0,
  79         msg='Battery charge time is negative.')
  80
  81   def testBatteryDischargesWhenACisOff(self):
  82     """Calculate battery time to empty when AC is OFF"""
  83     self._power_strip.PowerOff(self._outlet_battery_full)
  84
  85     # Get info about discharging battery
  86     self.WaitUntilBatteryState(False,'battery_time_to_empty')
  87     battery_status = self.GetBatteryInfo()
  88     self.assertTrue(battery_status.get('battery_is_present'),
  89         msg='Battery is not present.')
  90     self.assertFalse(battery_status.get('line_power_on'),
  91         msg='Line power is on.')
  92     self.assertTrue(battery_status.get('battery_time_to_empty') >= 0,
  93         msg='Battery discharge time is negative.')
  94
  95   def testBatteryTimesAreDifferent(self):
  96     """Time to full and time to empty should be different"""
  97     # Turn AC Power ON
  98     self._power_strip.PowerOn(self._outlet_battery_full)
  99
 100     # Get charging battery time to full
 101     self.WaitUntilBatteryState(True,'battery_time_to_full')
 102     battery_status = self.GetBatteryInfo()
 103     time_to_full = battery_status.get('battery_time_to_full')
 104
 105     # Turn AC Power OFF
 106     self._power_strip.PowerOff(self._outlet_battery_full)
 107
 108     # Get discharging battery time to empty
 109     self.WaitUntilBatteryState(False,'battery_time_to_empty')
 110     battery_status = self.GetBatteryInfo()
 111     time_to_empty = battery_status.get('battery_time_to_empty')
 112
 113     # Compare charge to discharge time
 114     """Confirm that time to full and time to empty are not
 115        returning the same value, but that the values are
 116        different (e.g., both are not zero)."""
 117     self.assertNotEqual(time_to_full, time_to_empty,
 118         msg='Battery time to full equals time to empty. '
 119             'Though very unlikely, this is not impossible. '
 120             'If test failed falsely, Kris owes Scott a beer.')
 121
 122
 123 if __name__ == '__main__':
 124   pyauto_functional.Main()