# coding=utf-8
import mock
import pytest

from common.microphone_utils import MicrophoneUtils
from robot_interface.utils.microphone import choose_mic_default, choose_mic_interactive


class MockPyAudio:
    def __init__(self):
        # You can predefine fake device info here
        self.devices = [
            {
                "index": 0,
                "name": u"Someone’s Microphone",  # Using a Unicode ’ character
                "maxInputChannels": 2,
                "maxOutputChannels": 0,
                "defaultSampleRate": 44100.0,
                "defaultLowInputLatency": 0.01,
                "defaultLowOutputLatency": 0.01,
                "defaultHighInputLatency": 0.1,
                "defaultHighOutputLatency": 0.1,
                "hostApi": 0,
            },
            {
                "index": 1,
                "name": u"Mock Speaker 1",
                "maxInputChannels": 0,
                "maxOutputChannels": 2,
                "defaultSampleRate": 48000.0,
                "defaultLowInputLatency": 0.01,
                "defaultLowOutputLatency": 0.01,
                "defaultHighInputLatency": 0.1,
                "defaultHighOutputLatency": 0.1,
                "hostApi": 0,
            },
        ]

    def get_device_count(self):
        """Return the number of available mock devices."""
        return len(self.devices)

    def get_device_info_by_index(self, index):
        """Return information for a given mock device index."""
        if 0 <= index < len(self.devices):
            return self.devices[index]
        else:
            raise IOError("Invalid device index: {}".format(index))

    def get_default_input_device_info(self):
        """Return info for a default mock input device."""
        for device in self.devices:
            if device.get("maxInputChannels", 0) > 0:
                return device
        raise IOError("No default input device found")


@pytest.fixture
def pyaudio_instance():
    return MockPyAudio()


def _raise_io_error():
    raise IOError()


class TestAudioUnit(MicrophoneUtils):
    """Run shared audio behavior tests with the mock implementation."""
    def test_choose_mic_default_no_mic(self):
        mock_pyaudio = mock.Mock()
        mock_pyaudio.get_device_count = mock.Mock(return_value=0L)
        mock_pyaudio.get_default_input_device_info = _raise_io_error

        result = choose_mic_default(mock_pyaudio)

        assert result is None

    def test_choose_mic_interactive_no_mic(self):
        mock_pyaudio = mock.Mock()
        mock_pyaudio.get_device_count = mock.Mock(return_value=0L)
        mock_pyaudio.get_default_input_device_info = _raise_io_error

        result = choose_mic_interactive(mock_pyaudio)

        assert result is None