xref: /aosp_15_r20/external/autotest/server/site_tests/firmware_MiniosMenu/firmware_MiniosMenu.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
1# Copyright (c) 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_MiniosMenu(FirmwareTest):
12    """
13    Servo based for MiniOS boot through the UI menu.
14
15    This test requires the device to support MiniOS. This test will boot to the
16    manual recovery screen and try to boot MiniOS through the UI menu.
17    """
18    version = 1
19
20    def initialize(self, host, cmdline_args, older_version):
21        super(firmware_MiniosMenu, self).initialize(host, cmdline_args)
22
23        self.test_skipped = True
24        if not self.menu_switcher:
25            raise error.TestNAError('Test skipped for menuless UI')
26        if not self.faft_config.chrome_ec:
27            raise error.TestNAError('Cannot check power state without EC')
28        if not self.faft_config.minios_enabled:
29            raise error.TestNAError('MiniOS is not enabled for this board')
30        self.test_skipped = False
31
32        self.host = host
33        self.switcher.setup_mode('normal')
34        self.setup_usbkey(usbkey=False)
35        self.older_version = older_version
36
37    def cleanup(self):
38        if not self.test_skipped:
39            try:
40                self.switcher.leave_minios()
41            except Exception as e:
42                logging.error('Caught exception: %s', str(e))
43        super(firmware_MiniosMenu, self).cleanup()
44
45    def run_once(self):
46        """Run a single iteration of the test."""
47        logging.info('Boot into recovery mode, older_version: %s',
48                     self.older_version)
49        self.switcher.reboot_to_mode(to_mode="rec", wait_for_dut_up=False)
50        self.wait_for('firmware_screen')
51        self.menu_switcher.trigger_rec_to_minios(self.older_version)
52        self.check_state(self.checkers.minios_checker)
53