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

Update tests p-s to use async_get() instead of async_get_registry() (#47654)

Browse source
This commit is contained in:
Erik Montnemery 2021-03-09 14:28:32 +01:00 • committed by GitHub
parent b3fecb1c95
commit ba2978c693
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
29 changed files with 103 additions and 86 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/components/smartthings/test_fan.py
View file

@ -22,6 +22,7 @@ from homeassistant.const import (
ATTR_SUPPORTED_FEATURES,
STATE_UNAVAILABLE,
)
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
from .conftest import setup_platform
@ -59,8 +60,8 @@ async def test_entity_and_device_attributes(hass, device_factory):
)
# Act
await setup_platform(hass, FAN_DOMAIN, devices=[device])
entity_registry = await hass.helpers.entity_registry.async_get_registry()
device_registry = await hass.helpers.device_registry.async_get_registry()
entity_registry = er.async_get(hass)
device_registry = dr.async_get(hass)
# Assert
entry = entity_registry.async_get("fan.fan_1")
assert entry