jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Bootstrap to start registry loading early (#22321)

Browse source 
* Registries store directly in data on loading.

* Loading registries concurent with stage 1.

* Removed comments
This commit is contained in:
Penny Wood 2019-03-27 22:06:20 +08:00 • committed by GitHub
parent 4de2efd07f
commit 646c4a7137
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 67 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/common.py
View file

@ -327,11 +327,7 @@ def mock_registry(hass, mock_entries=None):
registry = entity_registry.EntityRegistry(hass)
registry.entities = mock_entries or OrderedDict()
async def _get_reg():
return registry
hass.data[entity_registry.DATA_REGISTRY] = \
hass.loop.create_task(_get_reg())
hass.data[entity_registry.DATA_REGISTRY] = registry
return registry
@ -340,11 +336,7 @@ def mock_area_registry(hass, mock_entries=None):
registry = area_registry.AreaRegistry(hass)
registry.areas = mock_entries or OrderedDict()
async def _get_reg():
return registry
hass.data[area_registry.DATA_REGISTRY] = \
hass.loop.create_task(_get_reg())
hass.data[area_registry.DATA_REGISTRY] = registry
return registry
@ -353,11 +345,7 @@ def mock_device_registry(hass, mock_entries=None):
registry = device_registry.DeviceRegistry(hass)
registry.devices = mock_entries or OrderedDict()
async def _get_reg():
return registry
hass.data[device_registry.DATA_REGISTRY] = \
hass.loop.create_task(_get_reg())
hass.data[device_registry.DATA_REGISTRY] = registry
return registry