From 61f2f9d9ac54d30e06baeb31029aaaf82fa91034 Mon Sep 17 00:00:00 2001 From: Dave T <17680170+davet2001@users.noreply.github.com> Date: Thu, 16 Dec 2021 22:27:44 +0000 Subject: [PATCH] Use DeviceClass Enums in devolo_home_control tests (#62116) --- .../devolo_home_control/test_binary_sensor.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/components/devolo_home_control/test_binary_sensor.py b/tests/components/devolo_home_control/test_binary_sensor.py index 9b13220ad7c..4bce2ebd19e 100644 --- a/tests/components/devolo_home_control/test_binary_sensor.py +++ b/tests/components/devolo_home_control/test_binary_sensor.py @@ -4,14 +4,10 @@ from unittest.mock import patch import pytest from homeassistant.components.binary_sensor import DOMAIN -from homeassistant.const import ( - ENTITY_CATEGORY_DIAGNOSTIC, - STATE_OFF, - STATE_ON, - STATE_UNAVAILABLE, -) +from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE from homeassistant.core import HomeAssistant from homeassistant.helpers import entity_registry +from homeassistant.helpers.entity import EntityCategory from . import configure_integration from .mocks import ( @@ -42,9 +38,7 @@ async def test_binary_sensor(hass: HomeAssistant): state = hass.states.get(f"{DOMAIN}.test_2") assert state is not None er = entity_registry.async_get(hass) - assert ( - er.async_get(f"{DOMAIN}.test_2").entity_category == ENTITY_CATEGORY_DIAGNOSTIC - ) + assert er.async_get(f"{DOMAIN}.test_2").entity_category == EntityCategory.DIAGNOSTIC # Emulate websocket message: sensor turned on test_gateway.publisher.dispatch("Test", ("Test", True))