mirror of
https://github.com/home-assistant/core.git
synced 2026-05-18 06:20:17 +01:00
epenet
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
"""Common test tools for the Radio Frequency integration."""
|
|
|
|
from typing import NamedTuple, override
|
|
|
|
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
|
|
|
|
|
|
class MockCommand(NamedTuple):
|
|
"""Data structure to store calls to async_send_command."""
|
|
|
|
command: RadioFrequencyCommand
|
|
context: object | None
|
|
|
|
|
|
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[MockCommand] = []
|
|
|
|
@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(
|
|
MockCommand(command=command, context=self._context)
|
|
)
|
|
|
|
|
|
async def init_radio_frequency_fixture_helper(hass: HomeAssistant) -> None:
|
|
"""Set up the Radio Frequency integration for testing."""
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def mock_rf_entity_fixture_helper(
|
|
hass: HomeAssistant,
|
|
) -> MockRadioFrequencyEntity:
|
|
"""Add a mock radio frequency entity to the running integration."""
|
|
entity = MockRadioFrequencyEntity("test_rf_transmitter")
|
|
component = hass.data[DATA_COMPONENT]
|
|
await component.async_add_entities([entity])
|
|
return entity
|