Add model tests

This commit is contained in:
Jesse Hills 2024-04-22 04:35:48 +00:00
parent abb36ad92d
commit 60dd1f4f3c
No known key found for this signature in database
GPG Key ID: BEAAE804EFD8E83A

View File

@ -16,6 +16,7 @@ from aioesphomeapi.api_pb2 import (
DateStateResponse, DateStateResponse,
DateTimeStateResponse, DateTimeStateResponse,
DeviceInfoResponse, DeviceInfoResponse,
EventResponse,
FanStateResponse, FanStateResponse,
HomeassistantServiceMap, HomeassistantServiceMap,
HomeassistantServiceResponse, HomeassistantServiceResponse,
@ -27,6 +28,7 @@ from aioesphomeapi.api_pb2 import (
ListEntitiesCoverResponse, ListEntitiesCoverResponse,
ListEntitiesDateResponse, ListEntitiesDateResponse,
ListEntitiesDateTimeResponse, ListEntitiesDateTimeResponse,
ListEntitiesEventResponse,
ListEntitiesFanResponse, ListEntitiesFanResponse,
ListEntitiesLightResponse, ListEntitiesLightResponse,
ListEntitiesLockResponse, ListEntitiesLockResponse,
@ -82,6 +84,8 @@ from aioesphomeapi.model import (
DateTimeInfo, DateTimeInfo,
DateTimeState, DateTimeState,
DeviceInfo, DeviceInfo,
EventInfo,
Event,
FanInfo, FanInfo,
FanState, FanState,
HomeassistantServiceCall, HomeassistantServiceCall,
@ -280,6 +284,8 @@ def test_api_version_ord():
(TimeState, TimeStateResponse), (TimeState, TimeStateResponse),
(DateTimeInfo, ListEntitiesDateTimeResponse), (DateTimeInfo, ListEntitiesDateTimeResponse),
(DateTimeState, DateTimeStateResponse), (DateTimeState, DateTimeStateResponse),
(EventInfo, ListEntitiesEventResponse),
(Event, EventResponse),
], ],
) )
def test_basic_pb_conversions(model, pb): def test_basic_pb_conversions(model, pb):