From ac0f655fd65b57499e63ed7483ae60ed1a1d7638 Mon Sep 17 00:00:00 2001 From: Dave T <17680170+davet2001@users.noreply.github.com> Date: Mon, 20 Dec 2021 17:09:28 +0000 Subject: [PATCH] Use DeviceClass Enums in devolo_home_network tests (#62117) --- tests/components/devolo_home_network/test_sensor.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/components/devolo_home_network/test_sensor.py b/tests/components/devolo_home_network/test_sensor.py index 6ce4c36cdb5..a7b1be5f07d 100644 --- a/tests/components/devolo_home_network/test_sensor.py +++ b/tests/components/devolo_home_network/test_sensor.py @@ -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(