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