evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add missing hass type hint in history and recorder tests (#124204)

Browse source
This commit is contained in:
epenet 2024-08-19 15:41:09 +02:00 • committed by GitHub
parent 5470d14a11
commit c76d68503a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 27 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/components/recorder/test_migrate.py
View file

@ -29,7 +29,7 @@ from homeassistant.components.recorder.db_schema import (
States,
)
from homeassistant.components.recorder.util import session_scope
from homeassistant.core import HomeAssistant
from homeassistant.core import HomeAssistant, State
from homeassistant.helpers import recorder as recorder_helper
import homeassistant.util.dt as dt_util
@ -47,7 +47,7 @@ async def mock_recorder_before_hass(
"""Set up recorder."""
def _get_native_states(hass, entity_id):
def _get_native_states(hass: HomeAssistant, entity_id: str) -> list[State]:
with session_scope(hass=hass, read_only=True) as session:
instance = recorder.get_instance(hass)
metadata_id = instance.states_meta_manager.get(entity_id, session, True)