evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup EntityCategory in tests (#78808)

Browse source
This commit is contained in:
epenet 2022-09-20 16:49:44 +02:00 • committed by GitHub
parent fe747601ff
commit 25b1dfb53a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 13 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/components/zha/test_button.py
View file

@ -24,12 +24,11 @@ from homeassistant.components.button import DOMAIN, SERVICE_PRESS, ButtonDeviceC
from homeassistant.const import (
ATTR_DEVICE_CLASS,
ATTR_ENTITY_ID,
ENTITY_CATEGORY_CONFIG,
ENTITY_CATEGORY_DIAGNOSTIC,
STATE_UNKNOWN,
Platform,
)
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity import EntityCategory
from .common import find_entity_id
from .conftest import SIG_EP_INPUT, SIG_EP_OUTPUT, SIG_EP_TYPE
@ -137,7 +136,7 @@ async def test_button(hass, contact_sensor):
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC
assert entry.entity_category == EntityCategory.DIAGNOSTIC
with patch(
"zigpy.zcl.Cluster.request",
@ -177,7 +176,7 @@ async def test_frost_unlock(hass, tuya_water_valve):
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.entity_category == ENTITY_CATEGORY_CONFIG
assert entry.entity_category == EntityCategory.CONFIG
with patch(
"zigpy.zcl.Cluster.request",