1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00

Fix recorder fixture typing (#136174)

This commit is contained in:
Marc Mueller
2025-01-22 05:18:05 +01:00
committed by GitHub
parent ffcb4d676b
commit f822fd82bb
27 changed files with 100 additions and 87 deletions

View File

@@ -102,6 +102,7 @@ from .typing import (
MqttMockHAClient,
MqttMockHAClientGenerator,
MqttMockPahoClient,
RecorderInstanceContextManager,
RecorderInstanceGenerator,
WebSocketGenerator,
)
@@ -1536,7 +1537,7 @@ async def async_test_recorder(
enable_migrate_event_type_ids: bool,
enable_migrate_entity_ids: bool,
enable_migrate_event_ids: bool,
) -> AsyncGenerator[RecorderInstanceGenerator]:
) -> AsyncGenerator[RecorderInstanceContextManager]:
"""Yield context manager to setup recorder instance."""
# pylint: disable-next=import-outside-toplevel
from homeassistant.components import recorder
@@ -1702,7 +1703,7 @@ async def async_test_recorder(
@pytest.fixture
async def async_setup_recorder_instance(
async_test_recorder: RecorderInstanceGenerator,
async_test_recorder: RecorderInstanceContextManager,
) -> AsyncGenerator[RecorderInstanceGenerator]:
"""Yield callable to setup recorder instance."""
@@ -1715,7 +1716,7 @@ async def async_setup_recorder_instance(
expected_setup_result: bool = True,
wait_recorder: bool = True,
wait_recorder_setup: bool = True,
) -> AsyncGenerator[recorder.Recorder]:
) -> recorder.Recorder:
"""Set up and return recorder instance."""
return await stack.enter_async_context(
@@ -1734,7 +1735,7 @@ async def async_setup_recorder_instance(
@pytest.fixture
async def recorder_mock(
recorder_config: dict[str, Any] | None,
async_test_recorder: RecorderInstanceGenerator,
async_test_recorder: RecorderInstanceContextManager,
hass: HomeAssistant,
) -> AsyncGenerator[recorder.Recorder]:
"""Fixture with in-memory recorder."""