server/site_tests/firmware_FAFTSetup/firmware_FAFTSetup.py

# Copyright (c) 2012 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.

from itertools import groupby
import logging
from threading import Timer

from autotest_lib.client.common_lib import error
from autotest_lib.server.cros.faftsequence import FAFTSequence

class firmware_FAFTSetup(FAFTSequence):
    """This test checks the following FAFT hardware requirement:
      - Warm reset
      - Cold reset
      - Recovery boot with USB stick
      - USB stick is plugged into Servo board, not DUT
      - Keyboard simulation
    """
    version = 1


    def console_checker(self):
        """Verify EC console is available if using Chrome EC."""
        if not self.check_ec_capability():
            # Not Chrome EC. Nothing to check.
            return True
        try:
            expected_output = ["Chip:\s+[^\r\n]*\r\n",
                               "RO:\s+[^\r\n]*\r\n",
                               "RW:\s+[^\r\n]*\r\n",
                               "Build:\s+[^\r\n]*\r\n"]
            self.send_uart_command_get_output("version",
                                              expected_output,
                                              timeout=0.2)
            return True
        except:
            logging.error("Cannot talk to EC console.")
            return False

    def keyboard_checker(self):
        """Press 'd', Ctrl, ENTER, Refresh by servo and check from DUT."""
        # Stop UI so that key presses don't go to X.
        self.faft_client.run_shell_command("stop ui")
        # Press the four keys with one-second delay in between.
        Timer(2, self.servo.d_key).start()
        Timer(3, self.servo.ctrl_key).start()
        Timer(4, self.servo.enter_key).start()
        Timer(5, self.servo.refresh_key).start()
        lines = self.faft_client.run_shell_command_get_output("showkey")
        # Turn UI back on
        self.faft_client.run_shell_command("start ui")

        # We may be getting multiple key-press or key-release.
        # Let's remove duplicated items.
        dup_removed = [x[0] for x in groupby(lines)]

        keycode_seq = [32, 29, 28, 61]
        expected_output = []
        for keycode in keycode_seq:
            expected_output.extend([
                "keycode  %d press" % keycode,
                "keycode  %d release" % keycode])

        if dup_removed[-8:] != expected_output:
            logging.error("Keyboard simulation not working correctly")
            logging.error("Captured keycodes:\n" + "\n".join(dup_removed[-8:]))
            logging.error("Expected keycodes:\n" + "\n".join(expected_output))
            return False
        return True

    def run_once(self, host=None):
        self.register_faft_sequence((
            {   # Step 1, Check EC console is available and test warm reboot
                'state_checker': self.console_checker,
                'reboot_action': self.sync_and_warm_reboot,
            },
            {   # Step 2, Check test image in USB stick and recovery boot
                'userspace_action': self.assert_test_image_in_usb_disk,
                'reboot_action': self.enable_rec_mode_and_reboot,
                'firmware_action': self.wait_fw_screen_and_plug_usb,
                'install_deps_after_boot': True,
            },
            {   # Step 3, Test cold reboot
                'reboot_action': self.sync_and_cold_reboot,
            },
            {   # Step 4, Check keyboard simulation
                'state_checker': self.keyboard_checker,
            },
        ))
        self.run_faft_sequence()