Update nest library and switch events to async (#43583)
This commit is contained in:
parent
7214d6517a
commit
3dd14e05e3
7 changed files with 16 additions and 16 deletions
|
@ -3,7 +3,7 @@
|
|||
import time
|
||||
|
||||
from google_nest_sdm.device_manager import DeviceManager
|
||||
from google_nest_sdm.event import EventCallback, EventMessage
|
||||
from google_nest_sdm.event import AsyncEventCallback, EventMessage
|
||||
from google_nest_sdm.google_nest_subscriber import GoogleNestSubscriber
|
||||
|
||||
from homeassistant.components.nest import DOMAIN
|
||||
|
@ -61,7 +61,7 @@ class FakeSubscriber(GoogleNestSubscriber):
|
|||
self._device_manager = device_manager
|
||||
self._callback = None
|
||||
|
||||
def set_update_callback(self, callback: EventCallback):
|
||||
def set_update_callback(self, callback: AsyncEventCallback):
|
||||
"""Capture the callback set by Home Assistant."""
|
||||
self._callback = callback
|
||||
|
||||
|
@ -77,11 +77,11 @@ class FakeSubscriber(GoogleNestSubscriber):
|
|||
"""No-op to stop the subscriber."""
|
||||
return None
|
||||
|
||||
def receive_event(self, event_message: EventMessage):
|
||||
async def async_receive_event(self, event_message: EventMessage):
|
||||
"""Simulate a received pubsub message, invoked by tests."""
|
||||
# Update device state, then invoke HomeAssistant to refresh
|
||||
self._device_manager.handle_event(event_message)
|
||||
self._callback.handle_event(event_message)
|
||||
await self._device_manager.async_handle_event(event_message)
|
||||
await self._callback.async_handle_event(event_message)
|
||||
|
||||
|
||||
async def async_setup_sdm_platform(hass, platform, devices={}, structures={}):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue