From 4a7a3b046968dffc8b174cdc08bb120bff18448d Mon Sep 17 00:00:00 2001 From: Dave T <17680170+davet2001@users.noreply.github.com> Date: Thu, 16 Dec 2021 20:05:12 +0000 Subject: [PATCH] Use DeviceClass Enums in advantage_air tests (#61986) --- tests/components/advantage_air/test_cover.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/components/advantage_air/test_cover.py b/tests/components/advantage_air/test_cover.py index 363db076ada..2868179b3ee 100644 --- a/tests/components/advantage_air/test_cover.py +++ b/tests/components/advantage_air/test_cover.py @@ -8,11 +8,11 @@ from homeassistant.components.advantage_air.const import ( ) from homeassistant.components.cover import ( ATTR_POSITION, - DEVICE_CLASS_DAMPER, DOMAIN as COVER_DOMAIN, SERVICE_CLOSE_COVER, SERVICE_OPEN_COVER, SERVICE_SET_COVER_POSITION, + CoverDeviceClass, ) from homeassistant.const import ATTR_ENTITY_ID, STATE_OPEN from homeassistant.helpers import entity_registry as er @@ -49,7 +49,7 @@ async def test_cover_async_setup_entry(hass, aioclient_mock): state = hass.states.get(entity_id) assert state assert state.state == STATE_OPEN - assert state.attributes.get("device_class") == DEVICE_CLASS_DAMPER + assert state.attributes.get("device_class") == CoverDeviceClass.DAMPER assert state.attributes.get("current_position") == 100 entry = registry.async_get(entity_id)