185 lines
7.8 KiB
Python
185 lines
7.8 KiB
Python
# Copyright (c) 2011 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
|
|
import time
|
|
|
|
from autotest_lib.client.common_lib import error
|
|
from autotest_lib.server.cros.faft.firmware_test import FirmwareTest
|
|
|
|
|
|
class firmware_FwScreenCloseLid(FirmwareTest):
|
|
"""
|
|
Servo based lid close triggered shutdown test during firmware screens.
|
|
|
|
This test requires a USB disk plugged-in, which contains a Chrome OS test
|
|
image (built by "build_image --test"). On runtime, this test triggers
|
|
firmware screens (developer, remove, insert, yuck, to_norm screens),
|
|
and then closes the lid in order to power the machine down.
|
|
"""
|
|
version = 1
|
|
NEEDS_SERVO_USB = True
|
|
|
|
SHORT_SHUTDOWN_CONFIRMATION_PERIOD = 0.1
|
|
|
|
def wait_fw_screen_and_close_lid(self):
|
|
"""Wait for firmware warning screen and close lid."""
|
|
time.sleep(self.faft_config.firmware_screen)
|
|
self.servo.lid_close()
|
|
|
|
def wait_longer_fw_screen_and_close_lid(self):
|
|
"""Wait for firmware screen without timeout and close lid."""
|
|
time.sleep(self.faft_config.firmware_screen)
|
|
self.wait_fw_screen_and_close_lid()
|
|
|
|
def wait_second_screen_and_close_lid(self):
|
|
"""Wait and trigger TO_NORM or RECOVERY INSERT screen and close lid."""
|
|
self.switcher.trigger_dev_to_rec()
|
|
self.wait_longer_fw_screen_and_close_lid()
|
|
|
|
def wait_yuck_screen_and_close_lid(self):
|
|
"""Wait and trigger yuck screen and clod lid."""
|
|
# Insert a corrupted USB stick. A yuck screen is expected.
|
|
self.servo.switch_usbkey('dut')
|
|
time.sleep(self.faft_config.usb_plug)
|
|
self.wait_longer_fw_screen_and_close_lid()
|
|
|
|
def initialize(self, host, cmdline_args):
|
|
"""Initialize the test"""
|
|
super(firmware_FwScreenCloseLid, self).initialize(host, cmdline_args)
|
|
if self.faft_config.has_lid:
|
|
self.switcher.setup_mode('dev')
|
|
self.setup_usbkey(True, host=True)
|
|
usb_dev = self.servo.probe_host_usb_dev()
|
|
# Corrupt the kernel of USB stick. It is needed for triggering a
|
|
# yuck screen later.
|
|
self.corrupt_usb_kernel(usb_dev)
|
|
|
|
def cleanup(self):
|
|
"""Cleanup the test"""
|
|
try:
|
|
if self.faft_config.has_lid:
|
|
self.servo.switch_usbkey('host')
|
|
usb_dev = self.servo.probe_host_usb_dev()
|
|
# Restore kernel of USB stick which is corrupted on setup phase.
|
|
self.restore_usb_kernel(usb_dev)
|
|
except Exception as e:
|
|
logging.error("Caught exception: %s", str(e))
|
|
super(firmware_FwScreenCloseLid, self).cleanup()
|
|
|
|
def run_once(self):
|
|
"""Main test logic"""
|
|
if not self.faft_config.has_lid:
|
|
logging.info('This test does nothing on devices without lid.')
|
|
return
|
|
|
|
# Some platforms may not turn on with just lid open
|
|
power_action = False
|
|
if not self.faft_config.lid_wake_from_power_off:
|
|
power_action = True
|
|
|
|
if self.faft_config.mode_switcher_type not in (
|
|
'keyboard_dev_switcher', 'tablet_detachable_switcher',
|
|
'menu_switcher'):
|
|
raise error.TestNAError("This test is only valid on devices with "
|
|
"screens.")
|
|
|
|
if self.faft_config.chrome_ec and not self.check_ec_capability(['lid'
|
|
]):
|
|
raise error.TestNAError(
|
|
"TEST IT MANUALLY! ChromeEC can't control "
|
|
"lid on the device %s" % self.faft_config.platform)
|
|
|
|
logging.info("Expected dev mode and reboot. "
|
|
"When the next DEVELOPER SCREEN shown, close lid "
|
|
"to make DUT shutdown.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '1',
|
|
'mainfw_type': 'developer',
|
|
}))
|
|
self.switcher.simple_reboot()
|
|
self.run_shutdown_process(
|
|
self.wait_fw_screen_and_close_lid,
|
|
pre_power_action=self.servo.lid_open,
|
|
run_power_action=power_action,
|
|
post_power_action=self.switcher.bypass_dev_mode)
|
|
self.switcher.wait_for_client()
|
|
|
|
logging.info("Reboot. When the developer screen shown, press "
|
|
"enter key to trigger either TO_NORM screen (new) or "
|
|
"RECOVERY INSERT screen (old). Then close lid to "
|
|
"make DUT shutdown.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '1',
|
|
'mainfw_type': 'developer',
|
|
}))
|
|
self.switcher.simple_reboot()
|
|
self.run_shutdown_process(
|
|
self.wait_second_screen_and_close_lid,
|
|
pre_power_action=self.servo.lid_open,
|
|
run_power_action=power_action,
|
|
post_power_action=self.switcher.bypass_dev_mode,
|
|
shutdown_timeout=self.SHORT_SHUTDOWN_CONFIRMATION_PERIOD)
|
|
self.switcher.wait_for_client()
|
|
|
|
logging.info("Request recovery boot. When the RECOVERY INSERT "
|
|
"screen shows, close lid to make DUT shutdown.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '1',
|
|
'mainfw_type': 'developer',
|
|
}))
|
|
self.faft_client.system.request_recovery_boot()
|
|
self.switcher.simple_reboot()
|
|
self.run_shutdown_process(
|
|
self.wait_longer_fw_screen_and_close_lid,
|
|
pre_power_action=self.servo.lid_open,
|
|
run_power_action=power_action,
|
|
post_power_action=self.switcher.bypass_dev_mode,
|
|
shutdown_timeout=self.SHORT_SHUTDOWN_CONFIRMATION_PERIOD)
|
|
self.switcher.wait_for_client()
|
|
|
|
logging.info("Request recovery boot again. When the recovery "
|
|
"insert screen shows, insert a corrupted USB and trigger "
|
|
"a YUCK SCREEN. Then close lid to make DUT shutdown.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '1',
|
|
'mainfw_type': 'developer',
|
|
}))
|
|
self.faft_client.system.request_recovery_boot()
|
|
self.switcher.simple_reboot()
|
|
self.run_shutdown_process(
|
|
self.wait_yuck_screen_and_close_lid,
|
|
pre_power_action=self.servo.lid_open,
|
|
run_power_action=power_action,
|
|
post_power_action=self.switcher.bypass_dev_mode,
|
|
shutdown_timeout=self.SHORT_SHUTDOWN_CONFIRMATION_PERIOD)
|
|
self.switcher.wait_for_client()
|
|
|
|
logging.info("Switch back to normal mode.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '1',
|
|
'mainfw_type': 'developer',
|
|
}))
|
|
self.switcher.reboot_to_mode(to_mode='normal')
|
|
|
|
logging.info("Expected normal mode and request recovery boot. "
|
|
"Because an USB stick is inserted, a RECOVERY REMOVE "
|
|
"screen shows. Close lid to make DUT shutdown.")
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '0',
|
|
'mainfw_type': 'normal',
|
|
}))
|
|
self.faft_client.system.request_recovery_boot()
|
|
self.switcher.simple_reboot()
|
|
self.run_shutdown_process(
|
|
self.wait_longer_fw_screen_and_close_lid,
|
|
pre_power_action=self.servo.lid_open,
|
|
run_power_action=power_action,
|
|
shutdown_timeout=self.SHORT_SHUTDOWN_CONFIRMATION_PERIOD)
|
|
self.switcher.wait_for_client()
|
|
self.check_state((self.checkers.crossystem_checker, {
|
|
'devsw_boot': '0',
|
|
'mainfw_type': 'normal',
|
|
}))
|