From 84dad5d6785790f426e4c3bff8d97fa526f99b88 Mon Sep 17 00:00:00 2001
From: Dave T <17680170+davet2001@users.noreply.github.com>
Date: Thu, 16 Dec 2021 20:22:04 +0000
Subject: [PATCH] Use DeviceClass Enums in brother tests (#62110)

---
 tests/components/brother/test_sensor.py | 38 ++++++++++++-------------
 1 file changed, 19 insertions(+), 19 deletions(-)

diff --git a/tests/components/brother/test_sensor.py b/tests/components/brother/test_sensor.py
index d0eb0819303..2ffaf3a2a23 100644
--- a/tests/components/brother/test_sensor.py
+++ b/tests/components/brother/test_sensor.py
@@ -8,14 +8,14 @@ from homeassistant.components.brother.sensor import UNIT_PAGES
 from homeassistant.components.sensor import (
     ATTR_STATE_CLASS,
     DOMAIN as SENSOR_DOMAIN,
-    STATE_CLASS_MEASUREMENT,
+    SensorDeviceClass,
+    SensorStateClass,
 )
 from homeassistant.const import (
     ATTR_DEVICE_CLASS,
     ATTR_ENTITY_ID,
     ATTR_ICON,
     ATTR_UNIT_OF_MEASUREMENT,
-    DEVICE_CLASS_TIMESTAMP,
     PERCENTAGE,
     STATE_UNAVAILABLE,
 )
@@ -67,7 +67,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "75"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_black_toner_remaining")
     assert entry
@@ -78,7 +78,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "10"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_cyan_toner_remaining")
     assert entry
@@ -89,7 +89,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "8"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_magenta_toner_remaining")
     assert entry
@@ -100,7 +100,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d-nozzle"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "2"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_yellow_toner_remaining")
     assert entry
@@ -113,7 +113,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_COUNTER) == 986
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "92"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_drum_remaining_life")
     assert entry
@@ -126,7 +126,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_COUNTER) == 1611
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "92"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_black_drum_remaining_life")
     assert entry
@@ -139,7 +139,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_COUNTER) == 1611
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "92"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_cyan_drum_remaining_life")
     assert entry
@@ -152,7 +152,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_COUNTER) == 1611
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "92"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_magenta_drum_remaining_life")
     assert entry
@@ -165,7 +165,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_COUNTER) == 1611
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "92"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_yellow_drum_remaining_life")
     assert entry
@@ -176,7 +176,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:water-outline"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "97"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_fuser_remaining_life")
     assert entry
@@ -187,7 +187,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:current-ac"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "97"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_belt_unit_remaining_life")
     assert entry
@@ -198,7 +198,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:printer-3d"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
     assert state.state == "98"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_pf_kit_1_remaining_life")
     assert entry
@@ -209,7 +209,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
     assert state.state == "986"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_page_counter")
     assert entry
@@ -220,7 +220,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
     assert state.state == "538"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_duplex_unit_pages_counter")
     assert entry
@@ -231,7 +231,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
     assert state.state == "709"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_b_w_counter")
     assert entry
@@ -242,7 +242,7 @@ async def test_sensors(hass):
     assert state.attributes.get(ATTR_ICON) == "mdi:file-document-outline"
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == UNIT_PAGES
     assert state.state == "902"
-    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
 
     entry = registry.async_get("sensor.hl_l2340dw_color_counter")
     assert entry
@@ -252,7 +252,7 @@ async def test_sensors(hass):
     assert state
     assert state.attributes.get(ATTR_ICON) is None
     assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_TIMESTAMP
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TIMESTAMP
     assert state.state == "2019-09-24T12:14:56+00:00"
     assert state.attributes.get(ATTR_STATE_CLASS) is None