diff --git a/homeassistant/components/twentemilieu/sensor.py b/homeassistant/components/twentemilieu/sensor.py index 43bfba2b017..f25b84ace15 100644 --- a/homeassistant/components/twentemilieu/sensor.py +++ b/homeassistant/components/twentemilieu/sensor.py @@ -40,6 +40,13 @@ class TwenteMilieuSensorDescription( SENSORS: tuple[TwenteMilieuSensorDescription, ...] = ( + TwenteMilieuSensorDescription( + key="tree", + waste_type=WasteType.TREE, + name="Christmas Tree Pickup", + icon="mdi:pine-tree", + device_class=SensorDeviceClass.DATE, + ), TwenteMilieuSensorDescription( key="Non-recyclable", waste_type=WasteType.NON_RECYCLABLE, diff --git a/tests/components/twentemilieu/conftest.py b/tests/components/twentemilieu/conftest.py index d540658787b..530e9723251 100644 --- a/tests/components/twentemilieu/conftest.py +++ b/tests/components/twentemilieu/conftest.py @@ -69,6 +69,7 @@ def mock_twentemilieu() -> Generator[None, MagicMock, None]: WasteType.ORGANIC: date(2021, 11, 2), WasteType.PACKAGES: date(2021, 11, 3), WasteType.PAPER: None, + WasteType.TREE: date(2022, 1, 6), } yield twentemilieu diff --git a/tests/components/twentemilieu/test_sensor.py b/tests/components/twentemilieu/test_sensor.py index 502bc2e3089..5f09018358e 100644 --- a/tests/components/twentemilieu/test_sensor.py +++ b/tests/components/twentemilieu/test_sensor.py @@ -22,6 +22,17 @@ async def test_waste_pickup_sensors( entity_registry = er.async_get(hass) device_registry = dr.async_get(hass) + state = hass.states.get("sensor.christmas_tree_pickup") + entry = entity_registry.async_get("sensor.christmas_tree_pickup") + assert entry + assert state + assert entry.unique_id == "twentemilieu_12345_tree" + assert state.state == "2022-01-06" + assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Christmas Tree Pickup" + assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.DATE + assert state.attributes.get(ATTR_ICON) == "mdi:pine-tree" + assert ATTR_UNIT_OF_MEASUREMENT not in state.attributes + state = hass.states.get("sensor.non_recyclable_waste_pickup") entry = entity_registry.async_get("sensor.non_recyclable_waste_pickup") assert entry