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

@ -7,6 +7,7 @@ import pywemo
from homeassistant.components.wemo import CONF_DISCOVERY, CONF_STATIC
from homeassistant.components.wemo.const import DOMAIN
from homeassistant.helpers import entity_registry as er
from homeassistant.setup import async_setup_component
MOCK_HOST = "127.0.0.1"
@ -72,7 +73,7 @@ async def async_wemo_entity_fixture(hass, pywemo_device):
)
await hass.async_block_till_done()
entity_registry = await hass.helpers.entity_registry.async_get_registry()
entity_registry = er.async_get(hass)
entity_entries = list(entity_registry.entities.values())
assert len(entity_entries) == 1