jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-13 13:00:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in homekit and homekit_controller tests (#124213)

Browse Source 
Add missing hass type hint in homekit and homekit_controller tests
This commit is contained in:
epenet
2024-08-20 12:47:41 +02:00
committed by GitHub
parent be255613de
commit db92f29c00
24 changed files with 132 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/components/homekit/test_homekit.py
View File

@@ -73,6 +73,7 @@ from homeassistant.helpers.entityfilter import (
CONF_INCLUDE_DOMAINS,
CONF_INCLUDE_ENTITIES,
CONF_INCLUDE_ENTITY_GLOBS,
EntityFilter,
convert_filter,
)
from homeassistant.setup import async_setup_component
@@ -119,7 +120,13 @@ def patch_source_ip():
yield
def _mock_homekit(hass, entry, homekit_mode, entity_filter=None, devices=None):
def _mock_homekit(
hass: HomeAssistant,
entry: MockConfigEntry,
homekit_mode: str,
entity_filter: EntityFilter | None = None,
devices: list[str] | None = None,
) -> HomeKit:
return HomeKit(
hass=hass,
name=BRIDGE_NAME,
@@ -136,7 +143,7 @@ def _mock_homekit(hass, entry, homekit_mode, entity_filter=None, devices=None):
)
def _mock_homekit_bridge(hass, entry):
def _mock_homekit_bridge(hass: HomeAssistant, entry: MockConfigEntry) -> HomeKit:
homekit = _mock_homekit(hass, entry, HOMEKIT_MODE_BRIDGE)
homekit.driver = MagicMock()
homekit.iid_storage = MagicMock()