diff --git a/homeassistant/components/twentemilieu/sensor.py b/homeassistant/components/twentemilieu/sensor.py index e75d7f372ac..43bfba2b017 100644 --- a/homeassistant/components/twentemilieu/sensor.py +++ b/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, ), ) diff --git a/tests/components/twentemilieu/test_sensor.py b/tests/components/twentemilieu/test_sensor.py index 62acbe686a9..502bc2e3089 100644 --- a/tests/components/twentemilieu/test_sensor.py +++ b/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