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

Use new enums in smartthings tests (#62708)

Browse source 
* Use new enums in smartthings tests

* Convert == to is
This commit is contained in:
Dave T 2021-12-23 22:14:10 +00:00 • committed by GitHub
parent caa2157b5b
commit 79ef4dea98
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/components/smartthings/test_binary_sensor.py
View file

@ -13,13 +13,10 @@ from homeassistant.components.binary_sensor import (
from homeassistant.components.smartthings import binary_sensor
from homeassistant.components.smartthings.const import DOMAIN, SIGNAL_SMARTTHINGS_UPDATE
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import (
ATTR_FRIENDLY_NAME,
ENTITY_CATEGORY_DIAGNOSTIC,
STATE_UNAVAILABLE,
)
from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNAVAILABLE
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.entity import EntityCategory
from .conftest import setup_platform
@ -125,4 +122,4 @@ async def test_entity_category(hass, device_factory):
entry = entity_registry.async_get("binary_sensor.tamper_sensor_2_tamper")
assert entry
assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC
assert entry.entity_category is EntityCategory.DIAGNOSTIC

4
tests/components/smartthings/test_sensor.py
View file

@ -17,13 +17,13 @@ from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import (
ATTR_FRIENDLY_NAME,
ATTR_UNIT_OF_MEASUREMENT,
ENTITY_CATEGORY_DIAGNOSTIC,
PERCENTAGE,
STATE_UNAVAILABLE,
STATE_UNKNOWN,
)
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
from homeassistant.helpers.entity import EntityCategory
from .conftest import setup_platform
@ -95,7 +95,7 @@ async def test_entity_and_device_attributes(hass, device_factory):
entry = entity_registry.async_get("sensor.sensor_1_battery")
assert entry
assert entry.unique_id == f"{device.device_id}.{Attribute.battery}"
assert entry.entity_category == ENTITY_CATEGORY_DIAGNOSTIC
assert entry.entity_category is EntityCategory.DIAGNOSTIC
entry = device_registry.async_get_device({(DOMAIN, device.device_id)})
assert entry
assert entry.configuration_url == "https://account.smartthings.com"