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

Use DeviceClass Enums in devolo_home_network tests (#62117)

Browse source
This commit is contained in:
Dave T 2021-12-20 17:09:28 +00:00 • committed by GitHub
parent bae82d76b5
commit ac0f655fd6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/components/devolo_home_network/test_sensor.py
View file

@ -8,10 +8,11 @@ from homeassistant.components.devolo_home_network.const import (
LONG_UPDATE_INTERVAL,
SHORT_UPDATE_INTERVAL,
)
from homeassistant.components.sensor import DOMAIN, STATE_CLASS_MEASUREMENT
from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC, STATE_UNAVAILABLE
from homeassistant.components.sensor import DOMAIN, SensorStateClass
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry
from homeassistant.helpers.entity import EntityCategory
from homeassistant.util import dt
from . import configure_integration
@ -47,7 +48,7 @@ async def test_update_connected_wifi_clients(hass: HomeAssistant):
state = hass.states.get(state_key)
assert state is not None
assert state.state == "1"
assert state.attributes["state_class"] == STATE_CLASS_MEASUREMENT
assert state.attributes["state_class"] == SensorStateClass.MEASUREMENT
# Emulate device failure
with patch(
@ -89,7 +90,7 @@ async def test_update_neighboring_wifi_networks(hass: HomeAssistant):
assert state.state == "1"
er = entity_registry.async_get(hass)
assert er.async_get(state_key).entity_category == ENTITY_CATEGORY_DIAGNOSTIC
assert er.async_get(state_key).entity_category is EntityCategory.DIAGNOSTIC
# Emulate device failure
with patch(
@ -131,7 +132,7 @@ async def test_update_connected_plc_devices(hass: HomeAssistant):
assert state.state == "1"
er = entity_registry.async_get(hass)
assert er.async_get(state_key).entity_category == ENTITY_CATEGORY_DIAGNOSTIC
assert er.async_get(state_key).entity_category is EntityCategory.DIAGNOSTIC
# Emulate device failure
with patch(