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