Update tests t-z to use async_get() instead of async_get_registry() (#47655)

This commit is contained in:
Erik Montnemery 2021-03-09 14:25:03 +01:00 committed by GitHub
parent 78b21b1ad1
commit b3fecb1c95
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
34 changed files with 171 additions and 170 deletions

View file

@ -20,6 +20,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from tests.components.wled import init_integration
from tests.test_util.aiohttp import AiohttpClientMocker
@ -31,7 +32,7 @@ async def test_switch_state(
"""Test the creation and values of the WLED switches."""
await init_integration(hass, aioclient_mock)
entity_registry = await hass.helpers.entity_registry.async_get_registry()
entity_registry = er.async_get(hass)
state = hass.states.get("switch.wled_rgb_light_nightlight")
assert state