Revert "Remove unused test helper mock_area_registry" (#113453)

This commit is contained in:
Jan Bouwhuis 2024-03-14 22:11:44 +01:00 committed by GitHub
parent 566cbc71a5
commit 3ed7a7166d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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,