# Lint as: python2, python3
# Copyright 2015 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 server side test to check nodes created for internal card."""

from autotest_lib.client.cros.chameleon import audio_test_utils
from autotest_lib.server.cros.audio import audio_test


class audio_InternalCardNodes(audio_test.AudioTest):
    """Server side test to check audio nodes for internal card.

    This test talks to a Chameleon board and a Cros device to verify
    audio nodes created for internal cards are correct.

    """
    version = 1

    def get_expected_nodes(self, plugged):
        """Gets expected nodes should should be created for internal cards.

        @param plugged: True for plugged state, false otherwise.
        @returns:
            a tuple (output, input) containing lists of expected input and
            output nodes.
        """
        nodes = ([], [
                'POST_DSP_DELAYED_LOOPBACK', 'POST_DSP_LOOPBACK',
                'POST_MIX_LOOPBACK'
        ])
        if plugged:
            # Checks whether line-out or headphone is detected.
            hp_jack_node_type = audio_test_utils.check_hp_or_lineout_plugged(
                    self.facade)
            nodes[0].append(hp_jack_node_type)
            nodes[1].append('MIC')
        if audio_test_utils.has_internal_speaker(self.host):
            nodes[0].append('INTERNAL_SPEAKER')
        if audio_test_utils.has_internal_microphone(self.host):
            nodes[1].extend(
                    audio_test_utils.get_plugged_internal_mics(self.host))
        if audio_test_utils.has_hotwording(self.host):
            nodes[1].append('HOTWORD')
        if audio_test_utils.has_echo_reference(self.host):
            nodes[1].append('ECHO_REFERENCE')
        return nodes

    def run_once(self, plug=True, blocked_boards=[]):
        """Runs InternalCardNodes test."""
        if self.host.get_board().split(':')[1] in blocked_boards:
            raise error.TestNAError('Board not applicable to test!')
        if not audio_test_utils.has_audio_jack(self.host):
            audio_test_utils.check_plugged_nodes(
                    self.facade, self.get_expected_nodes(False))
            return

        jack_plugger = self.host.chameleon.get_audio_board().get_jack_plugger()

        if plug:
            jack_plugger.plug()
        else:
            jack_plugger.unplug()

        audio_test_utils.check_plugged_nodes(self.facade,
                                             self.get_expected_nodes(plug))