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

16
tests/components/homematicip_cloud/test_device.py
View file

@ -41,8 +41,8 @@ async def test_hmip_remove_device(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON
assert hmip_device
device_registry = await dr.async_get_registry(hass)
entity_registry = await er.async_get_registry(hass)
device_registry = dr.async_get(hass)
entity_registry = er.async_get(hass)
pre_device_count = len(device_registry.devices)
pre_entity_count = len(entity_registry.entities)
@ -73,8 +73,8 @@ async def test_hmip_add_device(hass, default_mock_hap_factory, hmip_config_entry
assert ha_state.state == STATE_ON
assert hmip_device
device_registry = await dr.async_get_registry(hass)
entity_registry = await er.async_get_registry(hass)
device_registry = dr.async_get(hass)
entity_registry = er.async_get(hass)
pre_device_count = len(device_registry.devices)
pre_entity_count = len(entity_registry.entities)
@ -119,8 +119,8 @@ async def test_hmip_remove_group(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON
assert hmip_device
device_registry = await dr.async_get_registry(hass)
entity_registry = await er.async_get_registry(hass)
device_registry = dr.async_get(hass)
entity_registry = er.async_get(hass)
pre_device_count = len(device_registry.devices)
pre_entity_count = len(entity_registry.entities)
@ -257,12 +257,12 @@ async def test_hmip_multi_area_device(hass, default_mock_hap_factory):
assert ha_state
# get the entity
entity_registry = await er.async_get_registry(hass)
entity_registry = er.async_get(hass)
entity = entity_registry.async_get(ha_state.entity_id)
assert entity
# get the device
device_registry = await dr.async_get_registry(hass)
device_registry = dr.async_get(hass)
device = device_registry.async_get(entity.device_id)
assert device.name == "Wired Eingangsmodul – 32-fach"