Add ButtonInfo to test file (#145)

This commit is contained in:
Jesse Hills 2021-11-29 17:06:41 +13:00 committed by GitHub
parent 8b37b0a7db
commit bdfa9d3c13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -13,6 +13,7 @@ from aioesphomeapi.api_pb2 import (
HomeassistantServiceResponse,
LightStateResponse,
ListEntitiesBinarySensorResponse,
ListEntitiesButtonResponse,
ListEntitiesClimateResponse,
ListEntitiesCoverResponse,
ListEntitiesFanResponse,
@ -37,6 +38,7 @@ from aioesphomeapi.model import (
APIVersion,
BinarySensorInfo,
BinarySensorState,
ButtonInfo,
ClimateInfo,
ClimatePreset,
ClimateState,
@ -217,6 +219,7 @@ def test_api_version_ord():
(HomeassistantServiceCall, HomeassistantServiceResponse),
(UserServiceArg, ListEntitiesServicesArgument),
(UserService, ListEntitiesServicesResponse),
(ButtonInfo, ListEntitiesButtonResponse),
],
)
def test_basic_pb_conversions(model, pb):