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:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user