Paulus Schoutsen
dd2a90a31f
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: balloob <1444314+balloob@users.noreply.github.com>
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
"""Common fixtures for the Radio Frequency tests."""
|
|
|
|
from typing import override
|
|
|
|
import pytest
|
|
from rf_protocols import ModulationType, RadioFrequencyCommand
|
|
|
|
from homeassistant.components.radio_frequency import (
|
|
DATA_COMPONENT,
|
|
RadioFrequencyTransmitterEntity,
|
|
)
|
|
from homeassistant.components.radio_frequency.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
|
|
@pytest.fixture
|
|
async def init_integration(hass: HomeAssistant) -> None:
|
|
"""Set up the Radio Frequency integration for testing."""
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
class MockRadioFrequencyCommand(RadioFrequencyCommand):
|
|
"""Mock RF command for testing."""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
frequency: int = 433_920_000,
|
|
modulation: ModulationType = ModulationType.OOK,
|
|
repeat_count: int = 0,
|
|
) -> None:
|
|
"""Initialize mock command."""
|
|
super().__init__(
|
|
frequency=frequency, modulation=modulation, repeat_count=repeat_count
|
|
)
|
|
|
|
@override
|
|
def get_raw_timings(self) -> list[int]:
|
|
"""Return mock timings."""
|
|
return [350, -1050, 350, -350]
|
|
|
|
|
|
class MockRadioFrequencyEntity(RadioFrequencyTransmitterEntity):
|
|
"""Mock radio frequency entity for testing."""
|
|
|
|
_attr_has_entity_name = True
|
|
_attr_name = "Test RF transmitter"
|
|
|
|
def __init__(
|
|
self,
|
|
unique_id: str,
|
|
frequency_ranges: list[tuple[int, int]] | None = None,
|
|
) -> None:
|
|
"""Initialize mock entity."""
|
|
self._attr_unique_id = unique_id
|
|
self._frequency_ranges = (
|
|
[(433_000_000, 434_000_000)]
|
|
if frequency_ranges is None
|
|
else frequency_ranges
|
|
)
|
|
self.send_command_calls: list[RadioFrequencyCommand] = []
|
|
|
|
@property
|
|
def supported_frequency_ranges(self) -> list[tuple[int, int]]:
|
|
"""Return supported frequency ranges."""
|
|
return self._frequency_ranges
|
|
|
|
async def async_send_command(self, command: RadioFrequencyCommand) -> None:
|
|
"""Mock send command."""
|
|
self.send_command_calls.append(command)
|
|
|
|
|
|
@pytest.fixture
|
|
async def mock_rf_entity(
|
|
hass: HomeAssistant, init_integration: None
|
|
) -> MockRadioFrequencyEntity:
|
|
"""Return a mock radio frequency entity."""
|
|
entity = MockRadioFrequencyEntity("test_rf_transmitter")
|
|
component = hass.data[DATA_COMPONENT]
|
|
await component.async_add_entities([entity])
|
|
return entity
|