jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-12 20:40:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

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