1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00

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
34 changed files with 171 additions and 170 deletions

View File

@@ -16,7 +16,7 @@ from homeassistant.components.unifi.const import (
DOMAIN as UNIFI_DOMAIN,
)
from homeassistant.components.unifi.switch import POE_SWITCH
from homeassistant.helpers import entity_registry
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
from .test_controller import (
@@ -799,7 +799,7 @@ async def test_restore_client_succeed(hass, aioclient_mock):
entry_id=1,
)
registry = await entity_registry.async_get_registry(hass)
registry = er.async_get(hass)
registry.async_get_or_create(
SWITCH_DOMAIN,
UNIFI_DOMAIN,
@@ -891,7 +891,7 @@ async def test_restore_client_no_old_state(hass, aioclient_mock):
entry_id=1,
)
registry = await entity_registry.async_get_registry(hass)
registry = er.async_get(hass)
registry.async_get_or_create(
SWITCH_DOMAIN,
UNIFI_DOMAIN,