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
|
from __future__ import annotations
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
from collections import OrderedDict
|
||||||
from collections.abc import AsyncGenerator, Generator, Mapping, Sequence
|
from collections.abc import AsyncGenerator, Generator, Mapping, Sequence
|
||||||
from contextlib import asynccontextmanager, contextmanager
|
from contextlib import asynccontextmanager, contextmanager
|
||||||
from datetime import UTC, datetime, timedelta
|
from datetime import UTC, datetime, timedelta
|
||||||
|
@ -620,6 +621,27 @@ def mock_registry(
|
||||||
return 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(
|
def mock_device_registry(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
mock_entries: dict[str, dr.DeviceEntry] | None = None,
|
mock_entries: dict[str, dr.DeviceEntry] | None = None,
|
||||||
|
|
Loading…
Add table
Reference in a new issue