# Copyright 2011 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# TODO(b/342929556): Remove from PVS testplans and remove this file.

import logging
from autotest_lib.server.cros.faft.firmware_test import FirmwareTest


class firmware_TryFwB(FirmwareTest):
    """
    Servo based RW firmware B boot test.
    """
    version = 1

    def initialize(self, host, cmdline_args, dev_mode=False):
        super(firmware_TryFwB, self).initialize(host, cmdline_args)
        self.switcher.setup_mode('dev' if dev_mode else 'normal',
                                 allow_gbb_force=True)
        self.setup_usbkey(usbkey=False)

    def cleanup(self):
        super(firmware_TryFwB, self).cleanup()

    def run_once(self):
        """Runs a single iteration of the test."""
        logging.info("Set fw_try_next to B")
        self.check_state((self.checkers.fw_tries_checker, 'A'))
        self.faft_client.system.set_fw_try_next('B')
        self.switcher.mode_aware_reboot()

        logging.info("Expected firmware B boot, reboot")
        self.check_state((self.checkers.fw_tries_checker, 'B'))
        self.switcher.mode_aware_reboot()

        logging.info("Expected firmware B boot, done")
        self.check_state((self.checkers.fw_tries_checker, 'B'))