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

Add entity category to UniFi sensors and switches (#58484)

Browse source
This commit is contained in:
Robert Svensson 2021-10-26 20:23:20 +02:00 • committed by GitHub
parent dd1154ad08
commit 47f6313e5b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/components/unifi/test_sensor.py
View file

@ -15,6 +15,8 @@ from homeassistant.components.unifi.const import (
CONF_TRACK_DEVICES,
DOMAIN as UNIFI_DOMAIN,
)
from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
import homeassistant.util.dt as dt_util
@ -74,6 +76,12 @@ async def test_bandwidth_sensors(hass, aioclient_mock, mock_unifi_websocket):
assert hass.states.get("sensor.wireless_client_rx").state == "2345.0"
assert hass.states.get("sensor.wireless_client_tx").state == "6789.0"
ent_reg = er.async_get(hass)
assert (
ent_reg.async_get("sensor.wired_client_rx").entity_category
== ENTITY_CATEGORY_DIAGNOSTIC
)
# Verify state update
wireless_client["rx_bytes"] = 3456000000
@ -179,6 +187,12 @@ async def test_uptime_sensors(
assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 1
assert hass.states.get("sensor.client1_uptime").state == "2021-01-01T01:00:00+00:00"
ent_reg = er.async_get(hass)
assert (
ent_reg.async_get("sensor.client1_uptime").entity_category
== ENTITY_CATEGORY_DIAGNOSTIC
)
# Verify normal new event doesn't change uptime
# 4 seconds has passed