Use enums in unifi tests (#62151)

This commit is contained in:
Robert Hillis 2021-12-17 05:12:43 -05:00 committed by GitHub
parent 499cae9900
commit 85c122f3e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 5 deletions

View File

@ -15,9 +15,9 @@ from homeassistant.components.unifi.const import (
CONF_TRACK_DEVICES, CONF_TRACK_DEVICES,
DOMAIN as UNIFI_DOMAIN, DOMAIN as UNIFI_DOMAIN,
) )
from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.entity import EntityCategory
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from .test_controller import setup_unifi_integration from .test_controller import setup_unifi_integration
@ -79,7 +79,7 @@ async def test_bandwidth_sensors(hass, aioclient_mock, mock_unifi_websocket):
ent_reg = er.async_get(hass) ent_reg = er.async_get(hass)
assert ( assert (
ent_reg.async_get("sensor.wired_client_rx").entity_category ent_reg.async_get("sensor.wired_client_rx").entity_category
== ENTITY_CATEGORY_DIAGNOSTIC is EntityCategory.DIAGNOSTIC
) )
# Verify state update # Verify state update
@ -190,7 +190,7 @@ async def test_uptime_sensors(
ent_reg = er.async_get(hass) ent_reg = er.async_get(hass)
assert ( assert (
ent_reg.async_get("sensor.client1_uptime").entity_category ent_reg.async_get("sensor.client1_uptime").entity_category
== ENTITY_CATEGORY_DIAGNOSTIC is EntityCategory.DIAGNOSTIC
) )
# Verify normal new event doesn't change uptime # Verify normal new event doesn't change uptime

View File

@ -16,9 +16,10 @@ from homeassistant.components.unifi.const import (
DOMAIN as UNIFI_DOMAIN, DOMAIN as UNIFI_DOMAIN,
) )
from homeassistant.components.unifi.switch import POE_SWITCH from homeassistant.components.unifi.switch import POE_SWITCH
from homeassistant.const import ENTITY_CATEGORY_CONFIG, STATE_OFF, STATE_ON from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.entity import EntityCategory
from .test_controller import ( from .test_controller import (
CONTROLLER_HOST, CONTROLLER_HOST,
@ -408,7 +409,7 @@ async def test_switches(hass, aioclient_mock):
"switch.block_client_1", "switch.block_client_1",
"switch.block_media_streaming", "switch.block_media_streaming",
): ):
assert ent_reg.async_get(entry_id).entity_category == ENTITY_CATEGORY_CONFIG assert ent_reg.async_get(entry_id).entity_category is EntityCategory.CONFIG
# Block and unblock client # Block and unblock client