Only publish nest camera event messages once per thread and bump nest version (#61587)

This commit is contained in:
Allen Porter 2021-12-12 14:12:05 -08:00 committed by GitHub
parent 4ad90b33c9
commit cd001d0243
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 77 additions and 5 deletions

View file

@ -4,7 +4,7 @@
"config_flow": true,
"dependencies": ["ffmpeg", "http", "media_source"],
"documentation": "https://www.home-assistant.io/integrations/nest",
"requirements": ["python-nest==4.1.0", "google-nest-sdm==0.4.5"],
"requirements": ["python-nest==4.1.0", "google-nest-sdm==0.4.6"],
"codeowners": ["@allenporter"],
"quality_scale": "platinum",
"dhcp": [

View file

@ -747,7 +747,7 @@ google-cloud-pubsub==2.1.0
google-cloud-texttospeech==0.4.0
# homeassistant.components.nest
google-nest-sdm==0.4.5
google-nest-sdm==0.4.6
# homeassistant.components.google_travel_time
googlemaps==2.5.1

View file

@ -470,7 +470,7 @@ google-api-python-client==1.6.4
google-cloud-pubsub==2.1.0
# homeassistant.components.nest
google-nest-sdm==0.4.5
google-nest-sdm==0.4.6
# homeassistant.components.google_travel_time
googlemaps==2.5.1

View file

@ -62,7 +62,7 @@ class FakeSubscriber(GoogleNestSubscriber):
def set_update_callback(self, callback: Callable[[EventMessage], Awaitable[None]]):
"""Capture the callback set by Home Assistant."""
self._callback = callback
self._device_manager.set_update_callback(callback)
async def create_subscription(self):
"""Create the subscription."""
@ -93,7 +93,6 @@ class FakeSubscriber(GoogleNestSubscriber):
"""Simulate a received pubsub message, invoked by tests."""
# Update device state, then invoke HomeAssistant to refresh
await self._device_manager.async_handle_event(event_message)
await self._callback(event_message)
async def async_setup_sdm_platform(

View file

@ -4,6 +4,8 @@ These tests fake out the subscriber/devicemanager, and are not using a real
pubsub subscriber.
"""
import datetime
from google_nest_sdm.device import Device
from google_nest_sdm.event import EventMessage
@ -298,3 +300,74 @@ async def test_event_message_without_device_event(hass):
await hass.async_block_till_done()
assert len(events) == 0
async def test_doorbell_event_thread(hass):
"""Test a series of pubsub messages in the same thread."""
events = async_capture_events(hass, NEST_EVENT)
subscriber = await async_setup_devices(
hass,
"sdm.devices.types.DOORBELL",
traits={
"sdm.devices.traits.Info": {
"customName": "Front",
},
"sdm.devices.traits.CameraLiveStream": {},
"sdm.devices.traits.CameraClipPreview": {},
"sdm.devices.traits.CameraPerson": {},
},
)
registry = er.async_get(hass)
entry = registry.async_get("camera.front")
assert entry is not None
event_message_data = {
"eventId": "some-event-id-ignored",
"resourceUpdate": {
"name": DEVICE_ID,
"events": {
"sdm.devices.events.CameraMotion.Motion": {
"eventSessionId": EVENT_SESSION_ID,
"eventId": "n:1",
},
"sdm.devices.events.CameraClipPreview.ClipPreview": {
"eventSessionId": EVENT_SESSION_ID,
"previewUrl": "image-url-1",
},
},
},
"eventThreadId": "CjY5Y3VKaTZwR3o4Y19YbTVfMF...",
"resourcegroup": [DEVICE_ID],
}
# Publish message #1 that starts the event thread
timestamp1 = utcnow()
message_data_1 = event_message_data.copy()
message_data_1.update(
{
"timestamp": timestamp1.isoformat(timespec="seconds"),
"eventThreadState": "STARTED",
}
)
await subscriber.async_receive_event(EventMessage(message_data_1, auth=None))
# Publish message #1 that sends a no-op update to end the event thread
timestamp2 = timestamp1 + datetime.timedelta(seconds=1)
message_data_2 = event_message_data.copy()
message_data_2.update(
{
"timestamp": timestamp2.isoformat(timespec="seconds"),
"eventThreadState": "ENDED",
}
)
await subscriber.async_receive_event(EventMessage(message_data_2, auth=None))
await hass.async_block_till_done()
# The event is only published once
assert len(events) == 1
assert events[0].data == {
"device_id": entry.device_id,
"type": "camera_motion",
"timestamp": timestamp1.replace(microsecond=0),
"nest_event_id": EVENT_SESSION_ID,
}