client/site_tests/display_ClientChameleonConnection/display_ClientChameleonConnection.py

# Copyright 2014 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.

"""This is a client-side test to check the Chameleon connection."""

import logging

from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib.cros import chrome
from autotest_lib.client.cros import constants
from autotest_lib.client.cros.chameleon import chameleon
from autotest_lib.client.cros.chameleon import chameleon_port_finder
from autotest_lib.client.cros.multimedia import local_facade_factory


class display_ClientChameleonConnection(test.test):
    """Chameleon connection client test.

    This test talks to a Chameleon board from DUT. Try to plug the Chameleon
    ports and see if DUT detects them.
    """
    version = 1

    def run_once(self, host, args):
        ext_paths = [constants.DISPLAY_TEST_EXTENSION]
        with chrome.Chrome(extension_paths=ext_paths, autotest_ext=True) as cr:
            factory = local_facade_factory.LocalFacadeFactory(cr)
            display_facade = factory.create_display_facade()

            chameleon_board = chameleon.create_chameleon_board(host.hostname,
                                                               args)
            chameleon_board.setup_and_reset(self.outputdir)
            finder = chameleon_port_finder.ChameleonVideoInputFinder(
                    chameleon_board, display_facade)
            ports = finder.find_all_ports()

            connected_ports = ports.connected
            dut_failed_ports = ports.failed

            msg = str(finder)
            logging.debug(msg)

            if dut_failed_ports or not connected_ports:
                raise error.TestFail(msg)