Reduce overhead to lookup items in the entity and device registry (#94568)
This commit is contained in:
parent
de62082605
commit
a0c023d5cb
4 changed files with 35 additions and 6 deletions
|
@ -509,6 +509,7 @@ def mock_registry(
|
|||
if mock_entries is None:
|
||||
mock_entries = {}
|
||||
registry.entities = er.EntityRegistryItems()
|
||||
registry._entities_data = registry.entities.data
|
||||
for key, entry in mock_entries.items():
|
||||
registry.entities[key] = entry
|
||||
|
||||
|
@ -554,6 +555,7 @@ def mock_device_registry(
|
|||
"""
|
||||
registry = dr.DeviceRegistry(hass)
|
||||
registry.devices = dr.DeviceRegistryItems()
|
||||
registry._device_data = registry.devices.data
|
||||
if mock_entries is None:
|
||||
mock_entries = {}
|
||||
for key, entry in mock_entries.items():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue