Add type hints to integration tests (part 23) (#88235)

This commit is contained in:
epenet 2023-02-17 16:40:46 +01:00 committed by GitHub
parent aa50096a31
commit 9a5f88f55f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
49 changed files with 651 additions and 408 deletions

View file

@ -288,7 +288,9 @@ async def test_config_children_only(hass: HomeAssistant) -> None:
assert config_start == config
async def test_config_children_and_attr(hass, config_children_and_attr):
async def test_config_children_and_attr(
hass: HomeAssistant, config_children_and_attr
) -> None:
"""Check config with children and attributes."""
config_start = copy(config_children_and_attr)
del config_start["platform"]
@ -379,7 +381,9 @@ async def test_master_state(hass: HomeAssistant) -> None:
assert ump.master_state is None
async def test_master_state_with_attrs(hass, config_children_and_attr, mock_states):
async def test_master_state_with_attrs(
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test master state property."""
config = validate_config(config_children_and_attr)
@ -390,7 +394,9 @@ async def test_master_state_with_attrs(hass, config_children_and_attr, mock_stat
assert ump.master_state == STATE_ON
async def test_master_state_with_bad_attrs(hass, config_children_and_attr):
async def test_master_state_with_bad_attrs(
hass: HomeAssistant, config_children_and_attr
) -> None:
"""Test master state property."""
config = copy(config_children_and_attr)
config["attributes"]["state"] = "bad.entity_id"
@ -401,7 +407,7 @@ async def test_master_state_with_bad_attrs(hass, config_children_and_attr):
assert ump.master_state == STATE_OFF
async def test_active_child_state(hass, mock_states):
async def test_active_child_state(hass: HomeAssistant, mock_states) -> None:
"""Test active child state property."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -460,7 +466,7 @@ async def test_polling(hass: HomeAssistant) -> None:
assert ump.should_poll is False
async def test_state_children_only(hass, mock_states):
async def test_state_children_only(hass: HomeAssistant, mock_states) -> None:
"""Test media player state with only children."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -478,8 +484,8 @@ async def test_state_children_only(hass, mock_states):
async def test_state_with_children_and_attrs(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test media player with children and master state."""
config = validate_config(config_children_and_attr)
@ -504,7 +510,7 @@ async def test_state_with_children_and_attrs(
assert ump.state == STATE_OFF
async def test_volume_level(hass, mock_states):
async def test_volume_level(hass: HomeAssistant, mock_states) -> None:
"""Test volume level property."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -527,7 +533,7 @@ async def test_volume_level(hass, mock_states):
assert ump.volume_level == 1
async def test_media_image_url(hass, mock_states):
async def test_media_image_url(hass: HomeAssistant, mock_states) -> None:
"""Test media_image_url property."""
test_url = "test_url"
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -548,7 +554,7 @@ async def test_media_image_url(hass, mock_states):
assert mock_states.mock_mp_1.entity_picture == ump.entity_picture
async def test_is_volume_muted_children_only(hass, mock_states):
async def test_is_volume_muted_children_only(hass: HomeAssistant, mock_states) -> None:
"""Test is volume muted property w/ children only."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -572,8 +578,8 @@ async def test_is_volume_muted_children_only(hass, mock_states):
async def test_sound_mode_list_children_and_attr(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test sound mode list property w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -588,8 +594,8 @@ async def test_sound_mode_list_children_and_attr(
async def test_source_list_children_and_attr(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test source list property w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -602,8 +608,8 @@ async def test_source_list_children_and_attr(
async def test_sound_mode_children_and_attr(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test sound modeproperty w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -615,7 +621,9 @@ async def test_sound_mode_children_and_attr(
assert ump.sound_mode == "movie"
async def test_source_children_and_attr(hass, config_children_and_attr, mock_states):
async def test_source_children_and_attr(
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test source property w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -628,8 +636,8 @@ async def test_source_children_and_attr(hass, config_children_and_attr, mock_sta
async def test_volume_level_children_and_attr(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test volume level property w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -642,8 +650,8 @@ async def test_volume_level_children_and_attr(
async def test_is_volume_muted_children_and_attr(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test is volume muted property w/ children and attrs."""
config = validate_config(config_children_and_attr)
@ -655,7 +663,9 @@ async def test_is_volume_muted_children_and_attr(
assert ump.is_volume_muted
async def test_supported_features_children_only(hass, mock_states):
async def test_supported_features_children_only(
hass: HomeAssistant, mock_states
) -> None:
"""Test supported media commands with only children."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -674,8 +684,8 @@ async def test_supported_features_children_only(hass, mock_states):
async def test_supported_features_children_and_cmds(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test supported media commands with children and attrs."""
config = copy(config_children_and_attr)
excmd = {"service": "media_player.test", "data": {}}
@ -732,7 +742,7 @@ async def test_supported_features_children_and_cmds(
assert check_flags == ump.supported_features
async def test_overrides(hass, config_children_and_attr):
async def test_overrides(hass: HomeAssistant, config_children_and_attr) -> None:
"""Test overrides."""
config = copy(config_children_and_attr)
excmd = {"service": "test.override", "data": {}}
@ -908,8 +918,8 @@ async def test_overrides(hass, config_children_and_attr):
async def test_supported_features_play_pause(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test supported media commands with play_pause function."""
config = copy(config_children_and_attr)
excmd = {"service": "media_player.test", "data": {"entity_id": "test"}}
@ -931,8 +941,8 @@ async def test_supported_features_play_pause(
async def test_service_call_no_active_child(
hass, config_children_and_attr, mock_states
):
hass: HomeAssistant, config_children_and_attr, mock_states
) -> None:
"""Test a service call to children with no active child."""
config = validate_config(config_children_and_attr)
@ -952,7 +962,7 @@ async def test_service_call_no_active_child(
assert len(mock_states.mock_mp_2.service_calls["turn_off"]) == 0
async def test_service_call_to_child(hass, mock_states):
async def test_service_call_to_child(hass: HomeAssistant, mock_states) -> None:
"""Test service calls that should be routed to a child."""
config = validate_config(CONFIG_CHILDREN_ONLY)
@ -1027,7 +1037,7 @@ async def test_service_call_to_child(hass, mock_states):
assert len(mock_states.mock_mp_2.service_calls["turn_off"]) == 2
async def test_service_call_to_command(hass, mock_states):
async def test_service_call_to_command(hass: HomeAssistant, mock_states) -> None:
"""Test service call to command."""
config = copy(CONFIG_CHILDREN_ONLY)
config["commands"] = {"turn_off": {"service": "test.turn_off", "data": {}}}