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

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

Browse Source
This commit is contained in:
epenet 2023-02-13 10:25:26 +01:00 committed by GitHub
parent 3f910b49ae
commit 9404a107b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
51 changed files with 811 additions and 411 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/homematicip_cloud/test_alarm_control_panel.py
View File

@ -48,7 +48,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_alarm_control_panel(hass, default_mock_hap_factory): async def test_hmip_alarm_control_panel(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipAlarmControlPanel.""" """Test HomematicipAlarmControlPanel."""
entity_id = "alarm_control_panel.hmip_alarm_control_panel" entity_id = "alarm_control_panel.hmip_alarm_control_panel"
entity_name = "HmIP Alarm Control Panel" entity_name = "HmIP Alarm Control Panel"

68
tests/components/homematicip_cloud/test_binary_sensor.py
View File

@ -39,7 +39,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_home_cloud_connection_sensor(hass, default_mock_hap_factory): async def test_hmip_home_cloud_connection_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCloudConnectionSensor.""" """Test HomematicipCloudConnectionSensor."""
entity_id = "binary_sensor.cloud_connection" entity_id = "binary_sensor.cloud_connection"
entity_name = "Cloud Connection" entity_name = "Cloud Connection"
@ -60,7 +62,9 @@ async def test_hmip_home_cloud_connection_sensor(hass, default_mock_hap_factory)
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_acceleration_sensor(hass, default_mock_hap_factory): async def test_hmip_acceleration_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipAccelerationSensor.""" """Test HomematicipAccelerationSensor."""
entity_id = "binary_sensor.garagentor" entity_id = "binary_sensor.garagentor"
entity_name = "Garagentor" entity_name = "Garagentor"
@ -97,7 +101,9 @@ async def test_hmip_acceleration_sensor(hass, default_mock_hap_factory):
assert len(hmip_device.mock_calls) == service_call_counter + 2 assert len(hmip_device.mock_calls) == service_call_counter + 2
async def test_hmip_tilt_vibration_sensor(hass, default_mock_hap_factory): async def test_hmip_tilt_vibration_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTiltVibrationSensor.""" """Test HomematicipTiltVibrationSensor."""
entity_id = "binary_sensor.garage_neigungs_und_erschutterungssensor" entity_id = "binary_sensor.garage_neigungs_und_erschutterungssensor"
entity_name = "Garage Neigungs- und Erschütterungssensor" entity_name = "Garage Neigungs- und Erschütterungssensor"
@ -133,7 +139,9 @@ async def test_hmip_tilt_vibration_sensor(hass, default_mock_hap_factory):
assert len(hmip_device.mock_calls) == service_call_counter + 2 assert len(hmip_device.mock_calls) == service_call_counter + 2
async def test_hmip_contact_interface(hass, default_mock_hap_factory): async def test_hmip_contact_interface(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipContactInterface.""" """Test HomematicipContactInterface."""
entity_id = "binary_sensor.kontakt_schnittstelle_unterputz_1_fach" entity_id = "binary_sensor.kontakt_schnittstelle_unterputz_1_fach"
entity_name = "Kontakt-Schnittstelle Unterputz – 1-fach" entity_name = "Kontakt-Schnittstelle Unterputz – 1-fach"
@ -156,7 +164,9 @@ async def test_hmip_contact_interface(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_shutter_contact(hass, default_mock_hap_factory): async def test_hmip_shutter_contact(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipShutterContact.""" """Test HomematicipShutterContact."""
entity_id = "binary_sensor.fenstergriffsensor" entity_id = "binary_sensor.fenstergriffsensor"
entity_name = "Fenstergriffsensor" entity_name = "Fenstergriffsensor"
@ -196,7 +206,9 @@ async def test_hmip_shutter_contact(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_SABOTAGE] assert ha_state.attributes[ATTR_SABOTAGE]
async def test_hmip_shutter_contact_optical(hass, default_mock_hap_factory): async def test_hmip_shutter_contact_optical(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipShutterContact.""" """Test HomematicipShutterContact."""
entity_id = "binary_sensor.sitzplatzture" entity_id = "binary_sensor.sitzplatzture"
entity_name = "Sitzplatzt\u00fcre" entity_name = "Sitzplatzt\u00fcre"
@ -226,7 +238,9 @@ async def test_hmip_shutter_contact_optical(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_SABOTAGE] assert ha_state.attributes[ATTR_SABOTAGE]
async def test_hmip_motion_detector(hass, default_mock_hap_factory): async def test_hmip_motion_detector(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipMotionDetector.""" """Test HomematicipMotionDetector."""
entity_id = "binary_sensor.bewegungsmelder_fur_55er_rahmen_innen" entity_id = "binary_sensor.bewegungsmelder_fur_55er_rahmen_innen"
entity_name = "Bewegungsmelder für 55er Rahmen – innen" entity_name = "Bewegungsmelder für 55er Rahmen – innen"
@ -245,7 +259,9 @@ async def test_hmip_motion_detector(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_presence_detector(hass, default_mock_hap_factory): async def test_hmip_presence_detector(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipPresenceDetector.""" """Test HomematicipPresenceDetector."""
entity_id = "binary_sensor.spi_1" entity_id = "binary_sensor.spi_1"
entity_name = "SPI_1" entity_name = "SPI_1"
@ -270,8 +286,8 @@ async def test_hmip_presence_detector(hass, default_mock_hap_factory):
async def test_hmip_pluggable_mains_failure_surveillance_sensor( async def test_hmip_pluggable_mains_failure_surveillance_sensor(
hass, default_mock_hap_factory hass: HomeAssistant, default_mock_hap_factory
): ) -> None:
"""Test HomematicipPresenceDetector.""" """Test HomematicipPresenceDetector."""
entity_id = "binary_sensor.netzausfalluberwachung" entity_id = "binary_sensor.netzausfalluberwachung"
entity_name = "Netzausfallüberwachung" entity_name = "Netzausfallüberwachung"
@ -290,7 +306,9 @@ async def test_hmip_pluggable_mains_failure_surveillance_sensor(
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_smoke_detector(hass, default_mock_hap_factory): async def test_hmip_smoke_detector(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSmokeDetector.""" """Test HomematicipSmokeDetector."""
entity_id = "binary_sensor.rauchwarnmelder" entity_id = "binary_sensor.rauchwarnmelder"
entity_name = "Rauchwarnmelder" entity_name = "Rauchwarnmelder"
@ -322,7 +340,9 @@ async def test_hmip_smoke_detector(hass, default_mock_hap_factory):
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_water_detector(hass, default_mock_hap_factory): async def test_hmip_water_detector(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipWaterDetector.""" """Test HomematicipWaterDetector."""
entity_id = "binary_sensor.wassersensor" entity_id = "binary_sensor.wassersensor"
entity_name = "Wassersensor" entity_name = "Wassersensor"
@ -357,7 +377,7 @@ async def test_hmip_water_detector(hass, default_mock_hap_factory):
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_storm_sensor(hass, default_mock_hap_factory): async def test_hmip_storm_sensor(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipStormSensor.""" """Test HomematicipStormSensor."""
entity_id = "binary_sensor.weather_sensor_plus_storm" entity_id = "binary_sensor.weather_sensor_plus_storm"
entity_name = "Weather Sensor – plus Storm" entity_name = "Weather Sensor – plus Storm"
@ -376,7 +396,7 @@ async def test_hmip_storm_sensor(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_rain_sensor(hass, default_mock_hap_factory): async def test_hmip_rain_sensor(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipRainSensor.""" """Test HomematicipRainSensor."""
entity_id = "binary_sensor.wettersensor_pro_raining" entity_id = "binary_sensor.wettersensor_pro_raining"
entity_name = "Wettersensor - pro Raining" entity_name = "Wettersensor - pro Raining"
@ -395,7 +415,9 @@ async def test_hmip_rain_sensor(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_sunshine_sensor(hass, default_mock_hap_factory): async def test_hmip_sunshine_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSunshineSensor.""" """Test HomematicipSunshineSensor."""
entity_id = "binary_sensor.wettersensor_pro_sunshine" entity_id = "binary_sensor.wettersensor_pro_sunshine"
entity_name = "Wettersensor - pro Sunshine" entity_name = "Wettersensor - pro Sunshine"
@ -415,7 +437,9 @@ async def test_hmip_sunshine_sensor(hass, default_mock_hap_factory):
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_battery_sensor(hass, default_mock_hap_factory): async def test_hmip_battery_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSunshineSensor.""" """Test HomematicipSunshineSensor."""
entity_id = "binary_sensor.wohnungsture_battery" entity_id = "binary_sensor.wohnungsture_battery"
entity_name = "Wohnungstüre Battery" entity_name = "Wohnungstüre Battery"
@ -434,7 +458,9 @@ async def test_hmip_battery_sensor(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_security_zone_sensor_group(hass, default_mock_hap_factory): async def test_hmip_security_zone_sensor_group(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSecurityZoneSensorGroup.""" """Test HomematicipSecurityZoneSensorGroup."""
entity_id = "binary_sensor.internal_securityzone" entity_id = "binary_sensor.internal_securityzone"
entity_name = "INTERNAL SecurityZone" entity_name = "INTERNAL SecurityZone"
@ -469,7 +495,9 @@ async def test_hmip_security_zone_sensor_group(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_WINDOW_STATE] == WindowState.OPEN assert ha_state.attributes[ATTR_WINDOW_STATE] == WindowState.OPEN
async def test_hmip_security_sensor_group(hass, default_mock_hap_factory): async def test_hmip_security_sensor_group(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSecuritySensorGroup.""" """Test HomematicipSecuritySensorGroup."""
entity_id = "binary_sensor.buro_sensors" entity_id = "binary_sensor.buro_sensors"
entity_name = "Büro Sensors" entity_name = "Büro Sensors"
@ -541,7 +569,9 @@ async def test_hmip_security_sensor_group(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_multi_contact_interface(hass, default_mock_hap_factory): async def test_hmip_multi_contact_interface(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipMultiContactInterface.""" """Test HomematicipMultiContactInterface."""
entity_id = "binary_sensor.wired_eingangsmodul_32_fach_channel5" entity_id = "binary_sensor.wired_eingangsmodul_32_fach_channel5"
entity_name = "Wired Eingangsmodul – 32-fach Channel5" entity_name = "Wired Eingangsmodul – 32-fach Channel5"

6
tests/components/homematicip_cloud/test_button.py
View File

@ -1,16 +1,18 @@
"""Tests for HomematicIP Cloud button.""" """Tests for HomematicIP Cloud button."""
from unittest.mock import patch from unittest.mock import patch
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
from homeassistant.components.button.const import SERVICE_PRESS from homeassistant.components.button.const import SERVICE_PRESS
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
from .helper import get_and_check_entity_basics from .helper import get_and_check_entity_basics
async def test_hmip_garage_door_controller_button(hass, default_mock_hap_factory): async def test_hmip_garage_door_controller_button(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipGarageDoorControllerButton.""" """Test HomematicipGarageDoorControllerButton."""
entity_id = "button.garagentor" entity_id = "button.garagentor"
entity_name = "Garagentor" entity_name = "Garagentor"

24
tests/components/homematicip_cloud/test_climate.py
View File

@ -36,7 +36,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_heating_group_heat(hass, default_mock_hap_factory): async def test_hmip_heating_group_heat(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingGroup.""" """Test HomematicipHeatingGroup."""
entity_id = "climate.badezimmer" entity_id = "climate.badezimmer"
entity_name = "Badezimmer" entity_name = "Badezimmer"
@ -229,7 +231,9 @@ async def test_hmip_heating_group_heat(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_HVAC_ACTION] == HVACAction.IDLE assert ha_state.attributes[ATTR_HVAC_ACTION] == HVACAction.IDLE
async def test_hmip_heating_group_cool(hass, default_mock_hap_factory): async def test_hmip_heating_group_cool(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingGroup.""" """Test HomematicipHeatingGroup."""
entity_id = "climate.badezimmer" entity_id = "climate.badezimmer"
entity_name = "Badezimmer" entity_name = "Badezimmer"
@ -349,7 +353,9 @@ async def test_hmip_heating_group_cool(hass, default_mock_hap_factory):
assert hmip_device.mock_calls[-1][1] == (4,) assert hmip_device.mock_calls[-1][1] == (4,)
async def test_hmip_heating_group_heat_with_switch(hass, default_mock_hap_factory): async def test_hmip_heating_group_heat_with_switch(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingGroup.""" """Test HomematicipHeatingGroup."""
entity_id = "climate.schlafzimmer" entity_id = "climate.schlafzimmer"
entity_name = "Schlafzimmer" entity_name = "Schlafzimmer"
@ -373,7 +379,9 @@ async def test_hmip_heating_group_heat_with_switch(hass, default_mock_hap_factor
assert ha_state.attributes[ATTR_PRESET_MODES] == [PRESET_BOOST, "STD", "P2"] assert ha_state.attributes[ATTR_PRESET_MODES] == [PRESET_BOOST, "STD", "P2"]
async def test_hmip_heating_group_heat_with_radiator(hass, default_mock_hap_factory): async def test_hmip_heating_group_heat_with_radiator(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingGroup.""" """Test HomematicipHeatingGroup."""
entity_id = "climate.vorzimmer" entity_id = "climate.vorzimmer"
entity_name = "Vorzimmer" entity_name = "Vorzimmer"
@ -396,7 +404,9 @@ async def test_hmip_heating_group_heat_with_radiator(hass, default_mock_hap_fact
assert ha_state.attributes[ATTR_PRESET_MODES] == [PRESET_NONE, PRESET_BOOST] assert ha_state.attributes[ATTR_PRESET_MODES] == [PRESET_NONE, PRESET_BOOST]
async def test_hmip_climate_services(hass, mock_hap_with_service): async def test_hmip_climate_services(
hass: HomeAssistant, mock_hap_with_service
) -> None:
"""Test HomematicipHeatingGroup.""" """Test HomematicipHeatingGroup."""
home = mock_hap_with_service.home home = mock_hap_with_service.home
@ -508,7 +518,9 @@ async def test_hmip_climate_services(hass, mock_hap_with_service):
assert len(home._connection.mock_calls) == 10 assert len(home._connection.mock_calls) == 10
async def test_hmip_heating_group_services(hass, default_mock_hap_factory): async def test_hmip_heating_group_services(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingGroup services.""" """Test HomematicipHeatingGroup services."""
entity_id = "climate.badezimmer" entity_id = "climate.badezimmer"
entity_name = "Badezimmer" entity_name = "Badezimmer"

4
tests/components/homematicip_cloud/test_config_flow.py
View File

@ -18,7 +18,7 @@ DEFAULT_CONFIG = {HMIPC_HAPID: "ABC123", HMIPC_PIN: "123", HMIPC_NAME: "hmip"}
IMPORT_CONFIG = {HMIPC_HAPID: "ABC123", HMIPC_AUTHTOKEN: "123", HMIPC_NAME: "hmip"} IMPORT_CONFIG = {HMIPC_HAPID: "ABC123", HMIPC_AUTHTOKEN: "123", HMIPC_NAME: "hmip"}
async def test_flow_works(hass, simple_mock_home): async def test_flow_works(hass: HomeAssistant, simple_mock_home) -> None:
"""Test config flow.""" """Test config flow."""
with patch( with patch(
@ -152,7 +152,7 @@ async def test_init_already_configured(hass: HomeAssistant) -> None:
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
async def test_import_config(hass, simple_mock_home): async def test_import_config(hass: HomeAssistant, simple_mock_home) -> None:
"""Test importing a host with an existing config file.""" """Test importing a host with an existing config file."""
with patch( with patch(
"homeassistant.components.homematicip_cloud.hap.HomematicipAuth.async_checkbutton", "homeassistant.components.homematicip_cloud.hap.HomematicipAuth.async_checkbutton",

28
tests/components/homematicip_cloud/test_cover.py
View File

@ -22,7 +22,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_cover_shutter(hass, default_mock_hap_factory): async def test_hmip_cover_shutter(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCoverShutte.""" """Test HomematicipCoverShutte."""
entity_id = "cover.broll_1" entity_id = "cover.broll_1"
entity_name = "BROLL_1" entity_name = "BROLL_1"
@ -87,7 +89,7 @@ async def test_hmip_cover_shutter(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_cover_slats(hass, default_mock_hap_factory): async def test_hmip_cover_slats(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipCoverSlats.""" """Test HomematicipCoverSlats."""
entity_id = "cover.sofa_links" entity_id = "cover.sofa_links"
entity_name = "Sofa links" entity_name = "Sofa links"
@ -161,7 +163,9 @@ async def test_hmip_cover_slats(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_multi_cover_slats(hass, default_mock_hap_factory): async def test_hmip_multi_cover_slats(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCoverSlats.""" """Test HomematicipCoverSlats."""
entity_id = "cover.wohnzimmer_fenster" entity_id = "cover.wohnzimmer_fenster"
entity_name = "Wohnzimmer Fenster" entity_name = "Wohnzimmer Fenster"
@ -239,7 +243,7 @@ async def test_hmip_multi_cover_slats(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_blind_module(hass, default_mock_hap_factory): async def test_hmip_blind_module(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipBlindModule.""" """Test HomematicipBlindModule."""
entity_id = "cover.sonnenschutz_balkontur" entity_id = "cover.sonnenschutz_balkontur"
entity_name = "Sonnenschutz Balkontür" entity_name = "Sonnenschutz Balkontür"
@ -349,7 +353,9 @@ async def test_hmip_blind_module(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_garage_door_tormatic(hass, default_mock_hap_factory): async def test_hmip_garage_door_tormatic(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCoverShutte.""" """Test HomematicipCoverShutte."""
entity_id = "cover.garage_door_module" entity_id = "cover.garage_door_module"
entity_name = "Garage Door Module" entity_name = "Garage Door Module"
@ -396,7 +402,9 @@ async def test_hmip_garage_door_tormatic(hass, default_mock_hap_factory):
assert hmip_device.mock_calls[-1][1] == (DoorCommand.STOP,) assert hmip_device.mock_calls[-1][1] == (DoorCommand.STOP,)
async def test_hmip_garage_door_hoermann(hass, default_mock_hap_factory): async def test_hmip_garage_door_hoermann(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCoverShutte.""" """Test HomematicipCoverShutte."""
entity_id = "cover.garage_door" entity_id = "cover.garage_door"
entity_name = "Garage door" entity_name = "Garage door"
@ -443,7 +451,9 @@ async def test_hmip_garage_door_hoermann(hass, default_mock_hap_factory):
assert hmip_device.mock_calls[-1][1] == (DoorCommand.STOP,) assert hmip_device.mock_calls[-1][1] == (DoorCommand.STOP,)
async def test_hmip_cover_shutter_group(hass, default_mock_hap_factory): async def test_hmip_cover_shutter_group(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipCoverShutteGroup.""" """Test HomematicipCoverShutteGroup."""
entity_id = "cover.rollos_shuttergroup" entity_id = "cover.rollos_shuttergroup"
entity_name = "Rollos ShutterGroup" entity_name = "Rollos ShutterGroup"
@ -506,7 +516,9 @@ async def test_hmip_cover_shutter_group(hass, default_mock_hap_factory):
assert ha_state.state == STATE_UNKNOWN assert ha_state.state == STATE_UNKNOWN
async def test_hmip_cover_slats_group(hass, default_mock_hap_factory): async def test_hmip_cover_slats_group(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test slats with HomematicipCoverShutteGroup.""" """Test slats with HomematicipCoverShutteGroup."""
entity_id = "cover.rollos_shuttergroup" entity_id = "cover.rollos_shuttergroup"
entity_name = "Rollos ShutterGroup" entity_name = "Rollos ShutterGroup"

33
tests/components/homematicip_cloud/test_device.py
View File

@ -6,6 +6,7 @@ from homematicip.base.enums import EventType
from homeassistant.components.homematicip_cloud import DOMAIN as HMIPC_DOMAIN from homeassistant.components.homematicip_cloud import DOMAIN as HMIPC_DOMAIN
from homeassistant.components.homematicip_cloud.hap import HomematicipHAP from homeassistant.components.homematicip_cloud.hap import HomematicipHAP
from homeassistant.const import STATE_ON, STATE_UNAVAILABLE from homeassistant.const import STATE_ON, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from .helper import ( from .helper import (
@ -16,7 +17,9 @@ from .helper import (
) )
async def test_hmip_load_all_supported_devices(hass, default_mock_hap_factory): async def test_hmip_load_all_supported_devices(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Ensure that all supported devices could be loaded.""" """Ensure that all supported devices could be loaded."""
mock_hap = await default_mock_hap_factory.async_get_mock_hap( mock_hap = await default_mock_hap_factory.async_get_mock_hap(
test_devices=None, test_groups=None test_devices=None, test_groups=None
@ -25,7 +28,9 @@ async def test_hmip_load_all_supported_devices(hass, default_mock_hap_factory):
assert len(mock_hap.hmip_device_by_entity_id) == 270 assert len(mock_hap.hmip_device_by_entity_id) == 270
async def test_hmip_remove_device(hass, default_mock_hap_factory): async def test_hmip_remove_device(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test Remove of hmip device.""" """Test Remove of hmip device."""
entity_id = "light.treppe_ch" entity_id = "light.treppe_ch"
entity_name = "Treppe CH" entity_name = "Treppe CH"
@ -57,7 +62,9 @@ async def test_hmip_remove_device(hass, default_mock_hap_factory):
assert len(mock_hap.hmip_device_by_entity_id) == pre_mapping_count - 3 assert len(mock_hap.hmip_device_by_entity_id) == pre_mapping_count - 3
async def test_hmip_add_device(hass, default_mock_hap_factory, hmip_config_entry): async def test_hmip_add_device(
hass: HomeAssistant, default_mock_hap_factory, hmip_config_entry
) -> None:
"""Test Remove of hmip device.""" """Test Remove of hmip device."""
entity_id = "light.treppe_ch" entity_id = "light.treppe_ch"
entity_name = "Treppe CH" entity_name = "Treppe CH"
@ -105,7 +112,7 @@ async def test_hmip_add_device(hass, default_mock_hap_factory, hmip_config_entry
assert len(new_hap.hmip_device_by_entity_id) == pre_mapping_count assert len(new_hap.hmip_device_by_entity_id) == pre_mapping_count
async def test_hmip_remove_group(hass, default_mock_hap_factory): async def test_hmip_remove_group(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test Remove of hmip group.""" """Test Remove of hmip group."""
entity_id = "switch.strom_group" entity_id = "switch.strom_group"
entity_name = "Strom Group" entity_name = "Strom Group"
@ -135,8 +142,8 @@ async def test_hmip_remove_group(hass, default_mock_hap_factory):
async def test_all_devices_unavailable_when_hap_not_connected( async def test_all_devices_unavailable_when_hap_not_connected(
hass, default_mock_hap_factory hass: HomeAssistant, default_mock_hap_factory
): ) -> None:
"""Test make all devices unavaulable when hap is not connected.""" """Test make all devices unavaulable when hap is not connected."""
entity_id = "light.treppe_ch" entity_id = "light.treppe_ch"
entity_name = "Treppe CH" entity_name = "Treppe CH"
@ -160,7 +167,7 @@ async def test_all_devices_unavailable_when_hap_not_connected(
assert ha_state.state == STATE_UNAVAILABLE assert ha_state.state == STATE_UNAVAILABLE
async def test_hap_reconnected(hass, default_mock_hap_factory): async def test_hap_reconnected(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test reconnect hap.""" """Test reconnect hap."""
entity_id = "light.treppe_ch" entity_id = "light.treppe_ch"
entity_name = "Treppe CH" entity_name = "Treppe CH"
@ -190,7 +197,9 @@ async def test_hap_reconnected(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hap_with_name(hass, mock_connection, hmip_config_entry): async def test_hap_with_name(
hass: HomeAssistant, mock_connection, hmip_config_entry
) -> None:
"""Test hap with name.""" """Test hap with name."""
home_name = "TestName" home_name = "TestName"
entity_id = f"light.{home_name.lower()}_treppe_ch" entity_id = f"light.{home_name.lower()}_treppe_ch"
@ -212,7 +221,9 @@ async def test_hap_with_name(hass, mock_connection, hmip_config_entry):
assert ha_state.attributes["friendly_name"] == entity_name assert ha_state.attributes["friendly_name"] == entity_name
async def test_hmip_reset_energy_counter_services(hass, default_mock_hap_factory): async def test_hmip_reset_energy_counter_services(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test reset_energy_counter service.""" """Test reset_energy_counter service."""
entity_id = "switch.pc" entity_id = "switch.pc"
entity_name = "Pc" entity_name = "Pc"
@ -242,7 +253,9 @@ async def test_hmip_reset_energy_counter_services(hass, default_mock_hap_factory
assert len(hmip_device._connection.mock_calls) == 4 # pylint: disable=W0212 assert len(hmip_device._connection.mock_calls) == 4 # pylint: disable=W0212
async def test_hmip_multi_area_device(hass, default_mock_hap_factory): async def test_hmip_multi_area_device(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test multi area device. Check if devices are created and referenced.""" """Test multi area device. Check if devices are created and referenced."""
entity_id = "binary_sensor.wired_eingangsmodul_32_fach_channel5" entity_id = "binary_sensor.wired_eingangsmodul_32_fach_channel5"
entity_name = "Wired Eingangsmodul – 32-fach Channel5" entity_name = "Wired Eingangsmodul – 32-fach Channel5"

16
tests/components/homematicip_cloud/test_hap.py
View File

@ -103,7 +103,9 @@ async def test_hap_setup_connection_error() -> None:
assert not hass.config_entries.flow.async_init.mock_calls assert not hass.config_entries.flow.async_init.mock_calls
async def test_hap_reset_unloads_entry_if_setup(hass, default_mock_hap_factory): async def test_hap_reset_unloads_entry_if_setup(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test calling reset while the entry has been setup.""" """Test calling reset while the entry has been setup."""
mock_hap = await default_mock_hap_factory.async_get_mock_hap() mock_hap = await default_mock_hap_factory.async_get_mock_hap()
assert hass.data[HMIPC_DOMAIN][HAPID] == mock_hap assert hass.data[HMIPC_DOMAIN][HAPID] == mock_hap
@ -116,7 +118,9 @@ async def test_hap_reset_unloads_entry_if_setup(hass, default_mock_hap_factory):
assert hass.data[HMIPC_DOMAIN] == {} assert hass.data[HMIPC_DOMAIN] == {}
async def test_hap_create(hass, hmip_config_entry, simple_mock_home): async def test_hap_create(
hass: HomeAssistant, hmip_config_entry, simple_mock_home
) -> None:
"""Mock AsyncHome to execute get_hap.""" """Mock AsyncHome to execute get_hap."""
hass.config.components.add(HMIPC_DOMAIN) hass.config.components.add(HMIPC_DOMAIN)
hap = HomematicipHAP(hass, hmip_config_entry) hap = HomematicipHAP(hass, hmip_config_entry)
@ -125,7 +129,9 @@ async def test_hap_create(hass, hmip_config_entry, simple_mock_home):
assert await hap.async_setup() assert await hap.async_setup()
async def test_hap_create_exception(hass, hmip_config_entry, mock_connection_init): async def test_hap_create_exception(
hass: HomeAssistant, hmip_config_entry, mock_connection_init
) -> None:
"""Mock AsyncHome to execute get_hap.""" """Mock AsyncHome to execute get_hap."""
hass.config.components.add(HMIPC_DOMAIN) hass.config.components.add(HMIPC_DOMAIN)
@ -145,7 +151,7 @@ async def test_hap_create_exception(hass, hmip_config_entry, mock_connection_ini
await hap.async_setup() await hap.async_setup()
async def test_auth_create(hass, simple_mock_auth): async def test_auth_create(hass: HomeAssistant, simple_mock_auth) -> None:
"""Mock AsyncAuth to execute get_auth.""" """Mock AsyncAuth to execute get_auth."""
config = {HMIPC_HAPID: HAPID, HMIPC_PIN: HAPPIN, HMIPC_NAME: "hmip"} config = {HMIPC_HAPID: HAPID, HMIPC_PIN: HAPPIN, HMIPC_NAME: "hmip"}
hmip_auth = HomematicipAuth(hass, config) hmip_auth = HomematicipAuth(hass, config)
@ -160,7 +166,7 @@ async def test_auth_create(hass, simple_mock_auth):
assert hmip_auth.auth.pin == HAPPIN assert hmip_auth.auth.pin == HAPPIN
async def test_auth_create_exception(hass, simple_mock_auth): async def test_auth_create_exception(hass: HomeAssistant, simple_mock_auth) -> None:
"""Mock AsyncAuth to execute get_auth.""" """Mock AsyncAuth to execute get_auth."""
config = {HMIPC_HAPID: HAPID, HMIPC_PIN: HAPPIN, HMIPC_NAME: "hmip"} config = {HMIPC_HAPID: HAPID, HMIPC_PIN: HAPPIN, HMIPC_NAME: "hmip"}
hmip_auth = HomematicipAuth(hass, config) hmip_auth = HomematicipAuth(hass, config)

20
tests/components/homematicip_cloud/test_init.py
View File

@ -21,8 +21,8 @@ from tests.common import MockConfigEntry
async def test_config_with_accesspoint_passed_to_config_entry( async def test_config_with_accesspoint_passed_to_config_entry(
hass, mock_connection, simple_mock_home hass: HomeAssistant, mock_connection, simple_mock_home
): ) -> None:
"""Test that config for a accesspoint are loaded via config entry.""" """Test that config for a accesspoint are loaded via config entry."""
entry_config = { entry_config = {
@ -55,8 +55,8 @@ async def test_config_with_accesspoint_passed_to_config_entry(
async def test_config_already_registered_not_passed_to_config_entry( async def test_config_already_registered_not_passed_to_config_entry(
hass, simple_mock_home hass: HomeAssistant, simple_mock_home
): ) -> None:
"""Test that an already registered accesspoint does not get imported.""" """Test that an already registered accesspoint does not get imported."""
mock_config = {HMIPC_AUTHTOKEN: "123", HMIPC_HAPID: "ABC123", HMIPC_NAME: "name"} mock_config = {HMIPC_AUTHTOKEN: "123", HMIPC_HAPID: "ABC123", HMIPC_NAME: "name"}
@ -99,8 +99,8 @@ async def test_config_already_registered_not_passed_to_config_entry(
async def test_load_entry_fails_due_to_connection_error( async def test_load_entry_fails_due_to_connection_error(
hass, hmip_config_entry, mock_connection_init hass: HomeAssistant, hmip_config_entry, mock_connection_init
): ) -> None:
"""Test load entry fails due to connection error.""" """Test load entry fails due to connection error."""
hmip_config_entry.add_to_hass(hass) hmip_config_entry.add_to_hass(hass)
@ -114,7 +114,9 @@ async def test_load_entry_fails_due_to_connection_error(
assert hmip_config_entry.state is ConfigEntryState.SETUP_RETRY assert hmip_config_entry.state is ConfigEntryState.SETUP_RETRY
async def test_load_entry_fails_due_to_generic_exception(hass, hmip_config_entry): async def test_load_entry_fails_due_to_generic_exception(
hass: HomeAssistant, hmip_config_entry
) -> None:
"""Test load entry fails due to generic exception.""" """Test load entry fails due to generic exception."""
hmip_config_entry.add_to_hass(hass) hmip_config_entry.add_to_hass(hass)
@ -160,7 +162,9 @@ async def test_unload_entry(hass: HomeAssistant) -> None:
assert hass.data[HMIPC_DOMAIN] == {} assert hass.data[HMIPC_DOMAIN] == {}
async def test_hmip_dump_hap_config_services(hass, mock_hap_with_service): async def test_hmip_dump_hap_config_services(
hass: HomeAssistant, mock_hap_with_service
) -> None:
"""Test dump configuration services.""" """Test dump configuration services."""
with patch("pathlib.Path.write_text", return_value=Mock()) as write_mock: with patch("pathlib.Path.write_text", return_value=Mock()) as write_mock:

28
tests/components/homematicip_cloud/test_light.py
View File

@ -26,7 +26,7 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_light(hass, default_mock_hap_factory): async def test_hmip_light(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipLight.""" """Test HomematicipLight."""
entity_id = "light.treppe_ch" entity_id = "light.treppe_ch"
entity_name = "Treppe CH" entity_name = "Treppe CH"
@ -71,7 +71,9 @@ async def test_hmip_light(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_notification_light(hass, default_mock_hap_factory): async def test_hmip_notification_light(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipNotificationLight.""" """Test HomematicipNotificationLight."""
entity_id = "light.alarm_status" entity_id = "light.alarm_status"
entity_name = "Alarm Status" entity_name = "Alarm Status"
@ -168,7 +170,7 @@ async def test_hmip_notification_light(hass, default_mock_hap_factory):
assert not ha_state.attributes.get(ATTR_BRIGHTNESS) assert not ha_state.attributes.get(ATTR_BRIGHTNESS)
async def test_hmip_dimmer(hass, default_mock_hap_factory): async def test_hmip_dimmer(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipDimmer.""" """Test HomematicipDimmer."""
entity_id = "light.schlafzimmerlicht" entity_id = "light.schlafzimmerlicht"
entity_name = "Schlafzimmerlicht" entity_name = "Schlafzimmerlicht"
@ -226,7 +228,9 @@ async def test_hmip_dimmer(hass, default_mock_hap_factory):
assert not ha_state.attributes.get(ATTR_BRIGHTNESS) assert not ha_state.attributes.get(ATTR_BRIGHTNESS)
async def test_hmip_light_measuring(hass, default_mock_hap_factory): async def test_hmip_light_measuring(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipLightMeasuring.""" """Test HomematicipLightMeasuring."""
entity_id = "light.flur_oben" entity_id = "light.flur_oben"
entity_name = "Flur oben" entity_name = "Flur oben"
@ -270,7 +274,9 @@ async def test_hmip_light_measuring(hass, default_mock_hap_factory):
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_wired_multi_dimmer(hass, default_mock_hap_factory): async def test_hmip_wired_multi_dimmer(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipMultiDimmer.""" """Test HomematicipMultiDimmer."""
entity_id = "light.raumlich_kuche" entity_id = "light.raumlich_kuche"
entity_name = "Raumlich (Küche)" entity_name = "Raumlich (Küche)"
@ -328,7 +334,9 @@ async def test_hmip_wired_multi_dimmer(hass, default_mock_hap_factory):
assert not ha_state.attributes.get(ATTR_BRIGHTNESS) assert not ha_state.attributes.get(ATTR_BRIGHTNESS)
async def test_hmip_din_rail_dimmer_3_channel1(hass, default_mock_hap_factory): async def test_hmip_din_rail_dimmer_3_channel1(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicIP DinRailDimmer3 Channel 1.""" """Test HomematicIP DinRailDimmer3 Channel 1."""
entity_id = "light.3_dimmer_channel1" entity_id = "light.3_dimmer_channel1"
entity_name = "3-Dimmer Channel1" entity_name = "3-Dimmer Channel1"
@ -385,7 +393,9 @@ async def test_hmip_din_rail_dimmer_3_channel1(hass, default_mock_hap_factory):
assert not ha_state.attributes.get(ATTR_BRIGHTNESS) assert not ha_state.attributes.get(ATTR_BRIGHTNESS)
async def test_hmip_din_rail_dimmer_3_channel2(hass, default_mock_hap_factory): async def test_hmip_din_rail_dimmer_3_channel2(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicIP DinRailDimmer3 Channel 2.""" """Test HomematicIP DinRailDimmer3 Channel 2."""
entity_id = "light.3_dimmer_channel2" entity_id = "light.3_dimmer_channel2"
entity_name = "3-Dimmer Channel2" entity_name = "3-Dimmer Channel2"
@ -442,7 +452,9 @@ async def test_hmip_din_rail_dimmer_3_channel2(hass, default_mock_hap_factory):
assert not ha_state.attributes.get(ATTR_BRIGHTNESS) assert not ha_state.attributes.get(ATTR_BRIGHTNESS)
async def test_hmip_din_rail_dimmer_3_channel3(hass, default_mock_hap_factory): async def test_hmip_din_rail_dimmer_3_channel3(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicIP DinRailDimmer3 Channel 3.""" """Test HomematicIP DinRailDimmer3 Channel 3."""
entity_id = "light.esstisch" entity_id = "light.esstisch"
entity_name = "Esstisch" entity_name = "Esstisch"

66
tests/components/homematicip_cloud/test_sensor.py
View File

@ -46,7 +46,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_accesspoint_status(hass, default_mock_hap_factory): async def test_hmip_accesspoint_status(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSwitch.""" """Test HomematicipSwitch."""
entity_id = "sensor.home_control_access_point_duty_cycle" entity_id = "sensor.home_control_access_point_duty_cycle"
entity_name = "HOME_CONTROL_ACCESS_POINT Duty Cycle" entity_name = "HOME_CONTROL_ACCESS_POINT Duty Cycle"
@ -63,7 +65,9 @@ async def test_hmip_accesspoint_status(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_UNIT_OF_MEASUREMENT] == PERCENTAGE assert ha_state.attributes[ATTR_UNIT_OF_MEASUREMENT] == PERCENTAGE
async def test_hmip_heating_thermostat(hass, default_mock_hap_factory): async def test_hmip_heating_thermostat(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingThermostat.""" """Test HomematicipHeatingThermostat."""
entity_id = "sensor.heizkorperthermostat_heating" entity_id = "sensor.heizkorperthermostat_heating"
entity_name = "Heizkörperthermostat Heating" entity_name = "Heizkörperthermostat Heating"
@ -97,7 +101,9 @@ async def test_hmip_heating_thermostat(hass, default_mock_hap_factory):
assert ha_state.attributes["icon"] == "mdi:battery-outline" assert ha_state.attributes["icon"] == "mdi:battery-outline"
async def test_hmip_humidity_sensor(hass, default_mock_hap_factory): async def test_hmip_humidity_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHumiditySensor.""" """Test HomematicipHumiditySensor."""
entity_id = "sensor.bwth_1_humidity" entity_id = "sensor.bwth_1_humidity"
entity_name = "BWTH 1 Humidity" entity_name = "BWTH 1 Humidity"
@ -120,7 +126,9 @@ async def test_hmip_humidity_sensor(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_RSSI_PEER] == -77 assert ha_state.attributes[ATTR_RSSI_PEER] == -77
async def test_hmip_temperature_sensor1(hass, default_mock_hap_factory): async def test_hmip_temperature_sensor1(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTemperatureSensor.""" """Test HomematicipTemperatureSensor."""
entity_id = "sensor.bwth_1_temperature" entity_id = "sensor.bwth_1_temperature"
entity_name = "BWTH 1 Temperature" entity_name = "BWTH 1 Temperature"
@ -145,7 +153,9 @@ async def test_hmip_temperature_sensor1(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10 assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10
async def test_hmip_temperature_sensor2(hass, default_mock_hap_factory): async def test_hmip_temperature_sensor2(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTemperatureSensor.""" """Test HomematicipTemperatureSensor."""
entity_id = "sensor.heizkorperthermostat_temperature" entity_id = "sensor.heizkorperthermostat_temperature"
entity_name = "Heizkörperthermostat Temperature" entity_name = "Heizkörperthermostat Temperature"
@ -170,7 +180,9 @@ async def test_hmip_temperature_sensor2(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10 assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10
async def test_hmip_temperature_sensor3(hass, default_mock_hap_factory): async def test_hmip_temperature_sensor3(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTemperatureSensor.""" """Test HomematicipTemperatureSensor."""
entity_id = "sensor.raumbediengerat_analog_temperature" entity_id = "sensor.raumbediengerat_analog_temperature"
entity_name = "Raumbediengerät Analog Temperature" entity_name = "Raumbediengerät Analog Temperature"
@ -195,7 +207,9 @@ async def test_hmip_temperature_sensor3(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10 assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 10
async def test_hmip_thermostat_evo_heating(hass, default_mock_hap_factory): async def test_hmip_thermostat_evo_heating(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipHeatingThermostat for HmIP-eTRV-E.""" """Test HomematicipHeatingThermostat for HmIP-eTRV-E."""
entity_id = "sensor.thermostat_evo_heating" entity_id = "sensor.thermostat_evo_heating"
entity_name = "thermostat_evo Heating" entity_name = "thermostat_evo Heating"
@ -215,7 +229,9 @@ async def test_hmip_thermostat_evo_heating(hass, default_mock_hap_factory):
assert ha_state.state == "40" assert ha_state.state == "40"
async def test_hmip_thermostat_evo_temperature(hass, default_mock_hap_factory): async def test_hmip_thermostat_evo_temperature(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTemperatureSensor.""" """Test HomematicipTemperatureSensor."""
entity_id = "sensor.thermostat_evo_temperature" entity_id = "sensor.thermostat_evo_temperature"
entity_name = "thermostat_evo Temperature" entity_name = "thermostat_evo Temperature"
@ -239,7 +255,7 @@ async def test_hmip_thermostat_evo_temperature(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 0.7 assert ha_state.attributes[ATTR_TEMPERATURE_OFFSET] == 0.7
async def test_hmip_power_sensor(hass, default_mock_hap_factory): async def test_hmip_power_sensor(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipPowerSensor.""" """Test HomematicipPowerSensor."""
entity_id = "sensor.flur_oben_power" entity_id = "sensor.flur_oben_power"
entity_name = "Flur oben Power" entity_name = "Flur oben Power"
@ -276,7 +292,9 @@ async def test_hmip_power_sensor(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_CONFIG_PENDING] assert ha_state.attributes[ATTR_CONFIG_PENDING]
async def test_hmip_illuminance_sensor1(hass, default_mock_hap_factory): async def test_hmip_illuminance_sensor1(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipIlluminanceSensor.""" """Test HomematicipIlluminanceSensor."""
entity_id = "sensor.wettersensor_illuminance" entity_id = "sensor.wettersensor_illuminance"
entity_name = "Wettersensor Illuminance" entity_name = "Wettersensor Illuminance"
@ -296,7 +314,9 @@ async def test_hmip_illuminance_sensor1(hass, default_mock_hap_factory):
assert ha_state.state == "231" assert ha_state.state == "231"
async def test_hmip_illuminance_sensor2(hass, default_mock_hap_factory): async def test_hmip_illuminance_sensor2(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipIlluminanceSensor.""" """Test HomematicipIlluminanceSensor."""
entity_id = "sensor.lichtsensor_nord_illuminance" entity_id = "sensor.lichtsensor_nord_illuminance"
entity_name = "Lichtsensor Nord Illuminance" entity_name = "Lichtsensor Nord Illuminance"
@ -319,7 +339,9 @@ async def test_hmip_illuminance_sensor2(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_LOWEST_ILLUMINATION] == 785.2 assert ha_state.attributes[ATTR_LOWEST_ILLUMINATION] == 785.2
async def test_hmip_windspeed_sensor(hass, default_mock_hap_factory): async def test_hmip_windspeed_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipWindspeedSensor.""" """Test HomematicipWindspeedSensor."""
entity_id = "sensor.wettersensor_pro_windspeed" entity_id = "sensor.wettersensor_pro_windspeed"
entity_name = "Wettersensor - pro Windspeed" entity_name = "Wettersensor - pro Windspeed"
@ -368,7 +390,9 @@ async def test_hmip_windspeed_sensor(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_WIND_DIRECTION] == txt assert ha_state.attributes[ATTR_WIND_DIRECTION] == txt
async def test_hmip_today_rain_sensor(hass, default_mock_hap_factory): async def test_hmip_today_rain_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTodayRainSensor.""" """Test HomematicipTodayRainSensor."""
entity_id = "sensor.weather_sensor_plus_today_rain" entity_id = "sensor.weather_sensor_plus_today_rain"
entity_name = "Weather Sensor – plus Today Rain" entity_name = "Weather Sensor – plus Today Rain"
@ -389,8 +413,8 @@ async def test_hmip_today_rain_sensor(hass, default_mock_hap_factory):
async def test_hmip_temperature_external_sensor_channel_1( async def test_hmip_temperature_external_sensor_channel_1(
hass, default_mock_hap_factory hass: HomeAssistant, default_mock_hap_factory
): ) -> None:
"""Test HomematicipTemperatureDifferenceSensor Channel 1 HmIP-STE2-PCB.""" """Test HomematicipTemperatureDifferenceSensor Channel 1 HmIP-STE2-PCB."""
entity_id = "sensor.ste2_channel_1_temperature" entity_id = "sensor.ste2_channel_1_temperature"
entity_name = "STE2 Channel 1 Temperature" entity_name = "STE2 Channel 1 Temperature"
@ -414,8 +438,8 @@ async def test_hmip_temperature_external_sensor_channel_1(
async def test_hmip_temperature_external_sensor_channel_2( async def test_hmip_temperature_external_sensor_channel_2(
hass, default_mock_hap_factory hass: HomeAssistant, default_mock_hap_factory
): ) -> None:
"""Test HomematicipTemperatureDifferenceSensor Channel 2 HmIP-STE2-PCB.""" """Test HomematicipTemperatureDifferenceSensor Channel 2 HmIP-STE2-PCB."""
entity_id = "sensor.ste2_channel_2_temperature" entity_id = "sensor.ste2_channel_2_temperature"
entity_name = "STE2 Channel 2 Temperature" entity_name = "STE2 Channel 2 Temperature"
@ -438,7 +462,9 @@ async def test_hmip_temperature_external_sensor_channel_2(
assert ha_state.state == "23.4" assert ha_state.state == "23.4"
async def test_hmip_temperature_external_sensor_delta(hass, default_mock_hap_factory): async def test_hmip_temperature_external_sensor_delta(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipTemperatureDifferenceSensor Delta HmIP-STE2-PCB.""" """Test HomematicipTemperatureDifferenceSensor Delta HmIP-STE2-PCB."""
entity_id = "sensor.ste2_delta_temperature" entity_id = "sensor.ste2_delta_temperature"
entity_name = "STE2 Delta Temperature" entity_name = "STE2 Delta Temperature"
@ -463,7 +489,9 @@ async def test_hmip_temperature_external_sensor_delta(hass, default_mock_hap_fac
assert ha_state.state == "-0.5" assert ha_state.state == "-0.5"
async def test_hmip_passage_detector_delta_counter(hass, default_mock_hap_factory): async def test_hmip_passage_detector_delta_counter(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipPassageDetectorDeltaCounter.""" """Test HomematicipPassageDetectorDeltaCounter."""
entity_id = "sensor.spdr_1" entity_id = "sensor.spdr_1"
entity_name = "SPDR_1" entity_name = "SPDR_1"

16
tests/components/homematicip_cloud/test_switch.py
View File

@ -19,7 +19,7 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_switch(hass, default_mock_hap_factory): async def test_hmip_switch(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipSwitch.""" """Test HomematicipSwitch."""
entity_id = "switch.schrank" entity_id = "switch.schrank"
entity_name = "Schrank" entity_name = "Schrank"
@ -56,7 +56,7 @@ async def test_hmip_switch(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_switch_input(hass, default_mock_hap_factory): async def test_hmip_switch_input(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipSwitch.""" """Test HomematicipSwitch."""
entity_id = "switch.wohnzimmer_beleuchtung" entity_id = "switch.wohnzimmer_beleuchtung"
entity_name = "Wohnzimmer Beleuchtung" entity_name = "Wohnzimmer Beleuchtung"
@ -93,7 +93,9 @@ async def test_hmip_switch_input(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_switch_measuring(hass, default_mock_hap_factory): async def test_hmip_switch_measuring(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipSwitchMeasuring.""" """Test HomematicipSwitchMeasuring."""
entity_id = "switch.pc" entity_id = "switch.pc"
entity_name = "Pc" entity_name = "Pc"
@ -131,7 +133,7 @@ async def test_hmip_switch_measuring(hass, default_mock_hap_factory):
assert ha_state.state == STATE_ON assert ha_state.state == STATE_ON
async def test_hmip_group_switch(hass, default_mock_hap_factory): async def test_hmip_group_switch(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipGroupSwitch.""" """Test HomematicipGroupSwitch."""
entity_id = "switch.strom_group" entity_id = "switch.strom_group"
entity_name = "Strom Group" entity_name = "Strom Group"
@ -171,7 +173,7 @@ async def test_hmip_group_switch(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_GROUP_MEMBER_UNREACHABLE] assert ha_state.attributes[ATTR_GROUP_MEMBER_UNREACHABLE]
async def test_hmip_multi_switch(hass, default_mock_hap_factory): async def test_hmip_multi_switch(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipMultiSwitch.""" """Test HomematicipMultiSwitch."""
entity_id = "switch.jalousien_1_kizi_2_schlazi_channel1" entity_id = "switch.jalousien_1_kizi_2_schlazi_channel1"
entity_name = "Jalousien - 1 KiZi, 2 SchlaZi Channel1" entity_name = "Jalousien - 1 KiZi, 2 SchlaZi Channel1"
@ -224,7 +226,9 @@ async def test_hmip_multi_switch(hass, default_mock_hap_factory):
assert ha_state.state == STATE_OFF assert ha_state.state == STATE_OFF
async def test_hmip_wired_multi_switch(hass, default_mock_hap_factory): async def test_hmip_wired_multi_switch(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipMultiSwitch.""" """Test HomematicipMultiSwitch."""
entity_id = "switch.fernseher_wohnzimmer" entity_id = "switch.fernseher_wohnzimmer"
entity_name = "Fernseher (Wohnzimmer)" entity_name = "Fernseher (Wohnzimmer)"

10
tests/components/homematicip_cloud/test_weather.py
View File

@ -22,7 +22,9 @@ async def test_manually_configured_platform(hass: HomeAssistant) -> None:
assert not hass.data.get(HMIPC_DOMAIN) assert not hass.data.get(HMIPC_DOMAIN)
async def test_hmip_weather_sensor(hass, default_mock_hap_factory): async def test_hmip_weather_sensor(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipWeatherSensor.""" """Test HomematicipWeatherSensor."""
entity_id = "weather.weather_sensor_plus" entity_id = "weather.weather_sensor_plus"
entity_name = "Weather Sensor – plus" entity_name = "Weather Sensor – plus"
@ -46,7 +48,9 @@ async def test_hmip_weather_sensor(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_WEATHER_TEMPERATURE] == 12.1 assert ha_state.attributes[ATTR_WEATHER_TEMPERATURE] == 12.1
async def test_hmip_weather_sensor_pro(hass, default_mock_hap_factory): async def test_hmip_weather_sensor_pro(
hass: HomeAssistant, default_mock_hap_factory
) -> None:
"""Test HomematicipWeatherSensorPro.""" """Test HomematicipWeatherSensorPro."""
entity_id = "weather.wettersensor_pro" entity_id = "weather.wettersensor_pro"
entity_name = "Wettersensor - pro" entity_name = "Wettersensor - pro"
@ -71,7 +75,7 @@ async def test_hmip_weather_sensor_pro(hass, default_mock_hap_factory):
assert ha_state.attributes[ATTR_WEATHER_TEMPERATURE] == 12.1 assert ha_state.attributes[ATTR_WEATHER_TEMPERATURE] == 12.1
async def test_hmip_home_weather(hass, default_mock_hap_factory): async def test_hmip_home_weather(hass: HomeAssistant, default_mock_hap_factory) -> None:
"""Test HomematicipHomeWeather.""" """Test HomematicipHomeWeather."""
entity_id = "weather.weather_1010_wien_osterreich" entity_id = "weather.weather_1010_wien_osterreich"
entity_name = "Weather 1010 Wien, Österreich" entity_name = "Weather 1010 Wien, Österreich"

21
tests/components/homewizard/test_button.py
View File

@ -6,6 +6,7 @@ import pytest
from homeassistant.components import button from homeassistant.components import button
from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNKNOWN from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
@ -13,8 +14,8 @@ from .generator import get_mock_device
async def test_identify_button_entity_not_loaded_when_not_available( async def test_identify_button_entity_not_loaded_when_not_available(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Does not load button when device has no support for it.""" """Does not load button when device has no support for it."""
api = get_mock_device(product_type="HWE-P1") api = get_mock_device(product_type="HWE-P1")
@ -34,8 +35,8 @@ async def test_identify_button_entity_not_loaded_when_not_available(
async def test_identify_button_is_loaded( async def test_identify_button_is_loaded(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Loads button when device has support.""" """Loads button when device has support."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")
@ -64,7 +65,9 @@ async def test_identify_button_is_loaded(
assert entry.unique_id == "aabbccddeeff_identify" assert entry.unique_id == "aabbccddeeff_identify"
async def test_identify_press(hass, mock_config_entry_data, mock_config_entry): async def test_identify_press(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test button press is handled correctly.""" """Test button press is handled correctly."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")
@ -96,8 +99,8 @@ async def test_identify_press(hass, mock_config_entry_data, mock_config_entry):
async def test_identify_press_catches_requesterror( async def test_identify_press_catches_requesterror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test button press is handled RequestError correctly.""" """Test button press is handled RequestError correctly."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")
@ -134,8 +137,8 @@ async def test_identify_press_catches_requesterror(
async def test_identify_press_catches_disablederror( async def test_identify_press_catches_disablederror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test button press is handled DisabledError correctly.""" """Test button press is handled DisabledError correctly."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")

4
tests/components/homewizard/test_config_flow.py
View File

@ -116,7 +116,7 @@ async def test_discovery_flow_works(
async def test_discovery_flow_during_onboarding( async def test_discovery_flow_during_onboarding(
hass, aioclient_mock: AiohttpClientMocker, mock_onboarding: MagicMock hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_onboarding: MagicMock
) -> None: ) -> None:
"""Test discovery setup flow during onboarding.""" """Test discovery setup flow during onboarding."""
@ -159,7 +159,7 @@ async def test_discovery_flow_during_onboarding(
async def test_discovery_flow_during_onboarding_disabled_api( async def test_discovery_flow_during_onboarding_disabled_api(
hass, aioclient_mock: AiohttpClientMocker, mock_onboarding: MagicMock hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, mock_onboarding: MagicMock
) -> None: ) -> None:
"""Test discovery setup flow during onboarding with a disabled API.""" """Test discovery setup flow during onboarding with a disabled API."""

2
tests/components/homewizard/test_diagnostics.py
View File

@ -12,7 +12,7 @@ async def test_diagnostics(
hass: HomeAssistant, hass: HomeAssistant,
hass_client: ClientSessionGenerator, hass_client: ClientSessionGenerator,
init_integration: MockConfigEntry, init_integration: MockConfigEntry,
): ) -> None:
"""Test diagnostics.""" """Test diagnostics."""
assert await get_diagnostics_for_config_entry( assert await get_diagnostics_for_config_entry(
hass, hass_client, init_integration hass, hass_client, init_integration

30
tests/components/homewizard/test_init.py
View File

@ -7,13 +7,17 @@ from homewizard_energy.errors import DisabledError, HomeWizardEnergyException
from homeassistant.components.homewizard.const import DOMAIN from homeassistant.components.homewizard.const import DOMAIN
from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntryState from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntryState
from homeassistant.const import CONF_IP_ADDRESS from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant
from .generator import get_mock_device from .generator import get_mock_device
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
from tests.test_util.aiohttp import AiohttpClientMocker
async def test_load_unload(aioclient_mock, hass): async def test_load_unload(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test loading and unloading of integration.""" """Test loading and unloading of integration."""
device = get_mock_device() device = get_mock_device()
@ -41,7 +45,9 @@ async def test_load_unload(aioclient_mock, hass):
assert entry.state is ConfigEntryState.NOT_LOADED assert entry.state is ConfigEntryState.NOT_LOADED
async def test_load_failed_host_unavailable(aioclient_mock, hass): async def test_load_failed_host_unavailable(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test setup handles unreachable host.""" """Test setup handles unreachable host."""
def MockInitialize(): def MockInitialize():
@ -68,7 +74,9 @@ async def test_load_failed_host_unavailable(aioclient_mock, hass):
assert entry.state is ConfigEntryState.SETUP_RETRY assert entry.state is ConfigEntryState.SETUP_RETRY
async def test_load_detect_api_disabled(aioclient_mock, hass): async def test_load_detect_api_disabled(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test setup detects disabled API.""" """Test setup detects disabled API."""
def MockInitialize(): def MockInitialize():
@ -106,7 +114,9 @@ async def test_load_detect_api_disabled(aioclient_mock, hass):
assert flow["context"].get("entry_id") == entry.entry_id assert flow["context"].get("entry_id") == entry.entry_id
async def test_load_removes_reauth_flow(aioclient_mock, hass): async def test_load_removes_reauth_flow(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test setup removes reauth flow when API is enabled.""" """Test setup removes reauth flow when API is enabled."""
device = get_mock_device() device = get_mock_device()
@ -141,7 +151,9 @@ async def test_load_removes_reauth_flow(aioclient_mock, hass):
assert len(flows) == 0 assert len(flows) == 0
async def test_load_handles_homewizardenergy_exception(aioclient_mock, hass): async def test_load_handles_homewizardenergy_exception(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test setup handles exception from API.""" """Test setup handles exception from API."""
def MockInitialize(): def MockInitialize():
@ -168,7 +180,9 @@ async def test_load_handles_homewizardenergy_exception(aioclient_mock, hass):
assert entry.state is ConfigEntryState.SETUP_RETRY or ConfigEntryState.SETUP_ERROR assert entry.state is ConfigEntryState.SETUP_RETRY or ConfigEntryState.SETUP_ERROR
async def test_load_handles_generic_exception(aioclient_mock, hass): async def test_load_handles_generic_exception(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test setup handles global exception.""" """Test setup handles global exception."""
def MockInitialize(): def MockInitialize():
@ -195,7 +209,9 @@ async def test_load_handles_generic_exception(aioclient_mock, hass):
assert entry.state is ConfigEntryState.SETUP_RETRY or ConfigEntryState.SETUP_ERROR assert entry.state is ConfigEntryState.SETUP_RETRY or ConfigEntryState.SETUP_ERROR
async def test_load_handles_initialization_error(aioclient_mock, hass): async def test_load_handles_initialization_error(
aioclient_mock: AiohttpClientMocker, hass: HomeAssistant
) -> None:
"""Test handles non-exception error.""" """Test handles non-exception error."""
device = get_mock_device() device = get_mock_device()

26
tests/components/homewizard/test_number.py
View File

@ -1,5 +1,4 @@
"""Test the update coordinator for HomeWizard.""" """Test the update coordinator for HomeWizard."""
from unittest.mock import AsyncMock, patch from unittest.mock import AsyncMock, patch
from homewizard_energy.errors import DisabledError, RequestError from homewizard_energy.errors import DisabledError, RequestError
@ -9,6 +8,7 @@ import pytest
from homeassistant.components import number from homeassistant.components import number
from homeassistant.components.number import ATTR_VALUE, SERVICE_SET_VALUE from homeassistant.components.number import ATTR_VALUE, SERVICE_SET_VALUE
from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
@ -16,8 +16,8 @@ from .generator import get_mock_device
async def test_number_entity_not_loaded_when_not_available( async def test_number_entity_not_loaded_when_not_available(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity does not load number when brightness is not available.""" """Test entity does not load number when brightness is not available."""
api = get_mock_device() api = get_mock_device()
@ -39,7 +39,9 @@ async def test_number_entity_not_loaded_when_not_available(
) )
async def test_number_loads_entities(hass, mock_config_entry_data, mock_config_entry): async def test_number_loads_entities(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test entity does load number when brightness is available.""" """Test entity does load number when brightness is available."""
api = get_mock_device() api = get_mock_device()
@ -74,7 +76,9 @@ async def test_number_loads_entities(hass, mock_config_entry_data, mock_config_e
assert not entry.disabled assert not entry.disabled
async def test_brightness_level_set(hass, mock_config_entry_data, mock_config_entry): async def test_brightness_level_set(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test entity turns sets light level.""" """Test entity turns sets light level."""
api = get_mock_device() api = get_mock_device()
@ -149,8 +153,8 @@ async def test_brightness_level_set(hass, mock_config_entry_data, mock_config_en
async def test_brightness_level_set_catches_requesterror( async def test_brightness_level_set_catches_requesterror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity raises HomeAssistantError when RequestError was raised.""" """Test entity raises HomeAssistantError when RequestError was raised."""
api = get_mock_device() api = get_mock_device()
@ -185,8 +189,8 @@ async def test_brightness_level_set_catches_requesterror(
async def test_brightness_level_set_catches_disablederror( async def test_brightness_level_set_catches_disablederror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity raises HomeAssistantError when DisabledError was raised.""" """Test entity raises HomeAssistantError when DisabledError was raised."""
api = get_mock_device() api = get_mock_device()
@ -221,8 +225,8 @@ async def test_brightness_level_set_catches_disablederror(
async def test_brightness_level_set_catches_invalid_value( async def test_brightness_level_set_catches_invalid_value(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity raises ValueError when value was invalid.""" """Test entity raises ValueError when value was invalid."""
api = get_mock_device() api = get_mock_device()

158
tests/components/homewizard/test_sensor.py
View File

@ -1,5 +1,4 @@
"""Test the update coordinator for HomeWizard.""" """Test the update coordinator for HomeWizard."""
from datetime import timedelta from datetime import timedelta
from unittest.mock import AsyncMock, patch from unittest.mock import AsyncMock, patch
@ -24,6 +23,7 @@ from homeassistant.const import (
UnitOfPower, UnitOfPower,
UnitOfVolume, UnitOfVolume,
) )
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
import homeassistant.util.dt as dt_util 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( 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.""" """Test entity loads smr version."""
api = get_mock_device() api = get_mock_device()
@ -71,8 +71,8 @@ async def test_sensor_entity_smr_version(
async def test_sensor_entity_meter_model( 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.""" """Test entity loads meter model."""
api = get_mock_device() api = get_mock_device()
@ -111,8 +111,8 @@ async def test_sensor_entity_meter_model(
async def test_sensor_entity_unique_meter_id( 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.""" """Test entity loads unique meter id."""
api = get_mock_device() 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" 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.""" """Test entity loads wifi ssid."""
api = get_mock_device() 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( 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.""" """Test entity loads active_tariff."""
api = get_mock_device() api = get_mock_device()
@ -226,8 +228,8 @@ async def test_sensor_entity_active_tariff(
async def test_sensor_entity_wifi_strength( 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.""" """Test entity loads wifi strength."""
api = get_mock_device() 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( 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.""" """Test entity loads total power import t1."""
api = get_mock_device() 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( 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.""" """Test entity loads total power import t2."""
api = get_mock_device() 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( 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.""" """Test entity loads total power export t1."""
api = get_mock_device() 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( 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.""" """Test entity loads total power export t2."""
api = get_mock_device() 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( 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.""" """Test entity loads active power."""
api = get_mock_device() api = get_mock_device()
@ -459,8 +461,8 @@ async def test_sensor_entity_active_power(
async def test_sensor_entity_active_power_l1( 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.""" """Test entity loads active power l1."""
api = get_mock_device() api = get_mock_device()
@ -499,8 +501,8 @@ async def test_sensor_entity_active_power_l1(
async def test_sensor_entity_active_power_l2( 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.""" """Test entity loads active power l2."""
api = get_mock_device() api = get_mock_device()
@ -539,8 +541,8 @@ async def test_sensor_entity_active_power_l2(
async def test_sensor_entity_active_power_l3( 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.""" """Test entity loads active power l3."""
api = get_mock_device() api = get_mock_device()
@ -578,7 +580,9 @@ async def test_sensor_entity_active_power_l3(
assert ATTR_ICON not in state.attributes 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.""" """Test entity loads total gas."""
api = get_mock_device() 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( 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.""" """Test entity loads unique gas meter id."""
api = get_mock_device() 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( 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.""" """Test entity loads active voltage l1."""
api = get_mock_device() api = get_mock_device()
@ -714,8 +718,8 @@ async def test_sensor_entity_active_voltage_l1(
async def test_sensor_entity_active_voltage_l2( 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.""" """Test entity loads active voltage l2."""
api = get_mock_device() api = get_mock_device()
@ -773,8 +777,8 @@ async def test_sensor_entity_active_voltage_l2(
async def test_sensor_entity_active_voltage_l3( 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.""" """Test entity loads active voltage l3."""
api = get_mock_device() api = get_mock_device()
@ -832,8 +836,8 @@ async def test_sensor_entity_active_voltage_l3(
async def test_sensor_entity_active_current_l1( 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.""" """Test entity loads active current l1."""
api = get_mock_device() api = get_mock_device()
@ -891,8 +895,8 @@ async def test_sensor_entity_active_current_l1(
async def test_sensor_entity_active_current_l2( 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.""" """Test entity loads active current l2."""
api = get_mock_device() api = get_mock_device()
@ -950,8 +954,8 @@ async def test_sensor_entity_active_current_l2(
async def test_sensor_entity_active_current_l3( 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.""" """Test entity loads active current l3."""
api = get_mock_device() api = get_mock_device()
@ -1009,8 +1013,8 @@ async def test_sensor_entity_active_current_l3(
async def test_sensor_entity_active_frequency( 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.""" """Test entity loads active frequency."""
api = get_mock_device() api = get_mock_device()
@ -1065,8 +1069,8 @@ async def test_sensor_entity_active_frequency(
async def test_sensor_entity_voltage_sag_count_l1( 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.""" """Test entity loads voltage_sag_count_l1."""
api = get_mock_device() 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( 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.""" """Test entity loads voltage_sag_count_l2."""
api = get_mock_device() 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( 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.""" """Test entity loads voltage_sag_count_l3."""
api = get_mock_device() 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( 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.""" """Test entity loads voltage_swell_count_l1."""
api = get_mock_device() 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( 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.""" """Test entity loads voltage_swell_count_l2."""
api = get_mock_device() 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( 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.""" """Test entity loads voltage_swell_count_l3."""
api = get_mock_device() 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( 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.""" """Test entity loads any power fail count."""
api = get_mock_device() 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( 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.""" """Test entity loads long power fail count."""
api = get_mock_device() 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( 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.""" """Test entity loads active power average."""
api = get_mock_device() api = get_mock_device()
@ -1428,8 +1432,8 @@ async def test_sensor_entity_active_power_average(
async def test_sensor_entity_monthly_power_peak( 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.""" """Test entity loads monthly power peak."""
api = get_mock_device() api = get_mock_device()
@ -1471,8 +1475,8 @@ async def test_sensor_entity_monthly_power_peak(
async def test_sensor_entity_active_liters( 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).""" """Test entity loads active liters (watermeter)."""
api = get_mock_device() api = get_mock_device()
@ -1512,8 +1516,8 @@ async def test_sensor_entity_active_liters(
async def test_sensor_entity_total_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).""" """Test entity loads total liters (watermeter)."""
api = get_mock_device() api = get_mock_device()
@ -1553,8 +1557,8 @@ async def test_sensor_entity_total_liters(
async def test_sensor_entity_disabled_when_null( 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.""" """Test sensor disables data with null by default."""
api = get_mock_device() 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( 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.""" """Test sensor disables export if value is 0."""
api = get_mock_device() api = get_mock_device()
@ -1641,7 +1645,9 @@ async def test_sensor_entity_export_disabled_when_unused(
assert entry.disabled 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.""" """Test sensor handles api unreachable."""
api = get_mock_device() 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.""" """Test sensor handles api unreachable."""
api = get_mock_device() api = get_mock_device()

34
tests/components/homewizard/test_switch.py
View File

@ -1,5 +1,4 @@
"""Test the update coordinator for HomeWizard.""" """Test the update coordinator for HomeWizard."""
from unittest.mock import AsyncMock, patch from unittest.mock import AsyncMock, patch
from homewizard_energy.errors import DisabledError, RequestError from homewizard_energy.errors import DisabledError, RequestError
@ -18,6 +17,7 @@ from homeassistant.const import (
STATE_ON, STATE_ON,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
) )
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
@ -25,8 +25,8 @@ from .generator import get_mock_device
async def test_switch_entity_not_loaded_when_not_available( async def test_switch_entity_not_loaded_when_not_available(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity loads smr version.""" """Test entity loads smr version."""
api = get_mock_device() api = get_mock_device()
@ -49,7 +49,9 @@ async def test_switch_entity_not_loaded_when_not_available(
assert state_switch_lock is None assert state_switch_lock is None
async def test_switch_loads_entities(hass, mock_config_entry_data, mock_config_entry): async def test_switch_loads_entities(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test entity loads smr version.""" """Test entity loads smr version."""
api = get_mock_device() api = get_mock_device()
@ -102,7 +104,9 @@ async def test_switch_loads_entities(hass, mock_config_entry_data, mock_config_e
assert ATTR_DEVICE_CLASS not in state_switch_lock.attributes assert ATTR_DEVICE_CLASS not in state_switch_lock.attributes
async def test_switch_power_on_off(hass, mock_config_entry_data, mock_config_entry): async def test_switch_power_on_off(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test entity turns switch on and off.""" """Test entity turns switch on and off."""
api = get_mock_device() api = get_mock_device()
@ -156,8 +160,8 @@ async def test_switch_power_on_off(hass, mock_config_entry_data, mock_config_ent
async def test_switch_lock_power_on_off( async def test_switch_lock_power_on_off(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity turns switch on and off.""" """Test entity turns switch on and off."""
api = get_mock_device() api = get_mock_device()
@ -220,8 +224,8 @@ async def test_switch_lock_power_on_off(
async def test_switch_lock_sets_power_on_unavailable( async def test_switch_lock_sets_power_on_unavailable(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity turns switch on and off.""" """Test entity turns switch on and off."""
api = get_mock_device() api = get_mock_device()
@ -289,7 +293,9 @@ async def test_switch_lock_sets_power_on_unavailable(
assert len(api.state_set.mock_calls) == 2 assert len(api.state_set.mock_calls) == 2
async def test_cloud_connection_on_off(hass, mock_config_entry_data, mock_config_entry): async def test_cloud_connection_on_off(
hass: HomeAssistant, mock_config_entry_data, mock_config_entry
) -> None:
"""Test entity turns switch on and off.""" """Test entity turns switch on and off."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")
@ -350,8 +356,8 @@ async def test_cloud_connection_on_off(hass, mock_config_entry_data, mock_config
async def test_switch_handles_requesterror( async def test_switch_handles_requesterror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity raises HomeAssistantError when RequestError was raised.""" """Test entity raises HomeAssistantError when RequestError was raised."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")
@ -427,8 +433,8 @@ async def test_switch_handles_requesterror(
async def test_switch_handles_disablederror( async def test_switch_handles_disablederror(
hass, mock_config_entry_data, mock_config_entry hass: HomeAssistant, mock_config_entry_data, mock_config_entry
): ) -> None:
"""Test entity raises HomeAssistantError when Disabled was raised.""" """Test entity raises HomeAssistantError when Disabled was raised."""
api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02") api = get_mock_device(product_type="HWE-SKT", firmware_version="3.02")

10
tests/components/honeywell/test_init.py
View File

@ -1,8 +1,8 @@
"""Test honeywell setup process.""" """Test honeywell setup process."""
from unittest.mock import create_autospec, patch from unittest.mock import create_autospec, patch
import aiosomecomfort import aiosomecomfort
import pytest
from homeassistant.components.honeywell.const import ( from homeassistant.components.honeywell.const import (
CONF_COOL_AWAY_TEMPERATURE, CONF_COOL_AWAY_TEMPERATURE,
@ -19,7 +19,7 @@ MIGRATE_OPTIONS_KEYS = {CONF_COOL_AWAY_TEMPERATURE, CONF_HEAT_AWAY_TEMPERATURE}
@patch("homeassistant.components.honeywell.UPDATE_LOOP_SLEEP_TIME", 0) @patch("homeassistant.components.honeywell.UPDATE_LOOP_SLEEP_TIME", 0)
async def test_setup_entry(hass: HomeAssistant, config_entry: MockConfigEntry): async def test_setup_entry(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
"""Initialize the config entry.""" """Initialize the config entry."""
config_entry.add_to_hass(hass) config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(config_entry.entry_id) await hass.config_entries.async_setup(config_entry.entry_id)
@ -43,7 +43,11 @@ async def test_setup_multiple_thermostats(
@patch("homeassistant.components.honeywell.UPDATE_LOOP_SLEEP_TIME", 0) @patch("homeassistant.components.honeywell.UPDATE_LOOP_SLEEP_TIME", 0)
async def test_setup_multiple_thermostats_with_same_deviceid( async def test_setup_multiple_thermostats_with_same_deviceid(
hass: HomeAssistant, caplog, config_entry: MockConfigEntry, device, client hass: HomeAssistant,
caplog: pytest.LogCaptureFixture,
config_entry: MockConfigEntry,
device,
client,
) -> None: ) -> None:
"""Test Honeywell TCC API returning duplicate device IDs.""" """Test Honeywell TCC API returning duplicate device IDs."""
mock_location2 = create_autospec(aiosomecomfort.Location, instance=True) mock_location2 = create_autospec(aiosomecomfort.Location, instance=True)

2
tests/components/honeywell/test_sensor.py
View File

@ -16,7 +16,7 @@ async def test_outdoor_sensor(
device_with_outdoor_sensor: Device, device_with_outdoor_sensor: Device,
unit, unit,
temp, temp,
): ) -> None:
"""Test outdoor temperature sensor.""" """Test outdoor temperature sensor."""
device_with_outdoor_sensor.temperature_unit = unit device_with_outdoor_sensor.temperature_unit = unit
location.devices_by_id[ location.devices_by_id[

121
tests/components/http/test_auth.py
View File

@ -13,6 +13,9 @@ import yarl
from homeassistant.auth.const import GROUP_ID_READ_ONLY from homeassistant.auth.const import GROUP_ID_READ_ONLY
from homeassistant.auth.models import User from homeassistant.auth.models import User
from homeassistant.auth.providers import trusted_networks from homeassistant.auth.providers import trusted_networks
from homeassistant.auth.providers.legacy_api_password import (
LegacyApiPasswordAuthProvider,
)
from homeassistant.components import websocket_api from homeassistant.components import websocket_api
from homeassistant.components.http.auth import ( from homeassistant.components.http.auth import (
CONTENT_USER_NAME, CONTENT_USER_NAME,
@ -34,6 +37,9 @@ from homeassistant.setup import async_setup_component
from . import HTTP_HEADER_HA_AUTH, mock_real_ip from . import HTTP_HEADER_HA_AUTH, mock_real_ip
from tests.common import MockUser
from tests.typing import ClientSessionGenerator, WebSocketGenerator
API_PASSWORD = "test-password" API_PASSWORD = "test-password"
# Don't add 127.0.0.1/::1 as trusted, as it may interfere with other test cases # Don't add 127.0.0.1/::1 as trusted, as it may interfere with other test cases
@ -107,8 +113,11 @@ async def test_auth_middleware_loaded_by_default(hass: HomeAssistant) -> None:
async def test_cant_access_with_password_in_header( async def test_cant_access_with_password_in_header(
app, aiohttp_client, legacy_auth, hass app,
): aiohttp_client: ClientSessionGenerator,
legacy_auth: LegacyApiPasswordAuthProvider,
hass: HomeAssistant,
) -> None:
"""Test access with password in header.""" """Test access with password in header."""
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
client = await aiohttp_client(app) client = await aiohttp_client(app)
@ -121,8 +130,11 @@ async def test_cant_access_with_password_in_header(
async def test_cant_access_with_password_in_query( async def test_cant_access_with_password_in_query(
app, aiohttp_client, legacy_auth, hass app,
): aiohttp_client: ClientSessionGenerator,
legacy_auth: LegacyApiPasswordAuthProvider,
hass: HomeAssistant,
) -> None:
"""Test access with password in URL.""" """Test access with password in URL."""
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
client = await aiohttp_client(app) client = await aiohttp_client(app)
@ -137,7 +149,12 @@ async def test_cant_access_with_password_in_query(
assert resp.status == HTTPStatus.UNAUTHORIZED assert resp.status == HTTPStatus.UNAUTHORIZED
async def test_basic_auth_does_not_work(app, aiohttp_client, hass, legacy_auth): async def test_basic_auth_does_not_work(
app,
aiohttp_client: ClientSessionGenerator,
hass: HomeAssistant,
legacy_auth: LegacyApiPasswordAuthProvider,
) -> None:
"""Test access with basic authentication.""" """Test access with basic authentication."""
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
client = await aiohttp_client(app) client = await aiohttp_client(app)
@ -156,8 +173,12 @@ async def test_basic_auth_does_not_work(app, aiohttp_client, hass, legacy_auth):
async def test_cannot_access_with_trusted_ip( async def test_cannot_access_with_trusted_ip(
hass, app2, trusted_networks_auth, aiohttp_client, hass_owner_user hass: HomeAssistant,
): app2,
trusted_networks_auth,
aiohttp_client: ClientSessionGenerator,
hass_owner_user: MockUser,
) -> None:
"""Test access with an untrusted ip address.""" """Test access with an untrusted ip address."""
await async_setup_auth(hass, app2) await async_setup_auth(hass, app2)
@ -180,8 +201,11 @@ async def test_cannot_access_with_trusted_ip(
async def test_auth_active_access_with_access_token_in_header( async def test_auth_active_access_with_access_token_in_header(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with access token in header.""" """Test access with access token in header."""
token = hass_access_token token = hass_access_token
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
@ -213,8 +237,12 @@ async def test_auth_active_access_with_access_token_in_header(
async def test_auth_active_access_with_trusted_ip( async def test_auth_active_access_with_trusted_ip(
hass, app2, trusted_networks_auth, aiohttp_client, hass_owner_user hass: HomeAssistant,
): app2,
trusted_networks_auth,
aiohttp_client: ClientSessionGenerator,
hass_owner_user: MockUser,
) -> None:
"""Test access with an untrusted ip address.""" """Test access with an untrusted ip address."""
await async_setup_auth(hass, app2) await async_setup_auth(hass, app2)
@ -237,8 +265,11 @@ async def test_auth_active_access_with_trusted_ip(
async def test_auth_legacy_support_api_password_cannot_access( async def test_auth_legacy_support_api_password_cannot_access(
app, aiohttp_client, legacy_auth, hass app,
): aiohttp_client: ClientSessionGenerator,
legacy_auth: LegacyApiPasswordAuthProvider,
hass: HomeAssistant,
) -> None:
"""Test access using api_password if auth.support_legacy.""" """Test access using api_password if auth.support_legacy."""
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
client = await aiohttp_client(app) client = await aiohttp_client(app)
@ -254,8 +285,11 @@ async def test_auth_legacy_support_api_password_cannot_access(
async def test_auth_access_signed_path_with_refresh_token( async def test_auth_access_signed_path_with_refresh_token(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with signed url.""" """Test access with signed url."""
app.router.add_post("/", mock_handler) app.router.add_post("/", mock_handler)
app.router.add_get("/another_path", mock_handler) app.router.add_get("/another_path", mock_handler)
@ -296,8 +330,11 @@ async def test_auth_access_signed_path_with_refresh_token(
async def test_auth_access_signed_path_with_query_param( async def test_auth_access_signed_path_with_query_param(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with signed url and query params.""" """Test access with signed url and query params."""
app.router.add_post("/", mock_handler) app.router.add_post("/", mock_handler)
app.router.add_get("/another_path", mock_handler) app.router.add_get("/another_path", mock_handler)
@ -317,8 +354,11 @@ async def test_auth_access_signed_path_with_query_param(
async def test_auth_access_signed_path_with_query_param_order( async def test_auth_access_signed_path_with_query_param_order(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with signed url and query params different order.""" """Test access with signed url and query params different order."""
app.router.add_post("/", mock_handler) app.router.add_post("/", mock_handler)
app.router.add_get("/another_path", mock_handler) app.router.add_get("/another_path", mock_handler)
@ -343,8 +383,11 @@ async def test_auth_access_signed_path_with_query_param_order(
async def test_auth_access_signed_path_with_query_param_safe_param( async def test_auth_access_signed_path_with_query_param_safe_param(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with signed url and changing a safe param.""" """Test access with signed url and changing a safe param."""
app.router.add_post("/", mock_handler) app.router.add_post("/", mock_handler)
app.router.add_get("/another_path", mock_handler) app.router.add_get("/another_path", mock_handler)
@ -377,8 +420,13 @@ async def test_auth_access_signed_path_with_query_param_safe_param(
], ],
) )
async def test_auth_access_signed_path_with_query_param_tamper( async def test_auth_access_signed_path_with_query_param_tamper(
hass, app, aiohttp_client, hass_access_token, base_url: str, test_url: str hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
base_url: str,
test_url: str,
) -> None:
"""Test access with signed url and query params that have been tampered with.""" """Test access with signed url and query params that have been tampered with."""
app.router.add_post("/", mock_handler) app.router.add_post("/", mock_handler)
app.router.add_get("/another_path", mock_handler) app.router.add_get("/another_path", mock_handler)
@ -398,8 +446,11 @@ async def test_auth_access_signed_path_with_query_param_tamper(
async def test_auth_access_signed_path_via_websocket( async def test_auth_access_signed_path_via_websocket(
hass, app, hass_ws_client, hass_read_only_access_token hass: HomeAssistant,
): app,
hass_ws_client: WebSocketGenerator,
hass_read_only_access_token: str,
) -> None:
"""Test signed url via websockets uses connection user.""" """Test signed url via websockets uses connection user."""
@websocket_api.websocket_command({"type": "diagnostics/list"}) @websocket_api.websocket_command({"type": "diagnostics/list"})
@ -435,8 +486,11 @@ async def test_auth_access_signed_path_via_websocket(
async def test_auth_access_signed_path_with_http( async def test_auth_access_signed_path_with_http(
hass, app, aiohttp_client, hass_access_token hass: HomeAssistant,
): app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test signed url via HTTP uses HTTP user.""" """Test signed url via HTTP uses HTTP user."""
setup_request_context(app, current_request) setup_request_context(app, current_request)
@ -467,7 +521,9 @@ async def test_auth_access_signed_path_with_http(
assert claims["iss"] == refresh_token.id assert claims["iss"] == refresh_token.id
async def test_auth_access_signed_path_with_content_user(hass, app, aiohttp_client): async def test_auth_access_signed_path_with_content_user(
hass: HomeAssistant, app, aiohttp_client: ClientSessionGenerator
) -> None:
"""Test access signed url uses content user.""" """Test access signed url uses content user."""
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
signed_path = async_sign_path(hass, "/", timedelta(seconds=5)) signed_path = async_sign_path(hass, "/", timedelta(seconds=5))
@ -481,7 +537,12 @@ async def test_auth_access_signed_path_with_content_user(hass, app, aiohttp_clie
assert claims["iss"] == hass.data[STORAGE_KEY] assert claims["iss"] == hass.data[STORAGE_KEY]
async def test_local_only_user_rejected(hass, app, aiohttp_client, hass_access_token): async def test_local_only_user_rejected(
hass: HomeAssistant,
app,
aiohttp_client: ClientSessionGenerator,
hass_access_token: str,
) -> None:
"""Test access with access token in header.""" """Test access with access token in header."""
token = hass_access_token token = hass_access_token
await async_setup_auth(hass, app) await async_setup_auth(hass, app)
@ -501,7 +562,7 @@ async def test_local_only_user_rejected(hass, app, aiohttp_client, hass_access_t
assert req.status == HTTPStatus.UNAUTHORIZED assert req.status == HTTPStatus.UNAUTHORIZED
async def test_async_user_not_allowed_do_auth(hass, app): async def test_async_user_not_allowed_do_auth(hass: HomeAssistant, app) -> None:
"""Test for not allowing auth.""" """Test for not allowing auth."""
user = await hass.auth.async_create_user("Hello") user = await hass.auth.async_create_user("Hello")
user.is_active = False user.is_active = False

27
tests/components/http/test_ban.py
View File

@ -76,8 +76,10 @@ async def test_access_from_banned_ip(
async def test_access_from_banned_ip_with_partially_broken_yaml_file( async def test_access_from_banned_ip_with_partially_broken_yaml_file(
hass, aiohttp_client, caplog hass: HomeAssistant,
): aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test accessing to server from banned IP. Both trusted and not. """Test accessing to server from banned IP. Both trusted and not.
We inject some garbage into the yaml file to make sure it can We inject some garbage into the yaml file to make sure it can
@ -150,7 +152,11 @@ async def test_failure_loading_ip_bans_file(
assert resp.status == HTTPStatus.NOT_FOUND assert resp.status == HTTPStatus.NOT_FOUND
async def test_ip_ban_manager_never_started(hass, aiohttp_client, caplog): async def test_ip_ban_manager_never_started(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test we handle the ip ban manager not being started.""" """Test we handle the ip ban manager not being started."""
app = web.Application() app = web.Application()
app["hass"] = hass app["hass"] = hass
@ -183,8 +189,13 @@ async def test_ip_ban_manager_never_started(hass, aiohttp_client, caplog):
), ),
) )
async def test_access_from_supervisor_ip( async def test_access_from_supervisor_ip(
remote_addr, bans, status, hass, aiohttp_client, hassio_env remote_addr,
): bans,
status,
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
hassio_env,
) -> None:
"""Test accessing to server from supervisor IP.""" """Test accessing to server from supervisor IP."""
app = web.Application() app = web.Application()
app["hass"] = hass app["hass"] = hass
@ -251,7 +262,11 @@ async def test_ban_middleware_loaded_by_default(hass: HomeAssistant) -> None:
assert len(mock_setup.mock_calls) == 1 assert len(mock_setup.mock_calls) == 1
async def test_ip_bans_file_creation(hass, aiohttp_client, caplog): async def test_ip_bans_file_creation(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Testing if banned IP file created.""" """Testing if banned IP file created."""
app = web.Application() app = web.Application()
app["hass"] = hass app["hass"] = hass

4
tests/components/http/test_cors.py
View File

@ -60,7 +60,7 @@ def client(event_loop, aiohttp_client):
return event_loop.run_until_complete(aiohttp_client(app)) return event_loop.run_until_complete(aiohttp_client(app))
async def test_cors_requests(client): async def test_cors_requests(client) -> None:
"""Test cross origin requests.""" """Test cross origin requests."""
req = await client.get("/", headers={ORIGIN: TRUSTED_ORIGIN}) req = await client.get("/", headers={ORIGIN: TRUSTED_ORIGIN})
assert req.status == HTTPStatus.OK assert req.status == HTTPStatus.OK
@ -88,7 +88,7 @@ async def test_cors_requests(client):
assert req.headers[ACCESS_CONTROL_ALLOW_ORIGIN] == TRUSTED_ORIGIN assert req.headers[ACCESS_CONTROL_ALLOW_ORIGIN] == TRUSTED_ORIGIN
async def test_cors_preflight_allowed(client): async def test_cors_preflight_allowed(client) -> None:
"""Test cross origin resource sharing preflight (OPTIONS) request.""" """Test cross origin resource sharing preflight (OPTIONS) request."""
req = await client.options( req = await client.options(
"/", "/",

6
tests/components/http/test_data_validator.py
View File

@ -8,6 +8,8 @@ import voluptuous as vol
from homeassistant.components.http import HomeAssistantView from homeassistant.components.http import HomeAssistantView
from homeassistant.components.http.data_validator import RequestDataValidator from homeassistant.components.http.data_validator import RequestDataValidator
from tests.typing import ClientSessionGenerator
async def get_client(aiohttp_client, validator): async def get_client(aiohttp_client, validator):
"""Generate a client that hits a view decorated with validator.""" """Generate a client that hits a view decorated with validator."""
@ -30,7 +32,7 @@ async def get_client(aiohttp_client, validator):
return client return client
async def test_validator(aiohttp_client): async def test_validator(aiohttp_client: ClientSessionGenerator) -> None:
"""Test the validator.""" """Test the validator."""
client = await get_client( client = await get_client(
aiohttp_client, RequestDataValidator(vol.Schema({vol.Required("test"): str})) aiohttp_client, RequestDataValidator(vol.Schema({vol.Required("test"): str}))
@ -46,7 +48,7 @@ async def test_validator(aiohttp_client):
assert resp.status == HTTPStatus.BAD_REQUEST assert resp.status == HTTPStatus.BAD_REQUEST
async def test_validator_allow_empty(aiohttp_client): async def test_validator_allow_empty(aiohttp_client: ClientSessionGenerator) -> None:
"""Test the validator with empty data.""" """Test the validator with empty data."""
client = await get_client( client = await get_client(
aiohttp_client, aiohttp_client,

83
tests/components/http/test_forwarded.py
View File

@ -9,13 +9,17 @@ import pytest
from homeassistant.components.http.forwarded import async_setup_forwarded from homeassistant.components.http.forwarded import async_setup_forwarded
from tests.typing import ClientSessionGenerator
async def mock_handler(request): async def mock_handler(request):
"""Return the real IP as text.""" """Return the real IP as text."""
return web.Response(text=request.remote) return web.Response(text=request.remote)
async def test_x_forwarded_for_without_trusted_proxy(aiohttp_client, caplog): async def test_x_forwarded_for_without_trusted_proxy(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we get the IP from the transport.""" """Test that we get the IP from the transport."""
async def handler(request): async def handler(request):
@ -60,8 +64,8 @@ async def test_x_forwarded_for_without_trusted_proxy(aiohttp_client, caplog):
], ],
) )
async def test_x_forwarded_for_with_trusted_proxy( async def test_x_forwarded_for_with_trusted_proxy(
trusted_proxies, x_forwarded_for, remote, aiohttp_client trusted_proxies, x_forwarded_for, remote, aiohttp_client: ClientSessionGenerator
): ) -> None:
"""Test that we get the IP from the forwarded for header.""" """Test that we get the IP from the forwarded for header."""
async def handler(request): async def handler(request):
@ -85,7 +89,9 @@ async def test_x_forwarded_for_with_trusted_proxy(
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_for_disabled_with_proxy(aiohttp_client, caplog): async def test_x_forwarded_for_disabled_with_proxy(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we warn when processing is disabled, but proxy has been detected.""" """Test that we warn when processing is disabled, but proxy has been detected."""
async def handler(request): async def handler(request):
@ -112,7 +118,9 @@ async def test_x_forwarded_for_disabled_with_proxy(aiohttp_client, caplog):
) )
async def test_x_forwarded_for_with_spoofed_header(aiohttp_client): async def test_x_forwarded_for_with_spoofed_header(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that we get the IP from the transport with a spoofed header.""" """Test that we get the IP from the transport with a spoofed header."""
async def handler(request): async def handler(request):
@ -150,8 +158,10 @@ async def test_x_forwarded_for_with_spoofed_header(aiohttp_client):
], ],
) )
async def test_x_forwarded_for_with_malformed_header( async def test_x_forwarded_for_with_malformed_header(
x_forwarded_for, aiohttp_client, caplog x_forwarded_for,
): aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test that we get a HTTP 400 bad request with a malformed header.""" """Test that we get a HTTP 400 bad request with a malformed header."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -165,7 +175,9 @@ async def test_x_forwarded_for_with_malformed_header(
assert "Invalid IP address in X-Forwarded-For" in caplog.text assert "Invalid IP address in X-Forwarded-For" in caplog.text
async def test_x_forwarded_for_with_multiple_headers(aiohttp_client, caplog): async def test_x_forwarded_for_with_multiple_headers(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we get a HTTP 400 bad request with multiple headers.""" """Test that we get a HTTP 400 bad request with multiple headers."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -214,8 +226,12 @@ async def test_x_forwarded_for_with_multiple_headers(aiohttp_client, caplog):
], ],
) )
async def test_x_forwarded_proto_with_trusted_proxy( async def test_x_forwarded_proto_with_trusted_proxy(
x_forwarded_for, remote, x_forwarded_proto, secure, aiohttp_client x_forwarded_for,
): remote,
x_forwarded_proto,
secure,
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that we get the proto header if proxy is trusted.""" """Test that we get the proto header if proxy is trusted."""
async def handler(request): async def handler(request):
@ -241,7 +257,9 @@ async def test_x_forwarded_proto_with_trusted_proxy(
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_proto_with_trusted_proxy_multiple_for(aiohttp_client): async def test_x_forwarded_proto_with_trusted_proxy_multiple_for(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that we get the proto with 1 element in the proto, multiple in the for.""" """Test that we get the proto with 1 element in the proto, multiple in the for."""
async def handler(request): async def handler(request):
@ -269,7 +287,9 @@ async def test_x_forwarded_proto_with_trusted_proxy_multiple_for(aiohttp_client)
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_proto_not_processed_without_for(aiohttp_client): async def test_x_forwarded_proto_not_processed_without_for(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that proto header isn't processed without a for header.""" """Test that proto header isn't processed without a for header."""
async def handler(request): async def handler(request):
@ -291,7 +311,9 @@ async def test_x_forwarded_proto_not_processed_without_for(aiohttp_client):
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_proto_with_multiple_headers(aiohttp_client, caplog): async def test_x_forwarded_proto_with_multiple_headers(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we get a HTTP 400 bad request with multiple headers.""" """Test that we get a HTTP 400 bad request with multiple headers."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -316,8 +338,10 @@ async def test_x_forwarded_proto_with_multiple_headers(aiohttp_client, caplog):
["", ",", "https, , https", "https, https, "], ["", ",", "https, , https", "https, https, "],
) )
async def test_x_forwarded_proto_empty_element( async def test_x_forwarded_proto_empty_element(
x_forwarded_proto, aiohttp_client, caplog x_forwarded_proto,
): aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test that we get a HTTP 400 bad request with empty proto.""" """Test that we get a HTTP 400 bad request with empty proto."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -341,8 +365,13 @@ async def test_x_forwarded_proto_empty_element(
], ],
) )
async def test_x_forwarded_proto_incorrect_number_of_elements( async def test_x_forwarded_proto_incorrect_number_of_elements(
x_forwarded_for, x_forwarded_proto, expected, got, aiohttp_client, caplog x_forwarded_for,
): x_forwarded_proto,
expected,
got,
aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
) -> None:
"""Test that we get a HTTP 400 bad request with incorrect number of elements.""" """Test that we get a HTTP 400 bad request with incorrect number of elements."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -364,7 +393,9 @@ async def test_x_forwarded_proto_incorrect_number_of_elements(
) )
async def test_x_forwarded_host_with_trusted_proxy(aiohttp_client): async def test_x_forwarded_host_with_trusted_proxy(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that we get the host header if proxy is trusted.""" """Test that we get the host header if proxy is trusted."""
async def handler(request): async def handler(request):
@ -388,7 +419,9 @@ async def test_x_forwarded_host_with_trusted_proxy(aiohttp_client):
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_host_not_processed_without_for(aiohttp_client): async def test_x_forwarded_host_not_processed_without_for(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that host header isn't processed without a for header.""" """Test that host header isn't processed without a for header."""
async def handler(request): async def handler(request):
@ -410,7 +443,9 @@ async def test_x_forwarded_host_not_processed_without_for(aiohttp_client):
assert resp.status == HTTPStatus.OK assert resp.status == HTTPStatus.OK
async def test_x_forwarded_host_with_multiple_headers(aiohttp_client, caplog): async def test_x_forwarded_host_with_multiple_headers(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we get a HTTP 400 bad request with multiple headers.""" """Test that we get a HTTP 400 bad request with multiple headers."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -430,7 +465,9 @@ async def test_x_forwarded_host_with_multiple_headers(aiohttp_client, caplog):
assert "Too many headers for X-Forwarded-Host" in caplog.text assert "Too many headers for X-Forwarded-Host" in caplog.text
async def test_x_forwarded_host_with_empty_header(aiohttp_client, caplog): async def test_x_forwarded_host_with_empty_header(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that we get a HTTP 400 bad request with empty host value.""" """Test that we get a HTTP 400 bad request with empty host value."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)
@ -445,7 +482,9 @@ async def test_x_forwarded_host_with_empty_header(aiohttp_client, caplog):
assert "Empty value received in X-Forward-Host header" in caplog.text assert "Empty value received in X-Forward-Host header" in caplog.text
async def test_x_forwarded_cloud(aiohttp_client, caplog): async def test_x_forwarded_cloud(
aiohttp_client: ClientSessionGenerator, caplog: pytest.LogCaptureFixture
) -> None:
"""Test that cloud requests are not processed.""" """Test that cloud requests are not processed."""
app = web.Application() app = web.Application()
app.router.add_get("/", mock_handler) app.router.add_get("/", mock_handler)

47
tests/components/http/test_init.py
View File

@ -8,6 +8,9 @@ from unittest.mock import Mock, patch
import pytest import pytest
from homeassistant.auth.providers.legacy_api_password import (
LegacyApiPasswordAuthProvider,
)
import homeassistant.components.http as http import homeassistant.components.http as http
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.network import NoURLAvailableError from homeassistant.helpers.network import NoURLAvailableError
@ -16,6 +19,7 @@ from homeassistant.util import dt as dt_util
from homeassistant.util.ssl import server_context_intermediate, server_context_modern from homeassistant.util.ssl import server_context_intermediate, server_context_modern
from tests.common import async_fire_time_changed from tests.common import async_fire_time_changed
from tests.typing import ClientSessionGenerator
def _setup_broken_ssl_pem_files(tmpdir): def _setup_broken_ssl_pem_files(tmpdir):
@ -76,8 +80,8 @@ class TestView(http.HomeAssistantView):
async def test_registering_view_while_running( async def test_registering_view_while_running(
hass, aiohttp_client, aiohttp_unused_port hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, aiohttp_unused_port
): ) -> None:
"""Test that we can register a view while the server is running.""" """Test that we can register a view while the server is running."""
await async_setup_component( await async_setup_component(
hass, http.DOMAIN, {http.DOMAIN: {http.CONF_SERVER_PORT: aiohttp_unused_port()}} hass, http.DOMAIN, {http.DOMAIN: {http.CONF_SERVER_PORT: aiohttp_unused_port()}}
@ -88,7 +92,12 @@ async def test_registering_view_while_running(
hass.http.register_view(TestView) hass.http.register_view(TestView)
async def test_not_log_password(hass, hass_client_no_auth, caplog, legacy_auth): async def test_not_log_password(
hass: HomeAssistant,
hass_client_no_auth: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
legacy_auth: LegacyApiPasswordAuthProvider,
) -> None:
"""Test access with password doesn't get logged.""" """Test access with password doesn't get logged."""
assert await async_setup_component(hass, "api", {"http": {}}) assert await async_setup_component(hass, "api", {"http": {}})
client = await hass_client_no_auth() client = await hass_client_no_auth()
@ -141,7 +150,7 @@ async def test_proxy_config_only_trust_proxies(hass: HomeAssistant) -> None:
) )
async def test_ssl_profile_defaults_modern(hass, tmpdir): async def test_ssl_profile_defaults_modern(hass: HomeAssistant, tmpdir) -> None:
"""Test default ssl profile.""" """Test default ssl profile."""
cert_path, key_path, _ = await hass.async_add_executor_job( cert_path, key_path, _ = await hass.async_add_executor_job(
@ -166,7 +175,7 @@ async def test_ssl_profile_defaults_modern(hass, tmpdir):
assert len(mock_context.mock_calls) == 1 assert len(mock_context.mock_calls) == 1
async def test_ssl_profile_change_intermediate(hass, tmpdir): async def test_ssl_profile_change_intermediate(hass: HomeAssistant, tmpdir) -> None:
"""Test setting ssl profile to intermediate.""" """Test setting ssl profile to intermediate."""
cert_path, key_path, _ = await hass.async_add_executor_job( cert_path, key_path, _ = await hass.async_add_executor_job(
@ -197,7 +206,7 @@ async def test_ssl_profile_change_intermediate(hass, tmpdir):
assert len(mock_context.mock_calls) == 1 assert len(mock_context.mock_calls) == 1
async def test_ssl_profile_change_modern(hass, tmpdir): async def test_ssl_profile_change_modern(hass: HomeAssistant, tmpdir) -> None:
"""Test setting ssl profile to modern.""" """Test setting ssl profile to modern."""
cert_path, key_path, _ = await hass.async_add_executor_job( cert_path, key_path, _ = await hass.async_add_executor_job(
@ -228,7 +237,7 @@ async def test_ssl_profile_change_modern(hass, tmpdir):
assert len(mock_context.mock_calls) == 1 assert len(mock_context.mock_calls) == 1
async def test_peer_cert(hass, tmpdir): async def test_peer_cert(hass: HomeAssistant, tmpdir) -> None:
"""Test required peer cert.""" """Test required peer cert."""
cert_path, key_path, peer_cert_path = await hass.async_add_executor_job( cert_path, key_path, peer_cert_path = await hass.async_add_executor_job(
_setup_empty_ssl_pem_files, tmpdir _setup_empty_ssl_pem_files, tmpdir
@ -262,7 +271,9 @@ async def test_peer_cert(hass, tmpdir):
assert len(mock_load_verify_locations.mock_calls) == 1 assert len(mock_load_verify_locations.mock_calls) == 1
async def test_emergency_ssl_certificate_when_invalid(hass, tmpdir, caplog): async def test_emergency_ssl_certificate_when_invalid(
hass: HomeAssistant, tmpdir, caplog: pytest.LogCaptureFixture
) -> None:
"""Test http can startup with an emergency self signed cert when the current one is broken.""" """Test http can startup with an emergency self signed cert when the current one is broken."""
cert_path, key_path = await hass.async_add_executor_job( cert_path, key_path = await hass.async_add_executor_job(
@ -292,8 +303,8 @@ async def test_emergency_ssl_certificate_when_invalid(hass, tmpdir, caplog):
async def test_emergency_ssl_certificate_not_used_when_not_safe_mode( async def test_emergency_ssl_certificate_not_used_when_not_safe_mode(
hass, tmpdir, caplog hass: HomeAssistant, tmpdir, caplog: pytest.LogCaptureFixture
): ) -> None:
"""Test an emergency cert is only used in safe mode.""" """Test an emergency cert is only used in safe mode."""
cert_path, key_path = await hass.async_add_executor_job( cert_path, key_path = await hass.async_add_executor_job(
@ -309,8 +320,8 @@ async def test_emergency_ssl_certificate_not_used_when_not_safe_mode(
async def test_emergency_ssl_certificate_when_invalid_get_url_fails( async def test_emergency_ssl_certificate_when_invalid_get_url_fails(
hass, tmpdir, caplog hass: HomeAssistant, tmpdir, caplog: pytest.LogCaptureFixture
): ) -> None:
"""Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken. """Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken.
Ensure we can still start of we cannot determine the external url as well. Ensure we can still start of we cannot determine the external url as well.
@ -345,7 +356,9 @@ async def test_emergency_ssl_certificate_when_invalid_get_url_fails(
assert hass.http.site is not None assert hass.http.site is not None
async def test_invalid_ssl_and_cannot_create_emergency_cert(hass, tmpdir, caplog): async def test_invalid_ssl_and_cannot_create_emergency_cert(
hass: HomeAssistant, tmpdir, caplog: pytest.LogCaptureFixture
) -> None:
"""Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken.""" """Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken."""
cert_path, key_path = await hass.async_add_executor_job( cert_path, key_path = await hass.async_add_executor_job(
@ -375,8 +388,8 @@ async def test_invalid_ssl_and_cannot_create_emergency_cert(hass, tmpdir, caplog
async def test_invalid_ssl_and_cannot_create_emergency_cert_with_ssl_peer_cert( async def test_invalid_ssl_and_cannot_create_emergency_cert_with_ssl_peer_cert(
hass, tmpdir, caplog hass: HomeAssistant, tmpdir, caplog: pytest.LogCaptureFixture
): ) -> None:
"""Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken. """Test http falls back to no ssl when an emergency cert cannot be created when the configured one is broken.
When there is a peer cert verification and we cannot create When there is a peer cert verification and we cannot create
@ -422,7 +435,9 @@ async def test_cors_defaults(hass: HomeAssistant) -> None:
assert mock_setup.mock_calls[0][1][1] == ["https://cast.home-assistant.io"] assert mock_setup.mock_calls[0][1][1] == ["https://cast.home-assistant.io"]
async def test_storing_config(hass, aiohttp_client, aiohttp_unused_port): async def test_storing_config(
hass: HomeAssistant, aiohttp_client: ClientSessionGenerator, aiohttp_unused_port
) -> None:
"""Test that we store last working config.""" """Test that we store last working config."""
config = { config = {
http.CONF_SERVER_PORT: aiohttp_unused_port(), http.CONF_SERVER_PORT: aiohttp_unused_port(),

6
tests/components/http/test_request_context.py
View File

@ -6,8 +6,12 @@ from aiohttp import web
from homeassistant.components.http.request_context import setup_request_context from homeassistant.components.http.request_context import setup_request_context
from tests.typing import ClientSessionGenerator
async def test_request_context_middleware(aiohttp_client):
async def test_request_context_middleware(
aiohttp_client: ClientSessionGenerator,
) -> None:
"""Test that request context is set from middleware.""" """Test that request context is set from middleware."""
context = ContextVar("request", default=None) context = ContextVar("request", default=None)
app = web.Application() app = web.Application()

15
tests/components/http/test_security_filter.py
View File

@ -7,6 +7,8 @@ import urllib3
from homeassistant.components.http.security_filter import setup_security_filter from homeassistant.components.http.security_filter import setup_security_filter
from tests.typing import ClientSessionGenerator
async def mock_handler(request): async def mock_handler(request):
"""Return OK.""" """Return OK."""
@ -23,7 +25,9 @@ async def mock_handler(request):
("/", {"test": "123"}), ("/", {"test": "123"}),
], ],
) )
async def test_ok_requests(request_path, request_params, aiohttp_client): async def test_ok_requests(
request_path, request_params, aiohttp_client: ClientSessionGenerator
) -> None:
"""Test request paths that should not be filtered.""" """Test request paths that should not be filtered."""
app = web.Application() app = web.Application()
app.router.add_get("/{all:.*}", mock_handler) app.router.add_get("/{all:.*}", mock_handler)
@ -56,8 +60,13 @@ async def test_ok_requests(request_path, request_params, aiohttp_client):
], ],
) )
async def test_bad_requests( async def test_bad_requests(
request_path, request_params, fail_on_query_string, aiohttp_client, caplog, loop request_path,
): request_params,
fail_on_query_string,
aiohttp_client: ClientSessionGenerator,
caplog: pytest.LogCaptureFixture,
loop,
) -> None:
"""Test request paths that should be filtered.""" """Test request paths that should be filtered."""
app = web.Application() app = web.Application()
app.router.add_get("/{all:.*}", mock_handler) app.router.add_get("/{all:.*}", mock_handler)

10
tests/components/http/test_view.py
View File

@ -31,7 +31,7 @@ def mock_request_with_stopping():
return Mock(app={"hass": Mock(is_stopping=True)}, match_info={}) return Mock(app={"hass": Mock(is_stopping=True)}, match_info={})
async def test_invalid_json(caplog): async def test_invalid_json(caplog: pytest.LogCaptureFixture) -> None:
"""Test trying to return invalid JSON.""" """Test trying to return invalid JSON."""
with pytest.raises(HTTPInternalServerError): with pytest.raises(HTTPInternalServerError):
HomeAssistantView.json({"hello": Decimal("2.0")}) HomeAssistantView.json({"hello": Decimal("2.0")})
@ -48,7 +48,7 @@ async def test_nan_serialized_to_null() -> None:
assert json.loads(response.body.decode("utf-8")) is None assert json.loads(response.body.decode("utf-8")) is None
async def test_handling_unauthorized(mock_request): async def test_handling_unauthorized(mock_request) -> None:
"""Test handling unauth exceptions.""" """Test handling unauth exceptions."""
with pytest.raises(HTTPUnauthorized): with pytest.raises(HTTPUnauthorized):
await request_handler_factory( await request_handler_factory(
@ -56,7 +56,7 @@ async def test_handling_unauthorized(mock_request):
)(mock_request) )(mock_request)
async def test_handling_invalid_data(mock_request): async def test_handling_invalid_data(mock_request) -> None:
"""Test handling unauth exceptions.""" """Test handling unauth exceptions."""
with pytest.raises(HTTPBadRequest): with pytest.raises(HTTPBadRequest):
await request_handler_factory( await request_handler_factory(
@ -64,7 +64,7 @@ async def test_handling_invalid_data(mock_request):
)(mock_request) )(mock_request)
async def test_handling_service_not_found(mock_request): async def test_handling_service_not_found(mock_request) -> None:
"""Test handling unauth exceptions.""" """Test handling unauth exceptions."""
with pytest.raises(HTTPInternalServerError): with pytest.raises(HTTPInternalServerError):
await request_handler_factory( await request_handler_factory(
@ -73,7 +73,7 @@ async def test_handling_service_not_found(mock_request):
)(mock_request) )(mock_request)
async def test_not_running(mock_request_with_stopping): async def test_not_running(mock_request_with_stopping) -> None:
"""Test we get a 503 when not running.""" """Test we get a 503 when not running."""
response = await request_handler_factory( response = await request_handler_factory(
Mock(requires_auth=False), AsyncMock(side_effect=Unauthorized) Mock(requires_auth=False), AsyncMock(side_effect=Unauthorized)

35
tests/components/huawei_lte/test_config_flow.py
View File

@ -6,6 +6,7 @@ from huawei_lte_api.enums.user import LoginErrorEnum, LoginStateEnum, PasswordTy
import pytest import pytest
import requests.exceptions import requests.exceptions
from requests.exceptions import ConnectionError from requests.exceptions import ConnectionError
import requests_mock
from requests_mock import ANY from requests_mock import ANY
from homeassistant import config_entries, data_entry_flow from homeassistant import config_entries, data_entry_flow
@ -46,7 +47,9 @@ async def test_show_set_form(hass: HomeAssistant) -> None:
assert result["step_id"] == "user" assert result["step_id"] == "user"
async def test_urlize_plain_host(hass, requests_mock): async def test_urlize_plain_host(
hass: HomeAssistant, requests_mock: requests_mock.Mocker
) -> None:
"""Test that plain host or IP gets converted to a URL.""" """Test that plain host or IP gets converted to a URL."""
requests_mock.request(ANY, ANY, exc=ConnectionError()) requests_mock.request(ANY, ANY, exc=ConnectionError())
host = "192.168.100.1" host = "192.168.100.1"
@ -60,7 +63,9 @@ async def test_urlize_plain_host(hass, requests_mock):
assert user_input[CONF_URL] == f"http://{host}/" assert user_input[CONF_URL] == f"http://{host}/"
async def test_already_configured(hass, requests_mock, login_requests_mock): async def test_already_configured(
hass: HomeAssistant, requests_mock: requests_mock.Mocker, login_requests_mock
) -> None:
"""Test we reject already configured devices.""" """Test we reject already configured devices."""
MockConfigEntry( MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,
@ -90,7 +95,9 @@ async def test_already_configured(hass, requests_mock, login_requests_mock):
assert result["reason"] == "already_configured" assert result["reason"] == "already_configured"
async def test_connection_error(hass, requests_mock): async def test_connection_error(
hass: HomeAssistant, requests_mock: requests_mock.Mocker
) -> None:
"""Test we show user form on connection error.""" """Test we show user form on connection error."""
requests_mock.request(ANY, ANY, exc=ConnectionError()) requests_mock.request(ANY, ANY, exc=ConnectionError())
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
@ -168,8 +175,8 @@ def login_requests_mock(requests_mock):
), ),
) )
async def test_login_error( async def test_login_error(
hass, login_requests_mock, request_outcome, fixture_override, errors hass: HomeAssistant, login_requests_mock, request_outcome, fixture_override, errors
): ) -> None:
"""Test we show user form with appropriate error on response failure.""" """Test we show user form with appropriate error on response failure."""
login_requests_mock.request( login_requests_mock.request(
ANY, ANY,
@ -187,7 +194,7 @@ async def test_login_error(
assert result["errors"] == errors assert result["errors"] == errors
async def test_success(hass, login_requests_mock): async def test_success(hass: HomeAssistant, login_requests_mock) -> None:
"""Test successful flow provides entry creation data.""" """Test successful flow provides entry creation data."""
login_requests_mock.request( login_requests_mock.request(
ANY, ANY,
@ -262,8 +269,12 @@ async def test_success(hass, login_requests_mock):
), ),
) )
async def test_ssdp( async def test_ssdp(
hass, login_requests_mock, requests_mock_request_kwargs, upnp_data, expected_result hass: HomeAssistant,
): login_requests_mock,
requests_mock_request_kwargs,
upnp_data,
expected_result,
) -> None:
"""Test SSDP discovery initiates config properly.""" """Test SSDP discovery initiates config properly."""
url = FIXTURE_USER_INPUT[CONF_URL][:-1] # strip trailing slash for appending port url = FIXTURE_USER_INPUT[CONF_URL][:-1] # strip trailing slash for appending port
context = {"source": config_entries.SOURCE_SSDP} context = {"source": config_entries.SOURCE_SSDP}
@ -317,8 +328,12 @@ async def test_ssdp(
), ),
) )
async def test_reauth( async def test_reauth(
hass, login_requests_mock, login_response_text, expected_result, expected_entry_data hass: HomeAssistant,
): login_requests_mock,
login_response_text,
expected_result,
expected_entry_data,
) -> None:
"""Test reauth.""" """Test reauth."""
mock_entry_data = {**FIXTURE_USER_INPUT, CONF_PASSWORD: "invalid-password"} mock_entry_data = {**FIXTURE_USER_INPUT, CONF_PASSWORD: "invalid-password"}
entry = MockConfigEntry( entry = MockConfigEntry(

2
tests/components/huawei_lte/test_device_tracker.py
View File

@ -15,6 +15,6 @@ from homeassistant.components.huawei_lte import device_tracker
("foo_bar", "foo_bar"), ("foo_bar", "foo_bar"),
), ),
) )
def test_better_snakecase(value, expected): def test_better_snakecase(value, expected) -> None:
"""Test that better snakecase works better.""" """Test that better snakecase works better."""
assert device_tracker._better_snakecase(value) == expected assert device_tracker._better_snakecase(value) == expected

2
tests/components/huawei_lte/test_sensor.py
View File

@ -17,6 +17,6 @@ from homeassistant.const import (
(">=-51dBm", (-51, SIGNAL_STRENGTH_DECIBELS_MILLIWATT)), (">=-51dBm", (-51, SIGNAL_STRENGTH_DECIBELS_MILLIWATT)),
), ),
) )
def test_format_default(value, expected): def test_format_default(value, expected) -> None:
"""Test that default formatter copes with expected values.""" """Test that default formatter copes with expected values."""
assert sensor.format_default(value) == expected assert sensor.format_default(value) == expected

4
tests/components/huawei_lte/test_switches.py
View File

@ -123,7 +123,7 @@ async def test_turn_off_switch_wifi_guest_network(client, hass: HomeAssistant) -
) )
async def test_huawei_lte_wifi_guest_network_config_entry_when_ssid_is_str( async def test_huawei_lte_wifi_guest_network_config_entry_when_ssid_is_str(
client, hass: HomeAssistant client, hass: HomeAssistant
): ) -> None:
"""Test switch wifi guest network config entry when ssid is a str. """Test switch wifi guest network config entry when ssid is a str.
Issue #76244. Huawai models: H312-371, E5372 and E8372. Issue #76244. Huawai models: H312-371, E5372 and E8372.
@ -143,7 +143,7 @@ async def test_huawei_lte_wifi_guest_network_config_entry_when_ssid_is_str(
) )
async def test_huawei_lte_wifi_guest_network_config_entry_when_ssid_is_none( async def test_huawei_lte_wifi_guest_network_config_entry_when_ssid_is_none(
client, hass: HomeAssistant client, hass: HomeAssistant
): ) -> None:
"""Test switch wifi guest network config entry when ssid is a None. """Test switch wifi guest network config entry when ssid is a None.
Issue #76244. Issue #76244.

8
tests/components/hue/test_binary_sensor.py
View File

@ -1,11 +1,13 @@
"""Philips Hue binary_sensor platform tests for V2 bridge/api.""" """Philips Hue binary_sensor platform tests for V2 bridge/api."""
from homeassistant.core import HomeAssistant
from .conftest import setup_platform from .conftest import setup_platform
from .const import FAKE_BINARY_SENSOR, FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY from .const import FAKE_BINARY_SENSOR, FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY
async def test_binary_sensors(hass, mock_bridge_v2, v2_resources_test_data): async def test_binary_sensors(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if all v2 binary_sensors get created with correct features.""" """Test if all v2 binary_sensors get created with correct features."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -33,7 +35,7 @@ async def test_binary_sensors(hass, mock_bridge_v2, v2_resources_test_data):
assert sensor.attributes["device_class"] == "running" assert sensor.attributes["device_class"] == "running"
async def test_binary_sensor_add_update(hass, mock_bridge_v2): async def test_binary_sensor_add_update(hass: HomeAssistant, mock_bridge_v2) -> None:
"""Test if binary_sensor get added/updated from events.""" """Test if binary_sensor get added/updated from events."""
await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY]) await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY])
await setup_platform(hass, mock_bridge_v2, "binary_sensor") await setup_platform(hass, mock_bridge_v2, "binary_sensor")

8
tests/components/hue/test_bridge.py
View File

@ -17,7 +17,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
async def test_bridge_setup_v1(hass, mock_api_v1): async def test_bridge_setup_v1(hass: HomeAssistant, mock_api_v1) -> None:
"""Test a successful setup for V1 bridge.""" """Test a successful setup for V1 bridge."""
config_entry = Mock() config_entry = Mock()
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1}
@ -37,7 +37,7 @@ async def test_bridge_setup_v1(hass, mock_api_v1):
assert forward_entries == {"light", "binary_sensor", "sensor"} assert forward_entries == {"light", "binary_sensor", "sensor"}
async def test_bridge_setup_v2(hass, mock_api_v2): async def test_bridge_setup_v2(hass: HomeAssistant, mock_api_v2) -> None:
"""Test a successful setup for V2 bridge.""" """Test a successful setup for V2 bridge."""
config_entry = Mock() config_entry = Mock()
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 2} config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 2}
@ -87,7 +87,7 @@ async def test_bridge_setup_timeout(hass: HomeAssistant) -> None:
await hue_bridge.async_initialize_bridge() await hue_bridge.async_initialize_bridge()
async def test_reset_unloads_entry_if_setup(hass, mock_api_v1): async def test_reset_unloads_entry_if_setup(hass: HomeAssistant, mock_api_v1) -> None:
"""Test calling reset while the entry has been setup.""" """Test calling reset while the entry has been setup."""
config_entry = Mock() config_entry = Mock()
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1}
@ -113,7 +113,7 @@ async def test_reset_unloads_entry_if_setup(hass, mock_api_v1):
assert len(hass.services.async_services()) == 0 assert len(hass.services.async_services()) == 0
async def test_handle_unauthorized(hass, mock_api_v1): async def test_handle_unauthorized(hass: HomeAssistant, mock_api_v1) -> None:
"""Test handling an unauthorized error on update.""" """Test handling an unauthorized error on update."""
config_entry = Mock(async_setup=AsyncMock()) config_entry = Mock(async_setup=AsyncMock())
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1}

4
tests/components/hue/test_config_flow.py
View File

@ -353,8 +353,8 @@ async def test_import_with_no_config(
async def test_creating_entry_removes_entries_for_same_host_or_bridge( async def test_creating_entry_removes_entries_for_same_host_or_bridge(
hass, aioclient_mock hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
): ) -> None:
"""Test that we clean up entries for same host and bridge. """Test that we clean up entries for same host and bridge.
An IP can only hold a single bridge and a single bridge can only be An IP can only hold a single bridge and a single bridge can only be

8
tests/components/hue/test_device_trigger_v1.py
View File

@ -1,8 +1,8 @@
"""The tests for Philips Hue device triggers for V1 bridge.""" """The tests for Philips Hue device triggers for V1 bridge."""
from homeassistant.components import automation, hue from homeassistant.components import automation, hue
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.hue.v1 import device_trigger from homeassistant.components.hue.v1 import device_trigger
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from .conftest import setup_platform from .conftest import setup_platform
@ -13,7 +13,7 @@ from tests.common import assert_lists_same, async_get_device_automations
REMOTES_RESPONSE = {"7": HUE_TAP_REMOTE_1, "8": HUE_DIMMER_REMOTE_1} REMOTES_RESPONSE = {"7": HUE_TAP_REMOTE_1, "8": HUE_DIMMER_REMOTE_1}
async def test_get_triggers(hass, mock_bridge_v1, device_reg): async def test_get_triggers(hass: HomeAssistant, mock_bridge_v1, device_reg) -> None:
"""Test we get the expected triggers from a hue remote.""" """Test we get the expected triggers from a hue remote."""
mock_bridge_v1.mock_sensor_responses.append(REMOTES_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(REMOTES_RESPONSE)
await setup_platform(hass, mock_bridge_v1, ["sensor", "binary_sensor"]) await setup_platform(hass, mock_bridge_v1, ["sensor", "binary_sensor"])
@ -76,7 +76,9 @@ async def test_get_triggers(hass, mock_bridge_v1, device_reg):
assert_lists_same(triggers, expected_triggers) assert_lists_same(triggers, expected_triggers)
async def test_if_fires_on_state_change(hass, mock_bridge_v1, device_reg, calls): async def test_if_fires_on_state_change(
hass: HomeAssistant, mock_bridge_v1, device_reg, calls
) -> None:
"""Test for button press trigger firing.""" """Test for button press trigger firing."""
mock_bridge_v1.mock_sensor_responses.append(REMOTES_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(REMOTES_RESPONSE)
await setup_platform(hass, mock_bridge_v1, ["sensor", "binary_sensor"]) await setup_platform(hass, mock_bridge_v1, ["sensor", "binary_sensor"])

9
tests/components/hue/test_device_trigger_v2.py
View File

@ -5,6 +5,7 @@ from homeassistant.components import hue
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.hue.v2.device import async_setup_devices from homeassistant.components.hue.v2.device import async_setup_devices
from homeassistant.components.hue.v2.hue_event import async_setup_hue_events from homeassistant.components.hue.v2.hue_event import async_setup_hue_events
from homeassistant.core import HomeAssistant
from .conftest import setup_platform from .conftest import setup_platform
@ -15,7 +16,9 @@ from tests.common import (
) )
async def test_hue_event(hass, mock_bridge_v2, v2_resources_test_data): async def test_hue_event(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test hue button events.""" """Test hue button events."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
await setup_platform(hass, mock_bridge_v2, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v2, ["binary_sensor", "sensor"])
@ -43,7 +46,9 @@ async def test_hue_event(hass, mock_bridge_v2, v2_resources_test_data):
assert events[0].data["subtype"] == btn_event["metadata"]["control_id"] assert events[0].data["subtype"] == btn_event["metadata"]["control_id"]
async def test_get_triggers(hass, mock_bridge_v2, v2_resources_test_data, device_reg): async def test_get_triggers(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data, device_reg
) -> None:
"""Test we get the expected triggers from a hue remote.""" """Test we get the expected triggers from a hue remote."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
await setup_platform(hass, mock_bridge_v2, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v2, ["binary_sensor", "sensor"])

10
tests/components/hue/test_diagnostics.py
View File

@ -1,11 +1,15 @@
"""Test Hue diagnostics.""" """Test Hue diagnostics."""
from homeassistant.core import HomeAssistant
from .conftest import setup_platform from .conftest import setup_platform
from tests.components.diagnostics import get_diagnostics_for_config_entry from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator
async def test_diagnostics_v1(hass, hass_client, mock_bridge_v1): async def test_diagnostics_v1(
hass: HomeAssistant, hass_client: ClientSessionGenerator, mock_bridge_v1
) -> None:
"""Test diagnostics v1.""" """Test diagnostics v1."""
await setup_platform(hass, mock_bridge_v1, []) await setup_platform(hass, mock_bridge_v1, [])
config_entry = hass.config_entries.async_entries("hue")[0] config_entry = hass.config_entries.async_entries("hue")[0]
@ -13,7 +17,9 @@ async def test_diagnostics_v1(hass, hass_client, mock_bridge_v1):
assert result == {} assert result == {}
async def test_diagnostics_v2(hass, hass_client, mock_bridge_v2): async def test_diagnostics_v2(
hass: HomeAssistant, hass_client: ClientSessionGenerator, mock_bridge_v2
) -> None:
"""Test diagnostics v2.""" """Test diagnostics v2."""
mock_bridge_v2.api.get_diagnostics.return_value = {"hello": "world"} mock_bridge_v2.api.get_diagnostics.return_value = {"hello": "world"}
await setup_platform(hass, mock_bridge_v2, []) await setup_platform(hass, mock_bridge_v2, [])

16
tests/components/hue/test_init.py
View File

@ -44,7 +44,7 @@ async def test_setup_with_no_config(hass: HomeAssistant) -> None:
assert hue.DOMAIN not in hass.data assert hue.DOMAIN not in hass.data
async def test_unload_entry(hass, mock_bridge_setup): async def test_unload_entry(hass: HomeAssistant, mock_bridge_setup) -> None:
"""Test being able to unload an entry.""" """Test being able to unload an entry."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=hue.DOMAIN, data={"host": "0.0.0.0", "api_version": 2} domain=hue.DOMAIN, data={"host": "0.0.0.0", "api_version": 2}
@ -65,7 +65,7 @@ async def test_unload_entry(hass, mock_bridge_setup):
assert hue.DOMAIN not in hass.data assert hue.DOMAIN not in hass.data
async def test_setting_unique_id(hass, mock_bridge_setup): async def test_setting_unique_id(hass: HomeAssistant, mock_bridge_setup) -> None:
"""Test we set unique ID if not set yet.""" """Test we set unique ID if not set yet."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=hue.DOMAIN, data={"host": "0.0.0.0", "api_version": 2} domain=hue.DOMAIN, data={"host": "0.0.0.0", "api_version": 2}
@ -75,7 +75,9 @@ async def test_setting_unique_id(hass, mock_bridge_setup):
assert entry.unique_id == "mock-id" assert entry.unique_id == "mock-id"
async def test_fixing_unique_id_no_other(hass, mock_bridge_setup): async def test_fixing_unique_id_no_other(
hass: HomeAssistant, mock_bridge_setup
) -> None:
"""Test we set unique ID if not set yet.""" """Test we set unique ID if not set yet."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=hue.DOMAIN, domain=hue.DOMAIN,
@ -87,7 +89,9 @@ async def test_fixing_unique_id_no_other(hass, mock_bridge_setup):
assert entry.unique_id == "mock-id" assert entry.unique_id == "mock-id"
async def test_fixing_unique_id_other_ignored(hass, mock_bridge_setup): async def test_fixing_unique_id_other_ignored(
hass: HomeAssistant, mock_bridge_setup
) -> None:
"""Test we set unique ID if not set yet.""" """Test we set unique ID if not set yet."""
MockConfigEntry( MockConfigEntry(
domain=hue.DOMAIN, domain=hue.DOMAIN,
@ -107,7 +111,9 @@ async def test_fixing_unique_id_other_ignored(hass, mock_bridge_setup):
assert hass.config_entries.async_entries() == [entry] assert hass.config_entries.async_entries() == [entry]
async def test_fixing_unique_id_other_correct(hass, mock_bridge_setup): async def test_fixing_unique_id_other_correct(
hass: HomeAssistant, mock_bridge_setup
) -> None:
"""Test we remove config entry if another one has correct ID.""" """Test we remove config entry if another one has correct ID."""
correct_entry = MockConfigEntry( correct_entry = MockConfigEntry(
domain=hue.DOMAIN, domain=hue.DOMAIN,

35
tests/components/hue/test_light_v1.py
View File

@ -8,6 +8,7 @@ from homeassistant.components import hue
from homeassistant.components.hue.const import CONF_ALLOW_HUE_GROUPS from homeassistant.components.hue.const import CONF_ALLOW_HUE_GROUPS
from homeassistant.components.hue.v1 import light as hue_light from homeassistant.components.hue.v1 import light as hue_light
from homeassistant.components.light import ColorMode from homeassistant.components.light import ColorMode
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.util import color from homeassistant.util import color
@ -185,7 +186,9 @@ async def setup_bridge(hass, mock_bridge_v1):
await hass.async_block_till_done() await hass.async_block_till_done()
async def test_not_load_groups_if_old_bridge(hass, mock_bridge_v1): async def test_not_load_groups_if_old_bridge(
hass: HomeAssistant, mock_bridge_v1
) -> None:
"""Test that we don't try to load groups if bridge runs old software.""" """Test that we don't try to load groups if bridge runs old software."""
mock_bridge_v1.api.config.apiversion = "1.12.0" mock_bridge_v1.api.config.apiversion = "1.12.0"
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
@ -195,7 +198,7 @@ async def test_not_load_groups_if_old_bridge(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_no_lights_or_groups(hass, mock_bridge_v1): async def test_no_lights_or_groups(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test the update_lights function when no lights are found.""" """Test the update_lights function when no lights are found."""
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
mock_bridge_v1.mock_group_responses.append({}) mock_bridge_v1.mock_group_responses.append({})
@ -204,7 +207,7 @@ async def test_no_lights_or_groups(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_lights(hass, mock_bridge_v1): async def test_lights(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test the update_lights function with some lights.""" """Test the update_lights function with some lights."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -224,7 +227,7 @@ async def test_lights(hass, mock_bridge_v1):
assert lamp_2.state == "off" assert lamp_2.state == "off"
async def test_lights_color_mode(hass, mock_bridge_v1): async def test_lights_color_mode(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test that lights only report appropriate color mode.""" """Test that lights only report appropriate color mode."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
mock_bridge_v1.mock_group_responses.append(GROUP_RESPONSE) mock_bridge_v1.mock_group_responses.append(GROUP_RESPONSE)
@ -269,7 +272,7 @@ async def test_lights_color_mode(hass, mock_bridge_v1):
] ]
async def test_groups(hass, mock_bridge_v1): async def test_groups(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test the update_lights function with some lights.""" """Test the update_lights function with some lights."""
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
mock_bridge_v1.mock_group_responses.append(GROUP_RESPONSE) mock_bridge_v1.mock_group_responses.append(GROUP_RESPONSE)
@ -294,7 +297,7 @@ async def test_groups(hass, mock_bridge_v1):
assert ent_reg.async_get("light.group_2").unique_id == "2" assert ent_reg.async_get("light.group_2").unique_id == "2"
async def test_new_group_discovered(hass, mock_bridge_v1): async def test_new_group_discovered(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has a new group.""" """Test if 2nd update has a new group."""
mock_bridge_v1.allow_groups = True mock_bridge_v1.allow_groups = True
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
@ -341,7 +344,7 @@ async def test_new_group_discovered(hass, mock_bridge_v1):
assert new_group.attributes["color_temp"] == 250 assert new_group.attributes["color_temp"] == 250
async def test_new_light_discovered(hass, mock_bridge_v1): async def test_new_light_discovered(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has a new light.""" """Test if 2nd update has a new light."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -387,7 +390,7 @@ async def test_new_light_discovered(hass, mock_bridge_v1):
assert light.state == "off" assert light.state == "off"
async def test_group_removed(hass, mock_bridge_v1): async def test_group_removed(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has removed group.""" """Test if 2nd update has removed group."""
mock_bridge_v1.allow_groups = True mock_bridge_v1.allow_groups = True
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
@ -416,7 +419,7 @@ async def test_group_removed(hass, mock_bridge_v1):
assert removed_group is None assert removed_group is None
async def test_light_removed(hass, mock_bridge_v1): async def test_light_removed(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has removed light.""" """Test if 2nd update has removed light."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -443,7 +446,7 @@ async def test_light_removed(hass, mock_bridge_v1):
assert removed_light is None assert removed_light is None
async def test_other_group_update(hass, mock_bridge_v1): async def test_other_group_update(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test changing one group that will impact the state of other light.""" """Test changing one group that will impact the state of other light."""
mock_bridge_v1.allow_groups = True mock_bridge_v1.allow_groups = True
mock_bridge_v1.mock_light_responses.append({}) mock_bridge_v1.mock_light_responses.append({})
@ -496,7 +499,7 @@ async def test_other_group_update(hass, mock_bridge_v1):
assert group_2.state == "off" assert group_2.state == "off"
async def test_other_light_update(hass, mock_bridge_v1): async def test_other_light_update(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test changing one light that will impact state of other light.""" """Test changing one light that will impact state of other light."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -549,7 +552,7 @@ async def test_other_light_update(hass, mock_bridge_v1):
assert lamp_2.attributes["brightness"] == 100 assert lamp_2.attributes["brightness"] == 100
async def test_update_timeout(hass, mock_bridge_v1): async def test_update_timeout(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test bridge marked as not available if timeout error during update.""" """Test bridge marked as not available if timeout error during update."""
mock_bridge_v1.api.lights.update = Mock(side_effect=asyncio.TimeoutError) mock_bridge_v1.api.lights.update = Mock(side_effect=asyncio.TimeoutError)
mock_bridge_v1.api.groups.update = Mock(side_effect=asyncio.TimeoutError) mock_bridge_v1.api.groups.update = Mock(side_effect=asyncio.TimeoutError)
@ -558,7 +561,7 @@ async def test_update_timeout(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_update_unauthorized(hass, mock_bridge_v1): async def test_update_unauthorized(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test bridge marked as not authorized if unauthorized during update.""" """Test bridge marked as not authorized if unauthorized during update."""
mock_bridge_v1.api.lights.update = Mock(side_effect=aiohue.Unauthorized) mock_bridge_v1.api.lights.update = Mock(side_effect=aiohue.Unauthorized)
await setup_bridge(hass, mock_bridge_v1) await setup_bridge(hass, mock_bridge_v1)
@ -567,7 +570,7 @@ async def test_update_unauthorized(hass, mock_bridge_v1):
assert len(mock_bridge_v1.handle_unauthorized_error.mock_calls) == 1 assert len(mock_bridge_v1.handle_unauthorized_error.mock_calls) == 1
async def test_light_turn_on_service(hass, mock_bridge_v1): async def test_light_turn_on_service(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test calling the turn on service on a light.""" """Test calling the turn on service on a light."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -620,7 +623,7 @@ async def test_light_turn_on_service(hass, mock_bridge_v1):
} }
async def test_light_turn_off_service(hass, mock_bridge_v1): async def test_light_turn_off_service(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test calling the turn on service on a light.""" """Test calling the turn on service on a light."""
mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE) mock_bridge_v1.mock_light_responses.append(LIGHT_RESPONSE)
@ -758,7 +761,7 @@ def test_hs_color() -> None:
assert light.hs_color == color.color_xy_to_hs(0.4, 0.5, LIGHT_GAMUT) assert light.hs_color == color.color_xy_to_hs(0.4, 0.5, LIGHT_GAMUT)
async def test_group_features(hass, mock_bridge_v1): async def test_group_features(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test group features.""" """Test group features."""
color_temp_type = "Color temperature light" color_temp_type = "Color temperature light"
extended_color_type = "Extended color light" extended_color_type = "Extended color light"

24
tests/components/hue/test_light_v2.py
View File

@ -1,13 +1,15 @@
"""Philips Hue lights platform tests for V2 bridge/api.""" """Philips Hue lights platform tests for V2 bridge/api."""
from homeassistant.components.light import ColorMode from homeassistant.components.light import ColorMode
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from .conftest import setup_platform from .conftest import setup_platform
from .const import FAKE_DEVICE, FAKE_LIGHT, FAKE_ZIGBEE_CONNECTIVITY from .const import FAKE_DEVICE, FAKE_LIGHT, FAKE_ZIGBEE_CONNECTIVITY
async def test_lights(hass, mock_bridge_v2, v2_resources_test_data): async def test_lights(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if all v2 lights get created with correct features.""" """Test if all v2 lights get created with correct features."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -73,7 +75,9 @@ async def test_lights(hass, mock_bridge_v2, v2_resources_test_data):
assert light_4.attributes["supported_color_modes"] == [] assert light_4.attributes["supported_color_modes"] == []
async def test_light_turn_on_service(hass, mock_bridge_v2, v2_resources_test_data): async def test_light_turn_on_service(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test calling the turn on service on a light.""" """Test calling the turn on service on a light."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -201,7 +205,9 @@ async def test_light_turn_on_service(hass, mock_bridge_v2, v2_resources_test_dat
assert mock_bridge_v2.mock_requests[8]["json"]["timed_effects"]["duration"] == 6000 assert mock_bridge_v2.mock_requests[8]["json"]["timed_effects"]["duration"] == 6000
async def test_light_turn_off_service(hass, mock_bridge_v2, v2_resources_test_data): async def test_light_turn_off_service(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test calling the turn off service on a light.""" """Test calling the turn off service on a light."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -271,7 +277,7 @@ async def test_light_turn_off_service(hass, mock_bridge_v2, v2_resources_test_da
assert mock_bridge_v2.mock_requests[3]["json"]["identify"]["action"] == "identify" assert mock_bridge_v2.mock_requests[3]["json"]["identify"]["action"] == "identify"
async def test_light_added(hass, mock_bridge_v2): async def test_light_added(hass: HomeAssistant, mock_bridge_v2) -> None:
"""Test new light added to bridge.""" """Test new light added to bridge."""
await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY]) await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY])
@ -293,7 +299,9 @@ async def test_light_added(hass, mock_bridge_v2):
assert test_entity.attributes["friendly_name"] == FAKE_DEVICE["metadata"]["name"] assert test_entity.attributes["friendly_name"] == FAKE_DEVICE["metadata"]["name"]
async def test_light_availability(hass, mock_bridge_v2, v2_resources_test_data): async def test_light_availability(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test light availability property.""" """Test light availability property."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -323,7 +331,9 @@ async def test_light_availability(hass, mock_bridge_v2, v2_resources_test_data):
assert test_light.state == "on" if status == "connected" else "unavailable" assert test_light.state == "on" if status == "connected" else "unavailable"
async def test_grouped_lights(hass, mock_bridge_v2, v2_resources_test_data): async def test_grouped_lights(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if all v2 grouped lights get created with correct features.""" """Test if all v2 grouped lights get created with correct features."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)

8
tests/components/hue/test_logbook.py
View File

@ -1,5 +1,4 @@
"""The tests for hue logbook.""" """The tests for hue logbook."""
from homeassistant.components.hue.const import ATTR_HUE_EVENT, CONF_SUBTYPE, DOMAIN from homeassistant.components.hue.const import ATTR_HUE_EVENT, CONF_SUBTYPE, DOMAIN
from homeassistant.components.hue.v1.hue_event import CONF_LAST_UPDATED from homeassistant.components.hue.v1.hue_event import CONF_LAST_UPDATED
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -10,6 +9,7 @@ from homeassistant.const import (
CONF_TYPE, CONF_TYPE,
CONF_UNIQUE_ID, CONF_UNIQUE_ID,
) )
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry from homeassistant.helpers import device_registry
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -35,7 +35,7 @@ SAMPLE_V2_EVENT = {
} }
async def test_humanify_hue_events(hass, mock_bridge_v2): async def test_humanify_hue_events(hass: HomeAssistant, mock_bridge_v2) -> None:
"""Test hue events when the devices are present in the registry.""" """Test hue events when the devices are present in the registry."""
await setup_platform(hass, mock_bridge_v2, "sensor") await setup_platform(hass, mock_bridge_v2, "sensor")
hass.config.components.add("recorder") hass.config.components.add("recorder")
@ -74,7 +74,9 @@ async def test_humanify_hue_events(hass, mock_bridge_v2):
assert v2_event["message"] == "first button pressed initially" assert v2_event["message"] == "first button pressed initially"
async def test_humanify_hue_events_devices_removed(hass, mock_bridge_v2): async def test_humanify_hue_events_devices_removed(
hass: HomeAssistant, mock_bridge_v2
) -> None:
"""Test hue events when the devices have been removed from the registry.""" """Test hue events when the devices have been removed from the registry."""
await setup_platform(hass, mock_bridge_v2, "sensor") await setup_platform(hass, mock_bridge_v2, "sensor")
hass.config.components.add("recorder") hass.config.components.add("recorder")

16
tests/components/hue/test_migration.py
View File

@ -44,8 +44,8 @@ async def test_auto_switchover(hass: HomeAssistant) -> None:
async def test_light_entity_migration( async def test_light_entity_migration(
hass, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data hass: HomeAssistant, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data
): ) -> None:
"""Test if entity schema for lights migrates from v1 to v2.""" """Test if entity schema for lights migrates from v1 to v2."""
config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2 config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2
@ -88,8 +88,8 @@ async def test_light_entity_migration(
async def test_sensor_entity_migration( async def test_sensor_entity_migration(
hass, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data hass: HomeAssistant, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data
): ) -> None:
"""Test if entity schema for sensors migrates from v1 to v2.""" """Test if entity schema for sensors migrates from v1 to v2."""
config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2 config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2
@ -147,8 +147,8 @@ async def test_sensor_entity_migration(
async def test_group_entity_migration_with_v1_id( async def test_group_entity_migration_with_v1_id(
hass, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data hass: HomeAssistant, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data
): ) -> None:
"""Test if entity schema for grouped_lights migrates from v1 to v2.""" """Test if entity schema for grouped_lights migrates from v1 to v2."""
config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2 config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2
@ -180,8 +180,8 @@ async def test_group_entity_migration_with_v1_id(
async def test_group_entity_migration_with_v2_group_id( async def test_group_entity_migration_with_v2_group_id(
hass, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data hass: HomeAssistant, mock_bridge_v2, mock_config_entry_v2, v2_resources_test_data
): ) -> None:
"""Test if entity schema for grouped_lights migrates from v1 to v2.""" """Test if entity schema for grouped_lights migrates from v1 to v2."""
config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2 config_entry = mock_bridge_v2.config_entry = mock_config_entry_v2

19
tests/components/hue/test_scene.py
View File

@ -1,14 +1,15 @@
"""Philips Hue scene platform tests for V2 bridge/api.""" """Philips Hue scene platform tests for V2 bridge/api."""
from homeassistant.const import STATE_UNKNOWN from homeassistant.const import STATE_UNKNOWN
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from .conftest import setup_platform from .conftest import setup_platform
from .const import FAKE_SCENE from .const import FAKE_SCENE
async def test_scene(hass, mock_bridge_v2, v2_resources_test_data): async def test_scene(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if (config) scenes get created.""" """Test if (config) scenes get created."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -67,7 +68,9 @@ async def test_scene(hass, mock_bridge_v2, v2_resources_test_data):
assert entity_entry.device_id is not None assert entity_entry.device_id is not None
async def test_scene_turn_on_service(hass, mock_bridge_v2, v2_resources_test_data): async def test_scene_turn_on_service(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test calling the turn on service on a scene.""" """Test calling the turn on service on a scene."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -103,8 +106,8 @@ async def test_scene_turn_on_service(hass, mock_bridge_v2, v2_resources_test_dat
async def test_scene_advanced_turn_on_service( async def test_scene_advanced_turn_on_service(
hass, mock_bridge_v2, v2_resources_test_data hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
): ) -> None:
"""Test calling the advanced turn on service on a scene.""" """Test calling the advanced turn on service on a scene."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -139,7 +142,9 @@ async def test_scene_advanced_turn_on_service(
} }
async def test_scene_updates(hass, mock_bridge_v2, v2_resources_test_data): async def test_scene_updates(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test scene events from bridge.""" """Test scene events from bridge."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)

21
tests/components/hue/test_sensor_v1.py
View File

@ -8,6 +8,7 @@ from homeassistant.components import hue
from homeassistant.components.hue.const import ATTR_HUE_EVENT from homeassistant.components.hue.const import ATTR_HUE_EVENT
from homeassistant.components.hue.v1 import sensor_base from homeassistant.components.hue.v1 import sensor_base
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_registry import async_get from homeassistant.helpers.entity_registry import async_get
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -281,7 +282,7 @@ SENSOR_RESPONSE = {
} }
async def test_no_sensors(hass, mock_bridge_v1): async def test_no_sensors(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test the update_items function when no sensors are found.""" """Test the update_items function when no sensors are found."""
mock_bridge_v1.mock_sensor_responses.append({}) mock_bridge_v1.mock_sensor_responses.append({})
await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"])
@ -289,7 +290,9 @@ async def test_no_sensors(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_sensors_with_multiple_bridges(hass, mock_bridge_v1): async def test_sensors_with_multiple_bridges(
hass: HomeAssistant, mock_bridge_v1
) -> None:
"""Test the update_items function with some sensors.""" """Test the update_items function with some sensors."""
mock_bridge_2 = create_mock_bridge(hass, api_version=1) mock_bridge_2 = create_mock_bridge(hass, api_version=1)
mock_bridge_2.mock_sensor_responses.append( mock_bridge_2.mock_sensor_responses.append(
@ -311,7 +314,7 @@ async def test_sensors_with_multiple_bridges(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 10 assert len(hass.states.async_all()) == 10
async def test_sensors(hass, mock_bridge_v1): async def test_sensors(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test the update_items function with some sensors.""" """Test the update_items function with some sensors."""
mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE)
await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"])
@ -355,7 +358,7 @@ async def test_sensors(hass, mock_bridge_v1):
) )
async def test_unsupported_sensors(hass, mock_bridge_v1): async def test_unsupported_sensors(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test that unsupported sensors don't get added and don't fail.""" """Test that unsupported sensors don't get added and don't fail."""
response_with_unsupported = dict(SENSOR_RESPONSE) response_with_unsupported = dict(SENSOR_RESPONSE)
response_with_unsupported["7"] = UNSUPPORTED_SENSOR response_with_unsupported["7"] = UNSUPPORTED_SENSOR
@ -366,7 +369,7 @@ async def test_unsupported_sensors(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 7 assert len(hass.states.async_all()) == 7
async def test_new_sensor_discovered(hass, mock_bridge_v1): async def test_new_sensor_discovered(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has a new sensor.""" """Test if 2nd update has a new sensor."""
mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE)
@ -400,7 +403,7 @@ async def test_new_sensor_discovered(hass, mock_bridge_v1):
assert temperature.state == "17.75" assert temperature.state == "17.75"
async def test_sensor_removed(hass, mock_bridge_v1): async def test_sensor_removed(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test if 2nd update has removed sensor.""" """Test if 2nd update has removed sensor."""
mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE)
@ -428,7 +431,7 @@ async def test_sensor_removed(hass, mock_bridge_v1):
assert removed_sensor is None assert removed_sensor is None
async def test_update_timeout(hass, mock_bridge_v1): async def test_update_timeout(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test bridge marked as not available if timeout error during update.""" """Test bridge marked as not available if timeout error during update."""
mock_bridge_v1.api.sensors.update = Mock(side_effect=asyncio.TimeoutError) mock_bridge_v1.api.sensors.update = Mock(side_effect=asyncio.TimeoutError)
await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"])
@ -436,7 +439,7 @@ async def test_update_timeout(hass, mock_bridge_v1):
assert len(hass.states.async_all()) == 0 assert len(hass.states.async_all()) == 0
async def test_update_unauthorized(hass, mock_bridge_v1): async def test_update_unauthorized(hass: HomeAssistant, mock_bridge_v1) -> None:
"""Test bridge marked as not authorized if unauthorized during update.""" """Test bridge marked as not authorized if unauthorized during update."""
mock_bridge_v1.api.sensors.update = Mock(side_effect=aiohue.Unauthorized) mock_bridge_v1.api.sensors.update = Mock(side_effect=aiohue.Unauthorized)
await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"]) await setup_platform(hass, mock_bridge_v1, ["binary_sensor", "sensor"])
@ -445,7 +448,7 @@ async def test_update_unauthorized(hass, mock_bridge_v1):
assert len(mock_bridge_v1.handle_unauthorized_error.mock_calls) == 1 assert len(mock_bridge_v1.handle_unauthorized_error.mock_calls) == 1
async def test_hue_events(hass, mock_bridge_v1, device_reg): async def test_hue_events(hass: HomeAssistant, mock_bridge_v1, device_reg) -> None:
"""Test that hue remotes fire events when pressed.""" """Test that hue remotes fire events when pressed."""
mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE) mock_bridge_v1.mock_sensor_responses.append(SENSOR_RESPONSE)

12
tests/components/hue/test_sensor_v2.py
View File

@ -1,6 +1,6 @@
"""Philips Hue sensor platform tests for V2 bridge/api.""" """Philips Hue sensor platform tests for V2 bridge/api."""
from homeassistant.components import hue from homeassistant.components import hue
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -8,7 +8,9 @@ from .conftest import setup_bridge, setup_platform
from .const import FAKE_DEVICE, FAKE_SENSOR, FAKE_ZIGBEE_CONNECTIVITY from .const import FAKE_DEVICE, FAKE_SENSOR, FAKE_ZIGBEE_CONNECTIVITY
async def test_sensors(hass, mock_bridge_v2, v2_resources_test_data): async def test_sensors(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if all v2 sensors get created with correct features.""" """Test if all v2 sensors get created with correct features."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -60,8 +62,8 @@ async def test_sensors(hass, mock_bridge_v2, v2_resources_test_data):
async def test_enable_sensor( async def test_enable_sensor(
hass, mock_bridge_v2, v2_resources_test_data, mock_config_entry_v2 hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data, mock_config_entry_v2
): ) -> None:
"""Test enabling of the by default disabled zigbee_connectivity sensor.""" """Test enabling of the by default disabled zigbee_connectivity sensor."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
await setup_bridge(hass, mock_bridge_v2, mock_config_entry_v2) await setup_bridge(hass, mock_bridge_v2, mock_config_entry_v2)
@ -95,7 +97,7 @@ async def test_enable_sensor(
assert state.attributes["mac_address"] == "00:17:88:01:0b:aa:bb:99" assert state.attributes["mac_address"] == "00:17:88:01:0b:aa:bb:99"
async def test_sensor_add_update(hass, mock_bridge_v2): async def test_sensor_add_update(hass: HomeAssistant, mock_bridge_v2) -> None:
"""Test if sensors get added/updated from events.""" """Test if sensors get added/updated from events."""
await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY]) await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY])
await setup_platform(hass, mock_bridge_v2, "sensor") await setup_platform(hass, mock_bridge_v2, "sensor")

37
tests/components/hue/test_services.py
View File

@ -8,6 +8,7 @@ from homeassistant.components.hue.const import (
CONF_ALLOW_HUE_GROUPS, CONF_ALLOW_HUE_GROUPS,
CONF_ALLOW_UNREACHABLE, CONF_ALLOW_UNREACHABLE,
) )
from homeassistant.core import HomeAssistant
from .conftest import setup_bridge, setup_component from .conftest import setup_bridge, setup_component
@ -45,7 +46,7 @@ SCENE_RESPONSE = {
} }
async def test_hue_activate_scene(hass, mock_api_v1): async def test_hue_activate_scene(hass: HomeAssistant, mock_api_v1) -> None:
"""Test successful hue_activate_scene.""" """Test successful hue_activate_scene."""
config_entry = config_entries.ConfigEntry( config_entry = config_entries.ConfigEntry(
1, 1,
@ -81,7 +82,7 @@ async def test_hue_activate_scene(hass, mock_api_v1):
assert mock_api_v1.mock_requests[2]["path"] == "groups/group_1/action" assert mock_api_v1.mock_requests[2]["path"] == "groups/group_1/action"
async def test_hue_activate_scene_transition(hass, mock_api_v1): async def test_hue_activate_scene_transition(hass: HomeAssistant, mock_api_v1) -> None:
"""Test successful hue_activate_scene with transition.""" """Test successful hue_activate_scene with transition."""
config_entry = config_entries.ConfigEntry( config_entry = config_entries.ConfigEntry(
1, 1,
@ -117,7 +118,9 @@ async def test_hue_activate_scene_transition(hass, mock_api_v1):
assert mock_api_v1.mock_requests[2]["path"] == "groups/group_1/action" assert mock_api_v1.mock_requests[2]["path"] == "groups/group_1/action"
async def test_hue_activate_scene_group_not_found(hass, mock_api_v1): async def test_hue_activate_scene_group_not_found(
hass: HomeAssistant, mock_api_v1
) -> None:
"""Test failed hue_activate_scene due to missing group.""" """Test failed hue_activate_scene due to missing group."""
config_entry = config_entries.ConfigEntry( config_entry = config_entries.ConfigEntry(
1, 1,
@ -148,7 +151,9 @@ async def test_hue_activate_scene_group_not_found(hass, mock_api_v1):
) )
async def test_hue_activate_scene_scene_not_found(hass, mock_api_v1): async def test_hue_activate_scene_scene_not_found(
hass: HomeAssistant, mock_api_v1
) -> None:
"""Test failed hue_activate_scene due to missing scene.""" """Test failed hue_activate_scene due to missing scene."""
config_entry = config_entries.ConfigEntry( config_entry = config_entries.ConfigEntry(
1, 1,
@ -180,8 +185,12 @@ async def test_hue_activate_scene_scene_not_found(hass, mock_api_v1):
async def test_hue_multi_bridge_activate_scene_all_respond( async def test_hue_multi_bridge_activate_scene_all_respond(
hass, mock_bridge_v1, mock_bridge_v2, mock_config_entry_v1, mock_config_entry_v2 hass: HomeAssistant,
): mock_bridge_v1,
mock_bridge_v2,
mock_config_entry_v1,
mock_config_entry_v2,
) -> None:
"""Test that makes multiple bridges successfully activate a scene.""" """Test that makes multiple bridges successfully activate a scene."""
await setup_component(hass) await setup_component(hass)
@ -210,8 +219,12 @@ async def test_hue_multi_bridge_activate_scene_all_respond(
async def test_hue_multi_bridge_activate_scene_one_responds( async def test_hue_multi_bridge_activate_scene_one_responds(
hass, mock_bridge_v1, mock_bridge_v2, mock_config_entry_v1, mock_config_entry_v2 hass: HomeAssistant,
): mock_bridge_v1,
mock_bridge_v2,
mock_config_entry_v1,
mock_config_entry_v2,
) -> None:
"""Test that makes only one bridge successfully activate a scene.""" """Test that makes only one bridge successfully activate a scene."""
await setup_component(hass) await setup_component(hass)
@ -239,8 +252,12 @@ async def test_hue_multi_bridge_activate_scene_one_responds(
async def test_hue_multi_bridge_activate_scene_zero_responds( async def test_hue_multi_bridge_activate_scene_zero_responds(
hass, mock_bridge_v1, mock_bridge_v2, mock_config_entry_v1, mock_config_entry_v2 hass: HomeAssistant,
): mock_bridge_v1,
mock_bridge_v2,
mock_config_entry_v1,
mock_config_entry_v2,
) -> None:
"""Test that makes no bridge successfully activate a scene.""" """Test that makes no bridge successfully activate a scene."""
await setup_component(hass) await setup_component(hass)
mock_api_v1 = mock_bridge_v1.api mock_api_v1 = mock_bridge_v1.api

15
tests/components/hue/test_switch.py
View File

@ -1,10 +1,13 @@
"""Philips Hue switch platform tests for V2 bridge/api.""" """Philips Hue switch platform tests for V2 bridge/api."""
from homeassistant.core import HomeAssistant
from .conftest import setup_platform from .conftest import setup_platform
from .const import FAKE_BINARY_SENSOR, FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY from .const import FAKE_BINARY_SENSOR, FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY
async def test_switch(hass, mock_bridge_v2, v2_resources_test_data): async def test_switch(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test if (config) switches get created.""" """Test if (config) switches get created."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -22,7 +25,9 @@ async def test_switch(hass, mock_bridge_v2, v2_resources_test_data):
assert test_entity.attributes["device_class"] == "switch" assert test_entity.attributes["device_class"] == "switch"
async def test_switch_turn_on_service(hass, mock_bridge_v2, v2_resources_test_data): async def test_switch_turn_on_service(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test calling the turn on service on a switch.""" """Test calling the turn on service on a switch."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -44,7 +49,9 @@ async def test_switch_turn_on_service(hass, mock_bridge_v2, v2_resources_test_da
assert mock_bridge_v2.mock_requests[0]["json"]["enabled"] is True assert mock_bridge_v2.mock_requests[0]["json"]["enabled"] is True
async def test_switch_turn_off_service(hass, mock_bridge_v2, v2_resources_test_data): async def test_switch_turn_off_service(
hass: HomeAssistant, mock_bridge_v2, v2_resources_test_data
) -> None:
"""Test calling the turn off service on a switch.""" """Test calling the turn off service on a switch."""
await mock_bridge_v2.api.load_test_data(v2_resources_test_data) await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
@ -83,7 +90,7 @@ async def test_switch_turn_off_service(hass, mock_bridge_v2, v2_resources_test_d
assert test_entity.state == "off" assert test_entity.state == "off"
async def test_switch_added(hass, mock_bridge_v2): async def test_switch_added(hass: HomeAssistant, mock_bridge_v2) -> None:
"""Test new switch added to bridge.""" """Test new switch added to bridge."""
await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY]) await mock_bridge_v2.api.load_test_data([FAKE_DEVICE, FAKE_ZIGBEE_CONNECTIVITY])