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

3
tests/components/directv/test_media_player.py
View file

@ -53,6 +53,7 @@ from homeassistant.const import (
STATE_PLAYING,
STATE_UNAVAILABLE,
)
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.util import dt as dt_util
@ -167,7 +168,7 @@ async def test_unique_id(
"""Test unique id."""
await setup_integration(hass, aioclient_mock)
entity_registry = await hass.helpers.entity_registry.async_get_registry()
entity_registry = er.async_get(hass)
main = entity_registry.async_get(MAIN_ENTITY_ID)
assert main.device_class == DEVICE_CLASS_RECEIVER