diff --git a/homeassistant/components/sun/sensor.py b/homeassistant/components/sun/sensor.py index 8a390ea2a27..344e0c2179e 100644 --- a/homeassistant/components/sun/sensor.py +++ b/homeassistant/components/sun/sensor.py @@ -13,7 +13,7 @@ from homeassistant.components.sensor import ( SensorStateClass, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import DEGREE +from homeassistant.const import DEGREE, EntityCategory from homeassistant.core import HomeAssistant from homeassistant.helpers.device_registry import DeviceEntryType from homeassistant.helpers.entity import DeviceInfo @@ -118,6 +118,7 @@ class SunSensor(SensorEntity): """Representation of a Sun Sensor.""" _attr_has_entity_name = True + _attr_entity_category = EntityCategory.DIAGNOSTIC entity_description: SunSensorEntityDescription def __init__( diff --git a/tests/components/sun/test_sensor.py b/tests/components/sun/test_sensor.py index 13f4fd0d62b..38453569269 100644 --- a/tests/components/sun/test_sensor.py +++ b/tests/components/sun/test_sensor.py @@ -6,6 +6,7 @@ import astral.sun from freezegun import freeze_time from homeassistant.components import sun +from homeassistant.const import EntityCategory from homeassistant.core import HomeAssistant import homeassistant.helpers.entity_registry as er from homeassistant.setup import async_setup_component @@ -98,4 +99,6 @@ async def test_setting_rising(hass: HomeAssistant) -> None: entity_reg = er.async_get(hass) entity = entity_reg.async_get("sensor.sun_next_dawn") + assert entity + assert entity.entity_category is EntityCategory.DIAGNOSTIC assert entity.unique_id == f"{entry_ids[0].entry_id}-next_dawn"