| # Copyright (c) 2012 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, time |
| |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.server.cros.faft.faft_classes import FAFTSequence |
| |
| |
| class firmware_DevBootUSB(FAFTSequence): |
| """ |
| Servo based Ctrl-U developer USB boot test. |
| |
| This test requires a USB disk plugged-in, which contains a Chrome OS test |
| image (built by "build_image test"). On runtime, this test first switches |
| DUT to developer mode. When dev_boot_usb=0, pressing Ctrl-U on developer |
| screen should not boot the USB disk. When dev_boot_usb=1, pressing Ctrl-U |
| should boot the USB disk. |
| """ |
| version = 1 |
| |
| |
| def initialize(self, host, cmdline_args, ec_wp=None): |
| super(firmware_DevBootUSB, self).initialize(host, cmdline_args, |
| ec_wp=ec_wp) |
| self.setup_dev_mode(dev_mode=True) |
| self.setup_usbkey(usbkey=True, host=False) |
| |
| self.original_dev_boot_usb = self.faft_client.system.get_dev_boot_usb() |
| logging.info('Original dev_boot_usb value: %s', |
| str(self.original_dev_boot_usb)) |
| |
| |
| def cleanup(self): |
| self.ensure_internal_device_boot() |
| super(firmware_DevBootUSB, self).cleanup() |
| |
| |
| def try_ctrl_u_and_ctrl_d(self): |
| """Try to press Ctrl-U first and then press Ctrl-D""" |
| self.wait_fw_screen_and_ctrl_u() |
| # If the above Ctrl-U doesn't work, the firmware beeps twice. |
| # Should wait the beep done before pressing Ctrl-D. |
| time.sleep(self.faft_config.beep) |
| self.press_ctrl_d() |
| |
| |
| def ensure_internal_device_boot(self): |
| """Ensure internal device boot; if not, reboot into it. |
| |
| If not, it may be a test failure during step 2 or 3, try to reboot |
| and press Ctrl-D to internal device boot. |
| """ |
| if self.faft_client.system.is_removable_device_boot(): |
| logging.info('Reboot into internal disk...') |
| self.run_faft_step({ |
| 'userspace_action': (self.faft_client.system.set_dev_boot_usb, |
| self.original_dev_boot_usb), |
| 'firmware_action': self.wait_fw_screen_and_ctrl_d, |
| }) |
| |
| |
| def run_once(self): |
| if (self.faft_config.has_keyboard and |
| not self.check_ec_capability(['keyboard'])): |
| raise error.TestNAError("TEST IT MANUALLY! This test can't be " |
| "automated on non-Chrome-EC devices.") |
| |
| self.register_faft_sequence(( |
| { # Step 1, expected developer mode, set dev_boot_usb to 0 |
| 'state_checker': (self.checkers.dev_boot_usb_checker, False), |
| 'userspace_action': (self.faft_client.system.set_dev_boot_usb, |
| 0), |
| # Ctrl-U doesn't take effect as dev_boot_usb=0. |
| # Falls back to Ctrl-D internal disk boot. |
| 'firmware_action': self.try_ctrl_u_and_ctrl_d, |
| 'install_deps_after_boot': True, |
| }, |
| { # Step 2, expected internal disk boot, set dev_boot_usb to 1 |
| 'state_checker': (self.checkers.dev_boot_usb_checker, False, |
| "Not internal disk boot, dev_boot_usb misbehaved"), |
| 'userspace_action': (self.faft_client.system.set_dev_boot_usb, |
| 1), |
| 'firmware_action': self.wait_fw_screen_and_ctrl_u, |
| 'install_deps_after_boot': True, |
| }, |
| { # Step 3, expected USB boot, set dev_boot_usb to the original |
| 'state_checker': (self.checkers.dev_boot_usb_checker, True, |
| "Not USB boot, Ctrl-U not work"), |
| 'userspace_action': (self.faft_client.system.set_dev_boot_usb, |
| self.original_dev_boot_usb), |
| 'firmware_action': self.wait_fw_screen_and_ctrl_d, |
| }, |
| { # Step 4, done |
| 'state_checker': (self.checkers.dev_boot_usb_checker, False), |
| } |
| )) |
| self.run_faft_sequence() |
