50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
# Copyright 2021 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
|
|
|
|
from autotest_lib.client.common_lib import error
|
|
from autotest_lib.server.cros.faft.firmware_test import FirmwareTest
|
|
|
|
|
|
class firmware_MenuPowerOff(FirmwareTest):
|
|
"""
|
|
Servo based test for powering off the device through the UI menu.
|
|
"""
|
|
version = 1
|
|
|
|
# Timeout of confirming DUT shutdown
|
|
POWER_OFF_TIMEOUT = 20
|
|
|
|
def initialize(self, host, cmdline_args):
|
|
super(firmware_MenuPowerOff, self).initialize(host, cmdline_args)
|
|
if not self.menu_switcher:
|
|
raise error.TestNAError('Test skipped for menuless UI')
|
|
if not self.faft_config.chrome_ec:
|
|
raise error.TestNAError('Cannot check power state without EC')
|
|
self.switcher.setup_mode('dev')
|
|
self.setup_usbkey(usbkey=False)
|
|
|
|
def run_once(self):
|
|
"""Method which actually runs the test."""
|
|
self.check_state((self.checkers.mode_checker, 'dev'))
|
|
self.switcher.simple_reboot()
|
|
|
|
# Now the device should be in dev screen
|
|
logging.info('Power off device in developer screen')
|
|
self.run_shutdown_process(self.menu_switcher.power_off,
|
|
run_power_action=False,
|
|
shutdown_timeout=self.POWER_OFF_TIMEOUT)
|
|
|
|
# Reboot to rec screen
|
|
self.switcher.enable_rec_mode_and_reboot(usb_state='host')
|
|
|
|
# Now the device should be in rec screen
|
|
logging.info('Power off device in recovery screen')
|
|
self.run_shutdown_process(
|
|
self.menu_switcher.power_off,
|
|
post_power_action=self.switcher.bypass_dev_mode,
|
|
shutdown_timeout=self.POWER_OFF_TIMEOUT)
|
|
self.switcher.wait_for_client()
|