Forcibly turn display on and reset X settings at the start of each iteration.

client/site_tests/power_LoadTest/power_LoadTest.py

 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import logging, os, shutil, sys, time
 from autotest_lib.client.bin import site_backchannel, site_ui_test, site_login
 from autotest_lib.client.common_lib import error, site_httpd, \
                             site_power_status, site_ui, utils
 
 # Workaround so flimflam.py doesn't need to be installed in the chroot.
@@ skipping 84 matching lines @@
                                                       ssid=wifi_ap,
                                                       security=wifi_sec,
                                                       passphrase=wifi_pw,
                                                       mode='managed')[0]:
                 raise error.TestError('Could not connect to WiFi network.')
 
         if check_network and site_backchannel.is_network_iface_running('eth0'):
             raise error.TestError(
                 'Ethernet interface is active. Please remove Ethernet cable')
 
-        # TODO (snanda):
-        # - set brightness level
-        # - turn off suspend on idle (not implemented yet in Chrome OS)
-
-        # record the current and max backlight levels
+        # record the max backlight level
         cmd = 'backlight-tool --get_max_brightness'
-        self._tmp_keyvals['level_backlight_max'] = int(
-                                             utils.system_output(cmd).rstrip())
-
-        cmd = 'backlight-tool --get_brightness'
-        self._tmp_keyvals['level_backlight_current'] = int(
-                                             utils.system_output(cmd).rstrip())
-
-        # disable screen locker and powerd
-        os.system('stop screen-locker')
-        os.system('stop powerd')
+        self._max_backlight = int(utils.system_output(cmd).rstrip())
+        self._tmp_keyvals['level_backlight_max'] = self._max_backlight
 
         # fix up file perms for the power test extension so that chrome
         # can access it
         os.system('chmod -R 755 %s' % self.bindir)
 
-
         # TODO (bleung) :
         # The new external extension packed crx means we can't pass params by
         # modifying params.js
         # Possible solution :
         # - modify extension to not start until we poke it from the browser.
         #       then pass through URL.
 
         # write test parameters to the power extension's params.js file
         # self._ext_path = os.path.join(self.bindir, 'extension')
         # self._write_ext_params()
@@ skipping 31 matching lines @@
 
         for i in range(self._loop_count):
             # the power test extension will report its status here
             latch = self._testServer.add_wait_url('/status')
 
             if site_login.logged_in():
                 site_login.attempt_logout()
             # the act of logging in will launch chrome with external extension.
             self.login(self.username, self.password)
 
+            # stop powerd
+            os.system('stop powerd')
+
+            # reset X settings since X gets restarted upon login
+            self._do_xset()
+
+            # reset backlight level since powerd might've modified it
+            # based on ambient light
+            self._set_backlight_level()
+
             low_battery = self._do_wait(self._verbose, self._loop_time,
                                         latch)
 
             if self._verbose:
                 logging.debug('loop %d completed' % i)
 
             if low_battery:
                 logging.info('Exiting due to low battery')
                 break
 
@@ skipping 46 matching lines @@
 
         self.write_perf_keyval(keyvals)
 
 
     def cleanup(self):
         # remove json file after test to stop external extension launch.
         if self._json_path:
             jsonfile = os.path.join(self._json_path, 'external_extensions.json')
             if os.path.exists(jsonfile):
                 os.system('rm -f %s' % jsonfile)
-        # re-enable screen locker and powerd. This also re-enables dpms.
+        # re-enable powerd
         os.system('start powerd')
-        os.system('start screen-locker')
         if site_login.logged_in():
             site_login.attempt_logout()
 
 
     def _percent_current_charge(self):
         return self._power_status.battery[0].charge_now * 100 / \
                self._power_status.battery[0].charge_full_design
 
 
     def _write_ext_params(self):
@@ skipping 42 matching lines @@
             for e in form_data:
                 key = 'ext_' + e
                 if key in self._tmp_keyvals:
                     self._tmp_keyvals[key] += form_data[e]
                 else:
                     self._tmp_keyvals[key] = form_data[e]
         else:
             logging.debug("Didn't get status back from power extension")
 
         return low_battery
+
+
+    def _do_xset(self):
+        XSET = 'LD_LIBRARY_PATH=/usr/local/lib xset'
+        # Disable X screen saver
+        site_ui.xsystem('%s s 0 0' % XSET)
+        # Disable DPMS Standby/Suspend/Off
+        site_ui.xsystem('%s dpms 0 0 0' % XSET)
+        # Force monitor on
+        site_ui.xsystem('%s dpms force on' % XSET)
+        # Save off X settings
+        site_ui.xsystem('%s q' % XSET)
+
+
+    def _set_backlight_level(self):
+        # set backlight level to 40% of max
+        cmd = 'backlight-tool --set_brightness %d ' % (
+              int(self._max_backlight * 0.4))
+        os.system(cmd)
+
+        # record brightness level
+        cmd = 'backlight-tool --get_brightness'
+        level = int(utils.system_output(cmd).rstrip())
+        logging.info('backlight level is %d' % level)
+        self._tmp_keyvals['level_backlight_current'] = level