Noah Husby af41a41046
Add select entity to Cambridge Audio (#128527)
* Add select entity to Cambridge Audio

* Add select entity to Cambridge Audio

* Update test name
2024-10-16 19:57:10 +02:00

70 lines
2.3 KiB
Python

"""Cambridge Audio tests configuration."""
from collections.abc import Generator
from unittest.mock import Mock, patch
from aiostreammagic.models import Display, Info, NowPlaying, PlayState, Source, State
import pytest
from homeassistant.components.cambridge_audio.const import DOMAIN
from homeassistant.const import CONF_HOST
from tests.common import MockConfigEntry, load_fixture, load_json_array_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
"""Override async_setup_entry."""
with patch(
"homeassistant.components.cambridge_audio.async_setup_entry",
return_value=True,
) as mock_setup_entry:
yield mock_setup_entry
@pytest.fixture
def mock_stream_magic_client() -> Generator[AsyncMock]:
"""Mock an Cambridge Audio client."""
with (
patch(
"homeassistant.components.cambridge_audio.StreamMagicClient",
autospec=True,
) as mock_client,
patch(
"homeassistant.components.cambridge_audio.config_flow.StreamMagicClient",
new=mock_client,
),
):
client = mock_client.return_value
client.host = "192.168.20.218"
client.info = Info.from_json(load_fixture("get_info.json", DOMAIN))
client.sources = [
Source.from_dict(x)
for x in load_json_array_fixture("get_sources.json", DOMAIN)
]
client.state = State.from_json(load_fixture("get_state.json", DOMAIN))
client.play_state = PlayState.from_json(
load_fixture("get_play_state.json", DOMAIN)
)
client.now_playing = NowPlaying.from_json(
load_fixture("get_now_playing.json", DOMAIN)
)
client.display = Display.from_json(load_fixture("get_display.json", DOMAIN))
client.is_connected = Mock(return_value=True)
client.position_last_updated = client.play_state.position
client.unregister_state_update_callbacks = AsyncMock(return_value=True)
yield client
@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
"""Mock a config entry."""
return MockConfigEntry(
domain=DOMAIN,
title="Cambridge Audio CXNv2",
data={CONF_HOST: "192.168.20.218"},
unique_id="0020c2d8",
)