evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add entity_category to Rituals Entities (#59756)

Browse source
This commit is contained in:
Milan Meulemans 2021-11-16 21:37:28 +01:00 • committed by GitHub
parent e1e6925097
commit f8501ded0e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
homeassistant/components/rituals_perfume_genie/binary_sensor.py
View file

@ -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

3
homeassistant/components/rituals_perfume_genie/select.py
View file

@ -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

3
homeassistant/components/rituals_perfume_genie/sensor.py
View file

@ -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

3
tests/components/rituals_perfume_genie/test_binary_sensor.py
View file

@ -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

2
tests/components/rituals_perfume_genie/test_select.py
View file

@ -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:

3
tests/components/rituals_perfume_genie/test_sensor.py
View file

@ -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: