1# Copyright 2021 The Chromium OS Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import logging 6 7from autotest_lib.client.common_lib import error 8from autotest_lib.server.cros.faft.firmware_test import FirmwareTest 9 10 11class firmware_MenuPowerOff(FirmwareTest): 12 """ 13 Servo based test for powering off the device through the UI menu. 14 """ 15 version = 1 16 17 # Timeout of confirming DUT shutdown 18 POWER_OFF_TIMEOUT = 20 19 20 def initialize(self, host, cmdline_args): 21 super(firmware_MenuPowerOff, self).initialize(host, cmdline_args) 22 if not self.menu_switcher: 23 raise error.TestNAError('Test skipped for menuless UI') 24 if not self.faft_config.chrome_ec: 25 raise error.TestNAError('Cannot check power state without EC') 26 self.switcher.setup_mode('dev') 27 self.setup_usbkey(usbkey=False) 28 29 def run_once(self): 30 """Method which actually runs the test.""" 31 self.check_state((self.checkers.mode_checker, 'dev')) 32 self.switcher.simple_reboot() 33 34 # Now the device should be in dev screen 35 logging.info('Power off device in developer screen') 36 self.run_shutdown_process(self.menu_switcher.power_off, 37 run_power_action=False, 38 shutdown_timeout=self.POWER_OFF_TIMEOUT) 39 40 # Reboot to rec screen 41 self.switcher.enable_rec_mode_and_reboot(usb_state='host') 42 43 # Now the device should be in rec screen 44 logging.info('Power off device in recovery screen') 45 self.run_shutdown_process( 46 self.menu_switcher.power_off, 47 post_power_action=self.switcher.bypass_dev_mode, 48 shutdown_timeout=self.POWER_OFF_TIMEOUT) 49 self.switcher.wait_for_client() 50