# 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()