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

Update tests p-s to use async_get() instead of async_get_registry() (#47654)

Browse source
This commit is contained in:
Erik Montnemery 2021-03-09 14:28:32 +01:00 • committed by GitHub
parent b3fecb1c95
commit ba2978c693
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
29 changed files with 103 additions and 86 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/components/person/test_init.py
View file

@ -22,7 +22,7 @@ from homeassistant.const import (
STATE_UNKNOWN,
)
from homeassistant.core import Context, CoreState, State
from homeassistant.helpers import collection, entity_registry
from homeassistant.helpers import collection, entity_registry as er
from homeassistant.setup import async_setup_component
from tests.common import mock_component, mock_restore_cache
@ -589,7 +589,7 @@ async def test_ws_delete(hass, hass_ws_client, storage_setup):
assert resp["success"]
assert len(hass.states.async_entity_ids("person")) == 0
ent_reg = await hass.helpers.entity_registry.async_get_registry()
ent_reg = er.async_get(hass)
assert not ent_reg.async_is_registered("person.tracked_person")
@ -681,7 +681,7 @@ async def test_update_person_when_user_removed(
async def test_removing_device_tracker(hass, storage_setup):
"""Test we automatically remove removed device trackers."""
storage_collection = hass.data[DOMAIN][1]
reg = await entity_registry.async_get_registry(hass)
reg = er.async_get(hass)
entry = reg.async_get_or_create(
"device_tracker", "mobile_app", "bla", suggested_object_id="pixel"
)