Add missing argument type hints to component tests (home-assistant#11…

…9671)

tests/components/accuweather/__init__.py

@@ -1,11 +1,12 @@
 """Tests for AccuWeather."""
 
 from homeassistant.components.accuweather.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
 
-async def init_integration(hass) -> MockConfigEntry:
+async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the AccuWeather integration in Home Assistant."""
     entry = MockConfigEntry(
         domain=DOMAIN,

tests/components/airly/__init__.py

@@ -1,8 +1,10 @@
 """Tests for Airly."""
 
 from homeassistant.components.airly.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry, load_fixture
+from tests.test_util.aiohttp import AiohttpClientMocker
 
 API_NEAREST_URL = "https://airapi.airly.eu/v2/measurements/nearest?lat=123.000000&lng=456.000000&maxDistanceKM=5.000000"
 API_POINT_URL = (
@@ -14,7 +16,9 @@
 }
 
 
-async def init_integration(hass, aioclient_mock) -> MockConfigEntry:
+async def init_integration(
+    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
+) -> MockConfigEntry:
     """Set up the Airly integration in Home Assistant."""
     entry = MockConfigEntry(
         domain=DOMAIN,

tests/components/bluetooth/test_passive_update_coordinator.py

@@ -52,7 +52,13 @@
 class MyCoordinator(PassiveBluetoothDataUpdateCoordinator):
     """An example coordinator that subclasses PassiveBluetoothDataUpdateCoordinator."""
 
-    def __init__(self, hass, logger, device_id, mode) -> None:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        logger: logging.Logger,
+        device_id: str,
+        mode: BluetoothScanningMode,
+    ) -> None:
         """Initialize the coordinator."""
         super().__init__(hass, logger, device_id, mode)
         self.data: dict[str, Any] = {}

tests/components/bthome/test_device_trigger.py

@@ -25,7 +25,7 @@ def get_device_id(mac: str) -> tuple[str, str]:
     return (BLUETOOTH_DOMAIN, mac)
 
 
-async def _async_setup_bthome_device(hass, mac: str):
+async def _async_setup_bthome_device(hass: HomeAssistant, mac: str) -> MockConfigEntry:
     config_entry = MockConfigEntry(
         domain=DOMAIN,
         unique_id=mac,

tests/components/device_tracker/common.py

@@ -20,7 +20,7 @@
     SourceType,
 )
 from homeassistant.core import HomeAssistant, callback
-from homeassistant.helpers.typing import GPSType
+from homeassistant.helpers.typing import ConfigType, GPSType
 from homeassistant.loader import bind_hass
 
 from tests.common import MockPlatform, mock_platform
@@ -143,7 +143,9 @@ def mock_legacy_device_tracker_setup(
 ) -> None:
     """Mock legacy device tracker platform setup."""
 
-    async def _async_get_scanner(hass, config) -> MockScanner:
+    async def _async_get_scanner(
+        hass: HomeAssistant, config: ConfigType
+    ) -> MockScanner:
         """Return the test scanner."""
         return legacy_device_scanner
 

tests/components/dexcom/__init__.py

@@ -7,6 +7,7 @@
 
 from homeassistant.components.dexcom.const import CONF_SERVER, DOMAIN, SERVER_US
 from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry, load_fixture
 
@@ -19,7 +20,7 @@
 GLUCOSE_READING = GlucoseReading(json.loads(load_fixture("data.json", "dexcom")))
 
 
-async def init_integration(hass) -> MockConfigEntry:
+async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the Dexcom integration in Home Assistant."""
     entry = MockConfigEntry(
         domain=DOMAIN,

tests/components/dlna_dms/conftest.py

@@ -38,7 +38,7 @@
 
 
 @pytest.fixture
-async def setup_media_source(hass) -> None:
+async def setup_media_source(hass: HomeAssistant) -> None:
     """Set up media source."""
     assert await async_setup_component(hass, "media_source", {})
 

tests/components/esphome/conftest.py

@@ -52,7 +52,7 @@ def esphome_mock_async_zeroconf(mock_async_zeroconf: MagicMock) -> None:
 
 
 @pytest.fixture(autouse=True)
-async def load_homeassistant(hass) -> None:
+async def load_homeassistant(hass: HomeAssistant) -> None:
     """Load the homeassistant integration."""
     assert await async_setup_component(hass, "homeassistant", {})
 
@@ -63,7 +63,7 @@ def mock_tts(mock_tts_cache_dir: Path) -> None:
 
 
 @pytest.fixture
-def mock_config_entry(hass) -> MockConfigEntry:
+def mock_config_entry(hass: HomeAssistant) -> MockConfigEntry:
     """Return the default mocked config entry."""
     config_entry = MockConfigEntry(
         title="ESPHome Device",

tests/components/fan/common.py

@@ -25,12 +25,13 @@
     SERVICE_TURN_OFF,
     SERVICE_TURN_ON,
 )
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockEntity
 
 
 async def async_turn_on(
-    hass,
+    hass: HomeAssistant,
     entity_id=ENTITY_MATCH_ALL,
     percentage: int | None = None,
     preset_mode: str | None = None,
@@ -50,7 +51,7 @@ async def async_turn_on(
     await hass.async_block_till_done()
 
 
-async def async_turn_off(hass, entity_id=ENTITY_MATCH_ALL) -> None:
+async def async_turn_off(hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL) -> None:
     """Turn all or specified fan off."""
     data = {ATTR_ENTITY_ID: entity_id} if entity_id else {}
 
@@ -59,7 +60,7 @@ async def async_turn_off(hass, entity_id=ENTITY_MATCH_ALL) -> None:
 
 
 async def async_oscillate(
-    hass, entity_id=ENTITY_MATCH_ALL, should_oscillate: bool = True
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, should_oscillate: bool = True
 ) -> None:
     """Set oscillation on all or specified fan."""
     data = {
@@ -76,7 +77,7 @@ async def async_oscillate(
 
 
 async def async_set_preset_mode(
-    hass, entity_id=ENTITY_MATCH_ALL, preset_mode: str | None = None
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, preset_mode: str | None = None
 ) -> None:
     """Set preset mode for all or specified fan."""
     data = {
@@ -90,7 +91,7 @@ async def async_set_preset_mode(
 
 
 async def async_set_percentage(
-    hass, entity_id=ENTITY_MATCH_ALL, percentage: int | None = None
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, percentage: int | None = None
 ) -> None:
     """Set percentage for all or specified fan."""
     data = {
@@ -104,7 +105,7 @@ async def async_set_percentage(
 
 
 async def async_increase_speed(
-    hass, entity_id=ENTITY_MATCH_ALL, percentage_step: int | None = None
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, percentage_step: int | None = None
 ) -> None:
     """Increase speed for all or specified fan."""
     data = {
@@ -121,7 +122,7 @@ async def async_increase_speed(
 
 
 async def async_decrease_speed(
-    hass, entity_id=ENTITY_MATCH_ALL, percentage_step: int | None = None
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, percentage_step: int | None = None
 ) -> None:
     """Decrease speed for all or specified fan."""
     data = {
@@ -138,7 +139,7 @@ async def async_decrease_speed(
 
 
 async def async_set_direction(
-    hass, entity_id=ENTITY_MATCH_ALL, direction: str | None = None
+    hass: HomeAssistant, entity_id=ENTITY_MATCH_ALL, direction: str | None = None
 ) -> None:
     """Set direction for all or specified fan."""
     data = {

tests/components/freedompro/conftest.py

@@ -10,6 +10,7 @@
 from typing_extensions import Generator
 
 from homeassistant.components.freedompro.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from .const import DEVICES, DEVICES_STATE
 
@@ -45,7 +46,7 @@ def mock_freedompro():
 
 
 @pytest.fixture
-async def init_integration(hass) -> MockConfigEntry:
+async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the Freedompro integration in Home Assistant."""
     entry = MockConfigEntry(
         domain=DOMAIN,
@@ -64,7 +65,7 @@ async def init_integration(hass) -> MockConfigEntry:
 
 
 @pytest.fixture
-async def init_integration_no_state(hass) -> MockConfigEntry:
+async def init_integration_no_state(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the Freedompro integration in Home Assistant without state."""
     entry = MockConfigEntry(
         domain=DOMAIN,

tests/components/gios/__init__.py

@@ -4,6 +4,7 @@
 from unittest.mock import patch
 
 from homeassistant.components.gios.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry, load_fixture
 
@@ -14,7 +15,7 @@
 
 
 async def init_integration(
-    hass, incomplete_data=False, invalid_indexes=False
+    hass: HomeAssistant, incomplete_data=False, invalid_indexes=False
 ) -> MockConfigEntry:
     """Set up the GIOS integration in Home Assistant."""
     entry = MockConfigEntry(

tests/components/imgw_pib/__init__.py

@@ -1,9 +1,13 @@
 """Tests for the IMGW-PIB integration."""
 
+from homeassistant.core import HomeAssistant
+
 from tests.common import MockConfigEntry
 
 
-async def init_integration(hass, config_entry: MockConfigEntry) -> MockConfigEntry:
+async def init_integration(
+    hass: HomeAssistant, config_entry: MockConfigEntry
+) -> MockConfigEntry:
     """Set up the IMGW-PIB integration in Home Assistant."""
     config_entry.add_to_hass(hass)
 

tests/components/kodi/__init__.py

@@ -11,13 +11,14 @@
     CONF_SSL,
     CONF_USERNAME,
 )
+from homeassistant.core import HomeAssistant
 
 from .util import MockConnection
 
 from tests.common import MockConfigEntry
 
 
-async def init_integration(hass) -> MockConfigEntry:
+async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the Kodi integration in Home Assistant."""
     entry_data = {
         CONF_NAME: "name",

tests/components/litejet/__init__.py

@@ -3,13 +3,14 @@
 from homeassistant.components import scene, switch
 from homeassistant.components.litejet import DOMAIN
 from homeassistant.const import CONF_PORT
+from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 
 from tests.common import MockConfigEntry
 
 
 async def async_init_integration(
-    hass, use_switch=False, use_scene=False
+    hass: HomeAssistant, use_switch: bool = False, use_scene: bool = False
 ) -> MockConfigEntry:
     """Set up the LiteJet integration in Home Assistant."""
 

tests/components/loqed/test_init.py

@@ -144,7 +144,7 @@ async def test_setup_cloudhook_from_entry_in_bridge(
 
 
 async def test_unload_entry(
-    hass, integration: MockConfigEntry, lock: loqed.Lock
+    hass: HomeAssistant, integration: MockConfigEntry, lock: loqed.Lock
 ) -> None:
     """Test successful unload of entry."""
 
@@ -157,7 +157,7 @@ async def test_unload_entry(
 
 
 async def test_unload_entry_fails(
-    hass, integration: MockConfigEntry, lock: loqed.Lock
+    hass: HomeAssistant, integration: MockConfigEntry, lock: loqed.Lock
 ) -> None:
     """Test unsuccessful unload of entry."""
     lock.deleteWebhook = AsyncMock(side_effect=Exception)

tests/components/lutron_caseta/__init__.py

@@ -9,6 +9,7 @@
     CONF_KEYFILE,
 )
 from homeassistant.const import CONF_HOST
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
@@ -83,7 +84,7 @@
 }
 
 
-async def async_setup_integration(hass, mock_bridge) -> MockConfigEntry:
+async def async_setup_integration(hass: HomeAssistant, mock_bridge) -> MockConfigEntry:
     """Set up a mock bridge."""
     mock_entry = MockConfigEntry(domain=DOMAIN, data=ENTRY_MOCK_DATA)
     mock_entry.add_to_hass(hass)

tests/components/met/__init__.py

@@ -4,11 +4,14 @@
 
 from homeassistant.components.met.const import CONF_TRACK_HOME, DOMAIN
 from homeassistant.const import CONF_ELEVATION, CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
 
-async def init_integration(hass, track_home=False) -> MockConfigEntry:
+async def init_integration(
+    hass: HomeAssistant, track_home: bool = False
+) -> MockConfigEntry:
     """Set up the Met integration in Home Assistant."""
     entry_data = {
         CONF_NAME: "test",

tests/components/met_eireann/__init__.py

@@ -4,11 +4,12 @@
 
 from homeassistant.components.met_eireann.const import DOMAIN
 from homeassistant.const import CONF_ELEVATION, CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
 
-async def init_integration(hass) -> MockConfigEntry:
+async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
     """Set up the Met Éireann integration in Home Assistant."""
     entry_data = {
         CONF_NAME: "test",

tests/components/motioneye/test_media_source.py

@@ -74,7 +74,7 @@
 
 
 @pytest.fixture(autouse=True)
-async def setup_media_source(hass) -> None:
+async def setup_media_source(hass: HomeAssistant) -> None:
     """Set up media source."""
     assert await async_setup_component(hass, "media_source", {})
 

tests/components/nam/__init__.py

@@ -3,6 +3,7 @@
 from unittest.mock import AsyncMock, Mock, patch
 
 from homeassistant.components.nam.const import DOMAIN
+from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry, load_json_object_fixture
 
@@ -12,7 +13,9 @@
 }
 
 
-async def init_integration(hass, co2_sensor=True) -> MockConfigEntry:
+async def init_integration(
+    hass: HomeAssistant, co2_sensor: bool = True
+) -> MockConfigEntry:
     """Set up the Nettigo Air Monitor integration in Home Assistant."""
     entry = MockConfigEntry(
         domain=DOMAIN,