Use device class enums in TwenteMilieu (#60686)

homeassistant/components/twentemilieu/sensor.py

@@ -6,9 +6,13 @@ from datetime import date
 
 from twentemilieu import WasteType
 
-from homeassistant.components.sensor import SensorEntity, SensorEntityDescription
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorEntityDescription,
+)
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_ID, DEVICE_CLASS_DATE
+from homeassistant.const import CONF_ID
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType
 from homeassistant.helpers.entity import DeviceInfo
@@ -41,28 +45,28 @@ SENSORS: tuple[TwenteMilieuSensorDescription, ...] = (
         waste_type=WasteType.NON_RECYCLABLE,
         name="Non-recyclable Waste Pickup",
         icon="mdi:delete-empty",
-        device_class=DEVICE_CLASS_DATE,
+        device_class=SensorDeviceClass.DATE,
     ),
     TwenteMilieuSensorDescription(
         key="Organic",
         waste_type=WasteType.ORGANIC,
         name="Organic Waste Pickup",
         icon="mdi:delete-empty",
-        device_class=DEVICE_CLASS_DATE,
+        device_class=SensorDeviceClass.DATE,
     ),
     TwenteMilieuSensorDescription(
         key="Paper",
         waste_type=WasteType.PAPER,
         name="Paper Waste Pickup",
         icon="mdi:delete-empty",
-        device_class=DEVICE_CLASS_DATE,
+        device_class=SensorDeviceClass.DATE,
     ),
     TwenteMilieuSensorDescription(
         key="Plastic",
         waste_type=WasteType.PACKAGES,
         name="Packages Waste Pickup",
         icon="mdi:delete-empty",
-        device_class=DEVICE_CLASS_DATE,
+        device_class=SensorDeviceClass.DATE,
     ),
 )
 

tests/components/twentemilieu/test_sensor.py

@@ -1,11 +1,11 @@
 """Tests for the Twente Milieu sensors."""
+from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.components.twentemilieu.const import DOMAIN
 from homeassistant.const import (
     ATTR_DEVICE_CLASS,
     ATTR_FRIENDLY_NAME,
     ATTR_ICON,
     ATTR_UNIT_OF_MEASUREMENT,
-    DEVICE_CLASS_DATE,
     STATE_UNKNOWN,
 )
 from homeassistant.core import HomeAssistant
@@ -29,7 +29,7 @@ async def test_waste_pickup_sensors(
     assert entry.unique_id == "twentemilieu_12345_Non-recyclable"
     assert state.state == "2021-11-01"
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Non-recyclable Waste Pickup"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_DATE
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.DATE
     assert state.attributes.get(ATTR_ICON) == "mdi:delete-empty"
     assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
 
@@ -40,7 +40,7 @@ async def test_waste_pickup_sensors(
     assert entry.unique_id == "twentemilieu_12345_Organic"
     assert state.state == "2021-11-02"
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Organic Waste Pickup"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_DATE
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.DATE
     assert state.attributes.get(ATTR_ICON) == "mdi:delete-empty"
     assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
 
@@ -51,7 +51,7 @@ async def test_waste_pickup_sensors(
     assert entry.unique_id == "twentemilieu_12345_Plastic"
     assert state.state == "2021-11-03"
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Packages Waste Pickup"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_DATE
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.DATE
     assert state.attributes.get(ATTR_ICON) == "mdi:delete-empty"
     assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes
 
@@ -62,7 +62,7 @@ async def test_waste_pickup_sensors(
     assert entry.unique_id == "twentemilieu_12345_Paper"
     assert state.state == STATE_UNKNOWN
     assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Paper Waste Pickup"
-    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_DATE
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.DATE
     assert state.attributes.get(ATTR_ICON) == "mdi:delete-empty"
     assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes