"""Configuration for Sonos tests.""" from unittest.mock import AsyncMock, MagicMock, Mock, patch as patch import pytest from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.sonos import DOMAIN from homeassistant.const import CONF_HOSTS from tests.common import MockConfigEntry @pytest.fixture(name="config_entry") def config_entry_fixture(): """Create a mock Sonos config entry.""" return MockConfigEntry(domain=DOMAIN, title="Sonos") @pytest.fixture(name="soco") def soco_fixture( music_library, speaker_info, battery_info, dummy_soco_service, alarmClock ): """Create a mock pysonos SoCo fixture.""" with patch("pysonos.SoCo", autospec=True) as mock, patch( "socket.gethostbyname", return_value="192.168.42.2" ): mock_soco = mock.return_value mock_soco.uid = "RINCON_test" mock_soco.play_mode = "NORMAL" mock_soco.music_library = music_library mock_soco.get_speaker_info.return_value = speaker_info mock_soco.avTransport = dummy_soco_service mock_soco.renderingControl = dummy_soco_service mock_soco.zoneGroupTopology = dummy_soco_service mock_soco.contentDirectory = dummy_soco_service mock_soco.deviceProperties = dummy_soco_service mock_soco.alarmClock = alarmClock mock_soco.mute = False mock_soco.night_mode = True mock_soco.dialog_mode = True mock_soco.volume = 19 mock_soco.get_battery_info.return_value = battery_info mock_soco.all_zones = [mock_soco] yield mock_soco @pytest.fixture(name="discover", autouse=True) def discover_fixture(soco): """Create a mock pysonos discover fixture.""" def do_callback(callback, **kwargs): callback(soco) return MagicMock() with patch("pysonos.discover_thread", side_effect=do_callback) as mock: yield mock @pytest.fixture(name="config") def config_fixture(): """Create hass config fixture.""" return {DOMAIN: {MP_DOMAIN: {CONF_HOSTS: ["192.168.42.1"]}}} @pytest.fixture(name="dummy_soco_service") def dummy_soco_service_fixture(): """Create dummy_soco_service fixture.""" service = Mock() service.subscribe = AsyncMock() return service @pytest.fixture(name="music_library") def music_library_fixture(): """Create music_library fixture.""" music_library = Mock() music_library.get_sonos_favorites.return_value = [] return music_library @pytest.fixture(name="alarmClock") def alarmClock_fixture(): """Create alarmClock fixture.""" alarmClock = Mock() alarmClock.subscribe = AsyncMock() alarmClock.ListAlarms.return_value = { "CurrentAlarmList": "" '' '' " " } return alarmClock @pytest.fixture(name="speaker_info") def speaker_info_fixture(): """Create speaker_info fixture.""" return { "zone_name": "Zone A", "model_name": "Model Name", "software_version": "49.2-64250", "mac_address": "00-11-22-33-44-55", "display_version": "13.1", } @pytest.fixture(name="battery_info") def battery_info_fixture(): """Create battery_info fixture.""" return { "Health": "GREEN", "Level": 100, "Temperature": "NORMAL", "PowerSource": "SONOS_CHARGING_RING", }