diff --git a/homeassistant/components/rituals_perfume_genie/binary_sensor.py b/homeassistant/components/rituals_perfume_genie/binary_sensor.py index a529ff3dca6..be251b86311 100644 --- a/homeassistant/components/rituals_perfume_genie/binary_sensor.py +++ b/homeassistant/components/rituals_perfume_genie/binary_sensor.py @@ -8,6 +8,7 @@ from homeassistant.components.binary_sensor import ( BinarySensorEntity, ) from homeassistant.config_entries import ConfigEntry +from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -38,6 +39,7 @@ class DiffuserBatteryChargingBinarySensor(DiffuserEntity, BinarySensorEntity): """Representation of a diffuser battery charging binary sensor.""" _attr_device_class = DEVICE_CLASS_BATTERY_CHARGING + _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC def __init__( self, diffuser: Diffuser, coordinator: RitualsDataUpdateCoordinator diff --git a/homeassistant/components/rituals_perfume_genie/select.py b/homeassistant/components/rituals_perfume_genie/select.py index eac95ee5ed4..9907e1d227f 100644 --- a/homeassistant/components/rituals_perfume_genie/select.py +++ b/homeassistant/components/rituals_perfume_genie/select.py @@ -5,7 +5,7 @@ from pyrituals import Diffuser from homeassistant.components.select import SelectEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.const import AREA_SQUARE_METERS +from homeassistant.const import AREA_SQUARE_METERS, ENTITY_CATEGORY_CONFIG from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -36,6 +36,7 @@ class DiffuserRoomSize(DiffuserEntity, SelectEntity): _attr_icon = "mdi:ruler-square" _attr_unit_of_measurement = AREA_SQUARE_METERS _attr_options = ["15", "30", "60", "100"] + _attr_entity_category = ENTITY_CATEGORY_CONFIG def __init__( self, diffuser: Diffuser, coordinator: RitualsDataUpdateCoordinator diff --git a/homeassistant/components/rituals_perfume_genie/sensor.py b/homeassistant/components/rituals_perfume_genie/sensor.py index 878fb2f1a86..5299539bcd9 100644 --- a/homeassistant/components/rituals_perfume_genie/sensor.py +++ b/homeassistant/components/rituals_perfume_genie/sensor.py @@ -8,6 +8,7 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( DEVICE_CLASS_BATTERY, DEVICE_CLASS_SIGNAL_STRENGTH, + ENTITY_CATEGORY_DIAGNOSTIC, PERCENTAGE, ) from homeassistant.core import HomeAssistant @@ -92,6 +93,7 @@ class DiffuserBatterySensor(DiffuserEntity, SensorEntity): _attr_device_class = DEVICE_CLASS_BATTERY _attr_native_unit_of_measurement = PERCENTAGE + _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC def __init__( self, diffuser: Diffuser, coordinator: RitualsDataUpdateCoordinator @@ -110,6 +112,7 @@ class DiffuserWifiSensor(DiffuserEntity, SensorEntity): _attr_device_class = DEVICE_CLASS_SIGNAL_STRENGTH _attr_native_unit_of_measurement = PERCENTAGE + _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC def __init__( self, diffuser: Diffuser, coordinator: RitualsDataUpdateCoordinator diff --git a/tests/components/rituals_perfume_genie/test_binary_sensor.py b/tests/components/rituals_perfume_genie/test_binary_sensor.py index f2e499655ca..769384dbbf8 100644 --- a/tests/components/rituals_perfume_genie/test_binary_sensor.py +++ b/tests/components/rituals_perfume_genie/test_binary_sensor.py @@ -1,7 +1,7 @@ """Tests for the Rituals Perfume Genie binary sensor platform.""" from homeassistant.components.binary_sensor import DEVICE_CLASS_BATTERY_CHARGING from homeassistant.components.rituals_perfume_genie.binary_sensor import CHARGING_SUFFIX -from homeassistant.const import ATTR_DEVICE_CLASS, STATE_ON +from homeassistant.const import ATTR_DEVICE_CLASS, ENTITY_CATEGORY_DIAGNOSTIC, STATE_ON from homeassistant.core import HomeAssistant from homeassistant.helpers import entity_registry @@ -28,3 +28,4 @@ async def test_binary_sensors(hass: HomeAssistant) -> None: entry = registry.async_get("binary_sensor.genie_battery_charging") assert entry assert entry.unique_id == f"{hublot}{CHARGING_SUFFIX}" + assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC diff --git a/tests/components/rituals_perfume_genie/test_select.py b/tests/components/rituals_perfume_genie/test_select.py index fb159166fb7..5c6d344adca 100644 --- a/tests/components/rituals_perfume_genie/test_select.py +++ b/tests/components/rituals_perfume_genie/test_select.py @@ -9,6 +9,7 @@ from homeassistant.const import ( AREA_SQUARE_METERS, ATTR_ENTITY_ID, ATTR_ICON, + ENTITY_CATEGORY_CONFIG, SERVICE_SELECT_OPTION, ) from homeassistant.core import HomeAssistant @@ -36,6 +37,7 @@ async def test_select_entity(hass: HomeAssistant) -> None: assert entry assert entry.unique_id == f"{diffuser.hublot}{ROOM_SIZE_SUFFIX}" assert entry.unit_of_measurement == AREA_SQUARE_METERS + assert entry.entity_category == ENTITY_CATEGORY_CONFIG async def test_select_option(hass: HomeAssistant) -> None: diff --git a/tests/components/rituals_perfume_genie/test_sensor.py b/tests/components/rituals_perfume_genie/test_sensor.py index 2c72d429a99..a76cf0d6c46 100644 --- a/tests/components/rituals_perfume_genie/test_sensor.py +++ b/tests/components/rituals_perfume_genie/test_sensor.py @@ -11,6 +11,7 @@ from homeassistant.const import ( ATTR_UNIT_OF_MEASUREMENT, DEVICE_CLASS_BATTERY, DEVICE_CLASS_SIGNAL_STRENGTH, + ENTITY_CATEGORY_DIAGNOSTIC, PERCENTAGE, ) from homeassistant.core import HomeAssistant @@ -59,6 +60,7 @@ async def test_sensors_diffuser_v1_battery_cartridge(hass: HomeAssistant) -> Non entry = registry.async_get("sensor.genie_battery") assert entry assert entry.unique_id == f"{hublot}{BATTERY_SUFFIX}" + assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC state = hass.states.get("sensor.genie_wifi") assert state @@ -69,6 +71,7 @@ async def test_sensors_diffuser_v1_battery_cartridge(hass: HomeAssistant) -> Non entry = registry.async_get("sensor.genie_wifi") assert entry assert entry.unique_id == f"{hublot}{WIFI_SUFFIX}" + assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC async def test_sensors_diffuser_v2_no_battery_no_cartridge(hass: HomeAssistant) -> None: