mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 21:06:19 +00:00
138 lines
3.6 KiB
Python
138 lines
3.6 KiB
Python
"""Fixtures for the Radio Browser integration tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Generator
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.radio_browser.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Return the default mocked config entry."""
|
|
return MockConfigEntry(
|
|
title="My Radios",
|
|
domain=DOMAIN,
|
|
data={},
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Mock setting up a config entry."""
|
|
with patch(
|
|
"homeassistant.components.radio_browser.async_setup_entry", return_value=True
|
|
) as mock_setup:
|
|
yield mock_setup
|
|
|
|
|
|
@pytest.fixture
|
|
async def init_integration(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> MockConfigEntry:
|
|
"""Set up the Radio Browser integration for testing."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return mock_config_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_countries():
|
|
"Generate mock countries for the countries method of the radios object."
|
|
|
|
class MockCountry:
|
|
"""Country Object for Radios."""
|
|
|
|
def __init__(self, code, name) -> None:
|
|
"""Initialize a mock country."""
|
|
self.code = code
|
|
self.name = name
|
|
self.favicon = "fake.png"
|
|
|
|
return [MockCountry("US", "United States")]
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_stations():
|
|
"Generate mock stations for the stations method of the radios object."
|
|
|
|
class MockStation:
|
|
"""Station object for Radios."""
|
|
|
|
def __init__(self, country_code, latitude, longitude, name, uuid) -> None:
|
|
"""Initialize a mock station."""
|
|
self.country_code = country_code
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.uuid = uuid
|
|
self.name = name
|
|
self.codec = "MP3"
|
|
self.favicon = "fake.png"
|
|
|
|
return [
|
|
MockStation(
|
|
country_code="US",
|
|
latitude=45.52000,
|
|
longitude=-122.63961,
|
|
name="Near Station 1",
|
|
uuid="1",
|
|
),
|
|
MockStation(
|
|
country_code="US",
|
|
latitude=None,
|
|
longitude=None,
|
|
name="Unknown location station",
|
|
uuid="2",
|
|
),
|
|
MockStation(
|
|
country_code="US",
|
|
latitude=47.57071,
|
|
longitude=-122.21148,
|
|
name="Moderate Far Station",
|
|
uuid="3",
|
|
),
|
|
MockStation(
|
|
country_code="US",
|
|
latitude=45.73943,
|
|
longitude=-121.51859,
|
|
name="Near Station 2",
|
|
uuid="4",
|
|
),
|
|
MockStation(
|
|
country_code="US",
|
|
latitude=44.99026,
|
|
longitude=-69.27804,
|
|
name="Really Far Station",
|
|
uuid="5",
|
|
),
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_radios(mock_countries, mock_stations):
|
|
"""Provide a radios mock object."""
|
|
radios = MagicMock()
|
|
radios.countries = AsyncMock(return_value=mock_countries)
|
|
radios.stations = AsyncMock(return_value=mock_stations)
|
|
return radios
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_radios(monkeypatch: pytest.MonkeyPatch, mock_radios):
|
|
"""Replace the radios object in the source with the mock object (with mock stations and countries)."""
|
|
|
|
def _patch(source):
|
|
monkeypatch.setattr(type(source), "radios", mock_radios)
|
|
|
|
return _patch
|