Dispatch unifiprotect websocket messages based on model (#119633)

This commit is contained in:
J. Nick Koston 2024-06-13 16:17:31 -05:00 committed by GitHub
parent de27f24a4c
commit 0c3a5ae5da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 91 additions and 36 deletions

View file

@ -6,7 +6,15 @@ from datetime import datetime, timedelta
from unittest.mock import Mock
import pytest
from uiprotect.data import NVR, Camera, Event, EventType, Sensor, SmartDetectObjectType
from uiprotect.data import (
NVR,
Camera,
Event,
EventType,
ModelType,
Sensor,
SmartDetectObjectType,
)
from uiprotect.data.nvr import EventMetadata, LicensePlateMetadata
from homeassistant.components.unifiprotect.const import DEFAULT_ATTRIBUTION
@ -438,6 +446,7 @@ async def test_sensor_update_alarm(
event_metadata = EventMetadata(sensor_id=sensor_all.id, alarm_type="smoke")
event = Event(
model=ModelType.EVENT,
id="test_event_id",
type=EventType.SENSOR_ALARM,
start=fixed_now - timedelta(seconds=1),
@ -521,6 +530,7 @@ async def test_camera_update_licenseplate(
license_plate=LicensePlateMetadata(name="ABCD1234", confidence_level=95)
)
event = Event(
model=ModelType.EVENT,
id="test_event_id",
type=EventType.SMART_DETECT,
start=fixed_now - timedelta(seconds=1),