1# Copyright 2020 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_MenuModeTransition(FirmwareTest):
12    """
13    Servo based test for manual mode transitions through the UI menu.
14    """
15    version = 1
16
17    def initialize(self, host, cmdline_args):
18        super(firmware_MenuModeTransition, self).initialize(host, cmdline_args)
19        if not self.menu_switcher:
20            raise error.TestNAError('Test skipped for menuless UI')
21        self.switcher.setup_mode('normal')
22        self.setup_usbkey(usbkey=False)
23
24    def run_once(self):
25        """Method which actually runs the test."""
26        self.check_state((self.checkers.mode_checker, 'normal'))
27
28        # Trigger to-dev by menu navigation
29        logging.info('Trigger to-dev by menu navigation.')
30        self.switcher.enable_rec_mode_and_reboot(usb_state='host')
31        self.switcher.wait_for_client_offline()
32        self.menu_switcher.trigger_rec_to_dev()
33
34        # Now the device should be in dev mode screen
35        self.menu_switcher.dev_boot_from_internal()
36        self.switcher.wait_for_client()
37
38        logging.info('Expected dev mode boot.')
39        self.check_state((self.checkers.mode_checker, 'dev'))
40
41        # Trigger to-norm by menu navigation
42        logging.info('Trigger to-norm by menu navigation.')
43        self.switcher.simple_reboot()
44        self.switcher.wait_for_client_offline()
45        self.menu_switcher.trigger_dev_to_normal()
46        self.switcher.wait_for_client()
47
48        logging.info('Expected normal mode boot, done.')
49        self.check_state((self.checkers.mode_checker, 'normal'))
50