From 4418b0a43e02b0f6adcab5889822ce93f08b21ea Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Fri, 17 Dec 2021 07:45:39 +0000
Subject: [PATCH] Use DeviceClass Enums in gogogate2 tests (#62137)

---
 tests/components/gogogate2/test_cover.py  | 7 +++----
 tests/components/gogogate2/test_sensor.py | 7 +++----
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/tests/components/gogogate2/test_cover.py b/tests/components/gogogate2/test_cover.py
index d3507283426..9f36dbc9008 100644
--- a/tests/components/gogogate2/test_cover.py
+++ b/tests/components/gogogate2/test_cover.py
@@ -16,11 +16,10 @@ from ismartgate.common import (
 )
 
 from homeassistant.components.cover import (
-    DEVICE_CLASS_GARAGE,
-    DEVICE_CLASS_GATE,
     DOMAIN as COVER_DOMAIN,
     SUPPORT_CLOSE,
     SUPPORT_OPEN,
+    CoverDeviceClass,
 )
 from homeassistant.components.gogogate2.const import (
     DEVICE_TYPE_GOGOGATE2,
@@ -282,11 +281,11 @@ async def test_availability(ismartgateapi_mock, hass: HomeAssistant) -> None:
     assert hass.states.get("cover.door1")
     assert (
         hass.states.get("cover.door1").attributes[ATTR_DEVICE_CLASS]
-        == DEVICE_CLASS_GARAGE
+        == CoverDeviceClass.GARAGE
     )
     assert (
         hass.states.get("cover.door2").attributes[ATTR_DEVICE_CLASS]
-        == DEVICE_CLASS_GATE
+        == CoverDeviceClass.GATE
     )
 
     api.async_info.side_effect = Exception("Error")
diff --git a/tests/components/gogogate2/test_sensor.py b/tests/components/gogogate2/test_sensor.py
index 129fcac504f..8df88b2b4b7 100644
--- a/tests/components/gogogate2/test_sensor.py
+++ b/tests/components/gogogate2/test_sensor.py
@@ -17,6 +17,7 @@ from ismartgate.common import (
 )
 
 from homeassistant.components.gogogate2.const import DEVICE_TYPE_ISMARTGATE, DOMAIN
+from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.config_entries import SOURCE_USER
 from homeassistant.const import (
     ATTR_DEVICE_CLASS,
@@ -25,8 +26,6 @@ from homeassistant.const import (
     CONF_IP_ADDRESS,
     CONF_PASSWORD,
     CONF_USERNAME,
-    DEVICE_CLASS_BATTERY,
-    DEVICE_CLASS_TEMPERATURE,
     STATE_UNAVAILABLE,
     STATE_UNKNOWN,
 )
@@ -296,11 +295,11 @@ async def test_availability(ismartgateapi_mock, hass: HomeAssistant) -> None:
     assert hass.states.get("sensor.door3_temperature") is None
     assert (
         hass.states.get("sensor.door1_battery").attributes[ATTR_DEVICE_CLASS]
-        == DEVICE_CLASS_BATTERY
+        == SensorDeviceClass.BATTERY
     )
     assert (
         hass.states.get("sensor.door1_temperature").attributes[ATTR_DEVICE_CLASS]
-        == DEVICE_CLASS_TEMPERATURE
+        == SensorDeviceClass.TEMPERATURE
     )
     assert (
         hass.states.get("sensor.door1_temperature").attributes[ATTR_UNIT_OF_MEASUREMENT]