diff --git a/tests/components/recorder/test_history.py b/tests/components/recorder/test_history.py index 05542cbecb5..af846353467 100644 --- a/tests/components/recorder/test_history.py +++ b/tests/components/recorder/test_history.py @@ -806,7 +806,9 @@ async def test_get_significant_states_only_minimal_response( assert len(hist["sensor.test"]) == 3 -def record_states(hass) -> tuple[datetime, datetime, dict[str, list[State]]]: +def record_states( + hass: HomeAssistant, +) -> tuple[datetime, datetime, dict[str, list[State]]]: """Record some test states. We inject a bunch of state updates from media player, zone and diff --git a/tests/components/recorder/test_history_db_schema_32.py b/tests/components/recorder/test_history_db_schema_32.py index b778a3ff6a3..8a3e6a58ab3 100644 --- a/tests/components/recorder/test_history_db_schema_32.py +++ b/tests/components/recorder/test_history_db_schema_32.py @@ -568,7 +568,9 @@ async def test_get_significant_states_only( ) -def record_states(hass) -> tuple[datetime, datetime, dict[str, list[State]]]: +def record_states( + hass: HomeAssistant, +) -> tuple[datetime, datetime, dict[str, list[State]]]: """Record some test states. We inject a bunch of state updates from media player, zone and diff --git a/tests/components/recorder/test_history_db_schema_42.py b/tests/components/recorder/test_history_db_schema_42.py index 04490b88a28..083d4c0930e 100644 --- a/tests/components/recorder/test_history_db_schema_42.py +++ b/tests/components/recorder/test_history_db_schema_42.py @@ -808,7 +808,9 @@ async def test_get_significant_states_only_minimal_response( assert len(hist["sensor.test"]) == 3 -def record_states(hass) -> tuple[datetime, datetime, dict[str, list[State]]]: +def record_states( + hass: HomeAssistant, +) -> tuple[datetime, datetime, dict[str, list[State]]]: """Record some test states. We inject a bunch of state updates from media player, zone and