evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle single state requests from ESPHome (#124660)

Browse source 
Co-authored-by: J. Nick Koston <nick@koston.org>
This commit is contained in:
Jesse Hills 2024-08-28 10:15:41 +12:00 • committed by GitHub
parent fa084143ef
commit e84d9e21f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 54 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/components/esphome/conftest.py
View file

@ -205,6 +205,7 @@ class MockESPHomeDevice:
self.home_assistant_state_subscription_callback: Callable[
[str, str | None], None
]
self.home_assistant_state_request_callback: Callable[[str, str | None], None]
self.voice_assistant_handle_start_callback: Callable[
[str, int, VoiceAssistantAudioSettings, str | None],
Coroutine[Any, Any, int | None],
@ -268,9 +269,11 @@ class MockESPHomeDevice:
def set_home_assistant_state_subscription_callback(
self,
on_state_sub: Callable[[str, str | None], None],
on_state_request: Callable[[str, str | None], None],
) -> None:
"""Set the state call callback."""
self.home_assistant_state_subscription_callback = on_state_sub
self.home_assistant_state_request_callback = on_state_request
def mock_home_assistant_state_subscription(
self, entity_id: str, attribute: str | None
@ -278,6 +281,12 @@ class MockESPHomeDevice:
"""Mock a state subscription."""
self.home_assistant_state_subscription_callback(entity_id, attribute)
def mock_home_assistant_state_request(
self, entity_id: str, attribute: str | None
) -> None:
"""Mock a state request."""
self.home_assistant_state_request_callback(entity_id, attribute)
def set_subscribe_voice_assistant_callbacks(
self,
handle_start: Callable[
@ -378,9 +387,12 @@ async def _mock_generic_device_entry(
def _subscribe_home_assistant_states(
on_state_sub: Callable[[str, str | None], None],
on_state_request: Callable[[str, str | None], None],
) -> None:
"""Subscribe to home assistant states."""
mock_device.set_home_assistant_state_subscription_callback(on_state_sub)
mock_device.set_home_assistant_state_subscription_callback(
on_state_sub, on_state_request
)
def _subscribe_voice_assistant(
*,