Add type hints to integration tests (part 10) (#87985)

tests/components/homewizard/test_sensor.py

@@ -1,5 +1,4 @@
 """Test the update coordinator for HomeWizard."""
-
 from datetime import timedelta
 from unittest.mock import AsyncMock, patch
 
@@ -24,6 +23,7 @@ from homeassistant.const import (
     UnitOfPower,
     UnitOfVolume,
 )
+from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 import homeassistant.util.dt as dt_util
 
@@ -33,8 +33,8 @@ from tests.common import async_fire_time_changed
 
 
 async def test_sensor_entity_smr_version(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads smr version."""
 
     api = get_mock_device()
@@ -71,8 +71,8 @@ async def test_sensor_entity_smr_version(
 
 
 async def test_sensor_entity_meter_model(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads meter model."""
 
     api = get_mock_device()
@@ -111,8 +111,8 @@ async def test_sensor_entity_meter_model(
 
 
 async def test_sensor_entity_unique_meter_id(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads unique meter id."""
 
     api = get_mock_device()
@@ -150,7 +150,9 @@ async def test_sensor_entity_unique_meter_id(
     assert state.attributes.get(ATTR_ICON) == "mdi:alphabetical-variant"
 
 
-async def test_sensor_entity_wifi_ssid(hass, mock_config_entry_data, mock_config_entry):
+async def test_sensor_entity_wifi_ssid(
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads wifi ssid."""
 
     api = get_mock_device()
@@ -187,8 +189,8 @@ async def test_sensor_entity_wifi_ssid(hass, mock_config_entry_data, mock_config
 
 
 async def test_sensor_entity_active_tariff(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active_tariff."""
 
     api = get_mock_device()
@@ -226,8 +228,8 @@ async def test_sensor_entity_active_tariff(
 
 
 async def test_sensor_entity_wifi_strength(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads wifi strength."""
 
     api = get_mock_device()
@@ -253,8 +255,8 @@ async def test_sensor_entity_wifi_strength(
 
 
 async def test_sensor_entity_total_power_import_t1_kwh(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total power import t1."""
 
     api = get_mock_device()
@@ -295,8 +297,8 @@ async def test_sensor_entity_total_power_import_t1_kwh(
 
 
 async def test_sensor_entity_total_power_import_t2_kwh(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total power import t2."""
 
     api = get_mock_device()
@@ -337,8 +339,8 @@ async def test_sensor_entity_total_power_import_t2_kwh(
 
 
 async def test_sensor_entity_total_power_export_t1_kwh(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total power export t1."""
 
     api = get_mock_device()
@@ -379,8 +381,8 @@ async def test_sensor_entity_total_power_export_t1_kwh(
 
 
 async def test_sensor_entity_total_power_export_t2_kwh(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total power export t2."""
 
     api = get_mock_device()
@@ -421,8 +423,8 @@ async def test_sensor_entity_total_power_export_t2_kwh(
 
 
 async def test_sensor_entity_active_power(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active power."""
 
     api = get_mock_device()
@@ -459,8 +461,8 @@ async def test_sensor_entity_active_power(
 
 
 async def test_sensor_entity_active_power_l1(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active power l1."""
 
     api = get_mock_device()
@@ -499,8 +501,8 @@ async def test_sensor_entity_active_power_l1(
 
 
 async def test_sensor_entity_active_power_l2(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active power l2."""
 
     api = get_mock_device()
@@ -539,8 +541,8 @@ async def test_sensor_entity_active_power_l2(
 
 
 async def test_sensor_entity_active_power_l3(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active power l3."""
 
     api = get_mock_device()
@@ -578,7 +580,9 @@ async def test_sensor_entity_active_power_l3(
     assert ATTR_ICON not in state.attributes
 
 
-async def test_sensor_entity_total_gas(hass, mock_config_entry_data, mock_config_entry):
+async def test_sensor_entity_total_gas(
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total gas."""
 
     api = get_mock_device()
@@ -615,8 +619,8 @@ async def test_sensor_entity_total_gas(hass, mock_config_entry_data, mock_config
 
 
 async def test_sensor_entity_unique_gas_meter_id(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads unique gas meter id."""
 
     api = get_mock_device()
@@ -655,8 +659,8 @@ async def test_sensor_entity_unique_gas_meter_id(
 
 
 async def test_sensor_entity_active_voltage_l1(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active voltage l1."""
 
     api = get_mock_device()
@@ -714,8 +718,8 @@ async def test_sensor_entity_active_voltage_l1(
 
 
 async def test_sensor_entity_active_voltage_l2(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active voltage l2."""
 
     api = get_mock_device()
@@ -773,8 +777,8 @@ async def test_sensor_entity_active_voltage_l2(
 
 
 async def test_sensor_entity_active_voltage_l3(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active voltage l3."""
 
     api = get_mock_device()
@@ -832,8 +836,8 @@ async def test_sensor_entity_active_voltage_l3(
 
 
 async def test_sensor_entity_active_current_l1(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active current l1."""
 
     api = get_mock_device()
@@ -891,8 +895,8 @@ async def test_sensor_entity_active_current_l1(
 
 
 async def test_sensor_entity_active_current_l2(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active current l2."""
 
     api = get_mock_device()
@@ -950,8 +954,8 @@ async def test_sensor_entity_active_current_l2(
 
 
 async def test_sensor_entity_active_current_l3(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active current l3."""
 
     api = get_mock_device()
@@ -1009,8 +1013,8 @@ async def test_sensor_entity_active_current_l3(
 
 
 async def test_sensor_entity_active_frequency(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active frequency."""
 
     api = get_mock_device()
@@ -1065,8 +1069,8 @@ async def test_sensor_entity_active_frequency(
 
 
 async def test_sensor_entity_voltage_sag_count_l1(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_sag_count_l1."""
 
     api = get_mock_device()
@@ -1104,8 +1108,8 @@ async def test_sensor_entity_voltage_sag_count_l1(
 
 
 async def test_sensor_entity_voltage_sag_count_l2(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_sag_count_l2."""
 
     api = get_mock_device()
@@ -1143,8 +1147,8 @@ async def test_sensor_entity_voltage_sag_count_l2(
 
 
 async def test_sensor_entity_voltage_sag_count_l3(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_sag_count_l3."""
 
     api = get_mock_device()
@@ -1182,8 +1186,8 @@ async def test_sensor_entity_voltage_sag_count_l3(
 
 
 async def test_sensor_entity_voltage_swell_count_l1(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_swell_count_l1."""
 
     api = get_mock_device()
@@ -1223,8 +1227,8 @@ async def test_sensor_entity_voltage_swell_count_l1(
 
 
 async def test_sensor_entity_voltage_swell_count_l2(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_swell_count_l2."""
 
     api = get_mock_device()
@@ -1264,8 +1268,8 @@ async def test_sensor_entity_voltage_swell_count_l2(
 
 
 async def test_sensor_entity_voltage_swell_count_l3(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads voltage_swell_count_l3."""
 
     api = get_mock_device()
@@ -1305,8 +1309,8 @@ async def test_sensor_entity_voltage_swell_count_l3(
 
 
 async def test_sensor_entity_any_power_fail_count(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads any power fail count."""
 
     api = get_mock_device()
@@ -1344,8 +1348,8 @@ async def test_sensor_entity_any_power_fail_count(
 
 
 async def test_sensor_entity_long_power_fail_count(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads long power fail count."""
 
     api = get_mock_device()
@@ -1385,8 +1389,8 @@ async def test_sensor_entity_long_power_fail_count(
 
 
 async def test_sensor_entity_active_power_average(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active power average."""
 
     api = get_mock_device()
@@ -1428,8 +1432,8 @@ async def test_sensor_entity_active_power_average(
 
 
 async def test_sensor_entity_monthly_power_peak(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads monthly power peak."""
 
     api = get_mock_device()
@@ -1471,8 +1475,8 @@ async def test_sensor_entity_monthly_power_peak(
 
 
 async def test_sensor_entity_active_liters(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads active liters (watermeter)."""
 
     api = get_mock_device()
@@ -1512,8 +1516,8 @@ async def test_sensor_entity_active_liters(
 
 
 async def test_sensor_entity_total_liters(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test entity loads total liters (watermeter)."""
 
     api = get_mock_device()
@@ -1553,8 +1557,8 @@ async def test_sensor_entity_total_liters(
 
 
 async def test_sensor_entity_disabled_when_null(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test sensor disables data with null by default."""
 
     api = get_mock_device()
@@ -1592,8 +1596,8 @@ async def test_sensor_entity_disabled_when_null(
 
 
 async def test_sensor_entity_export_disabled_when_unused(
-    hass, mock_config_entry_data, mock_config_entry
-):
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test sensor disables export if value is 0."""
 
     api = get_mock_device()
@@ -1641,7 +1645,9 @@ async def test_sensor_entity_export_disabled_when_unused(
     assert entry.disabled
 
 
-async def test_sensors_unreachable(hass, mock_config_entry_data, mock_config_entry):
+async def test_sensors_unreachable(
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test sensor handles api unreachable."""
 
     api = get_mock_device()
@@ -1689,7 +1695,9 @@ async def test_sensors_unreachable(hass, mock_config_entry_data, mock_config_ent
         )
 
 
-async def test_api_disabled(hass, mock_config_entry_data, mock_config_entry):
+async def test_api_disabled(
+    hass: HomeAssistant, mock_config_entry_data, mock_config_entry
+) -> None:
     """Test sensor handles api unreachable."""
 
     api = get_mock_device()