Revert "Remove unused test helper mock_area_registry" (#113453)
This commit is contained in:
parent
566cbc71a5
commit
3ed7a7166d
1 changed files with 22 additions and 0 deletions
|
@ -3,6 +3,7 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from collections import OrderedDict
|
||||
from collections.abc import AsyncGenerator, Generator, Mapping, Sequence
|
||||
from contextlib import asynccontextmanager, contextmanager
|
||||
from datetime import UTC, datetime, timedelta
|
||||
|
@ -620,6 +621,27 @@ def mock_registry(
|
|||
return registry
|
||||
|
||||
|
||||
def mock_area_registry(
|
||||
hass: HomeAssistant, mock_entries: dict[str, ar.AreaEntry] | None = None
|
||||
) -> ar.AreaRegistry:
|
||||
"""Mock the Area Registry.
|
||||
|
||||
This should only be used if you need to mock/re-stage a clean mocked
|
||||
area registry in your current hass object. It can be useful to,
|
||||
for example, pre-load the registry with items.
|
||||
|
||||
This mock will thus replace the existing registry in the running hass.
|
||||
|
||||
If you just need to access the existing registry, use the `area_registry`
|
||||
fixture instead.
|
||||
"""
|
||||
registry = ar.AreaRegistry(hass)
|
||||
registry.areas = mock_entries or OrderedDict()
|
||||
|
||||
hass.data[ar.DATA_REGISTRY] = registry
|
||||
return registry
|
||||
|
||||
|
||||
def mock_device_registry(
|
||||
hass: HomeAssistant,
|
||||
mock_entries: dict[str, dr.DeviceEntry] | None = None,
|
||||
|
|
Loading…
Add table
Reference in a new issue