Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

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
49 changed files with 165 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/components/deconz/test_services.py
View File

@@ -1,5 +1,4 @@
"""deCONZ service tests."""
from copy import deepcopy
from unittest.mock import Mock, patch
@@ -23,6 +22,7 @@ from homeassistant.components.deconz.services import (
async_unload_services,
)
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.entity_registry import async_entries_for_config_entry
from .test_gateway import (
@@ -245,7 +245,7 @@ async def test_remove_orphaned_entries_service(hass, aioclient_mock):
data = {CONF_BRIDGE_ID: BRIDGEID}
device_registry = await hass.helpers.device_registry.async_get_registry()
device_registry = dr.async_get(hass)
device = device_registry.async_get_or_create(
config_entry_id=config_entry.entry_id, identifiers={("mac", "123")}
)
@@ -261,7 +261,7 @@ async def test_remove_orphaned_entries_service(hass, aioclient_mock):
== 5 # Host, gateway, light, switch and orphan
)
entity_registry = await hass.helpers.entity_registry.async_get_registry()
entity_registry = er.async_get(hass)
entity_registry.async_get_or_create(
SENSOR_DOMAIN,
DECONZ_DOMAIN,