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

Update tests c-h to use registry async_get (#47652)

Browse source
This commit is contained in:
Erik Montnemery 2021-03-09 14:31:17 +01:00 • committed by GitHub
parent ba2978c693
commit 87e7cebd36
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
49 changed files with 165 additions and 138 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/components/hue/test_light.py
View file

@ -7,12 +7,7 @@ import aiohue
from homeassistant import config_entries
from homeassistant.components import hue
from homeassistant.components.hue import light as hue_light
from homeassistant.helpers.device_registry import (
async_get_registry as async_get_device_registry,
)
from homeassistant.helpers.entity_registry import (
async_get_registry as async_get_entity_registry,
)
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.util import color
HUE_LIGHT_NS = "homeassistant.components.light.hue."
@ -937,8 +932,8 @@ async def test_group_features(hass, mock_bridge):
group_3 = hass.states.get("light.dining_room")
assert group_3.attributes["supported_features"] == extended_color_feature
entity_registry = await async_get_entity_registry(hass)
device_registry = await async_get_device_registry(hass)
entity_registry = er.async_get(hass)
device_registry = dr.async_get(hass)
entry = entity_registry.async_get("light.hue_lamp_1")
device_entry = device_registry.async_get(entry.device_id)