evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve typing for calls fixture in tests (a-l) (#118349)

Browse source 
* Improve typing for `calls` fixture in tests (a-l)

* More

* More
This commit is contained in:
epenet 2024-05-29 09:06:48 +02:00 • committed by GitHub
parent 89ae425ac2
commit 7e62061b9a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
43 changed files with 546 additions and 338 deletions
Download patch file Download diff file Expand all files Collapse all files

99
tests/components/automation/test_init.py
View file

@ -72,13 +72,13 @@ from tests.typing import WebSocketGenerator
@pytest.fixture
def calls(hass):
def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service."""
return async_mock_service(hass, "test", "automation")
async def test_service_data_not_a_dict(
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls: list[ServiceCall]
) -> None:
"""Test service data not dict."""
with assert_setup_component(1, automation.DOMAIN):
@ -99,7 +99,9 @@ async def test_service_data_not_a_dict(
assert "Result is not a Dictionary" in caplog.text
async def test_service_data_single_template(hass: HomeAssistant, calls) -> None:
async def test_service_data_single_template(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test service data not dict."""
with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component(
@ -122,7 +124,9 @@ async def test_service_data_single_template(hass: HomeAssistant, calls) -> None:
assert calls[0].data["foo"] == "bar"
async def test_service_specify_data(hass: HomeAssistant, calls) -> None:
async def test_service_specify_data(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test service data."""
assert await async_setup_component(
hass,
@ -156,7 +160,9 @@ async def test_service_specify_data(hass: HomeAssistant, calls) -> None:
assert state.attributes.get("last_triggered") == time
async def test_service_specify_entity_id(hass: HomeAssistant, calls) -> None:
async def test_service_specify_entity_id(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test service data."""
assert await async_setup_component(
hass,
@ -175,7 +181,9 @@ async def test_service_specify_entity_id(hass: HomeAssistant, calls) -> None:
assert ["hello.world"] == calls[0].data.get(ATTR_ENTITY_ID)
async def test_service_specify_entity_id_list(hass: HomeAssistant, calls) -> None:
async def test_service_specify_entity_id_list(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test service data."""
assert await async_setup_component(
hass,
@ -197,7 +205,7 @@ async def test_service_specify_entity_id_list(hass: HomeAssistant, calls) -> Non
assert ["hello.world", "hello.world2"] == calls[0].data.get(ATTR_ENTITY_ID)
async def test_two_triggers(hass: HomeAssistant, calls) -> None:
async def test_two_triggers(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test triggers."""
assert await async_setup_component(
hass,
@ -222,7 +230,7 @@ async def test_two_triggers(hass: HomeAssistant, calls) -> None:
async def test_trigger_service_ignoring_condition(
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls: list[ServiceCall]
) -> None:
"""Test triggers."""
assert await async_setup_component(
@ -274,7 +282,9 @@ async def test_trigger_service_ignoring_condition(
assert len(calls) == 2
async def test_two_conditions_with_and(hass: HomeAssistant, calls) -> None:
async def test_two_conditions_with_and(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test two and conditions."""
entity_id = "test.entity"
assert await async_setup_component(
@ -312,7 +322,9 @@ async def test_two_conditions_with_and(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1
async def test_shorthand_conditions_template(hass: HomeAssistant, calls) -> None:
async def test_shorthand_conditions_template(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test shorthand nation form in conditions."""
assert await async_setup_component(
hass,
@ -337,7 +349,9 @@ async def test_shorthand_conditions_template(hass: HomeAssistant, calls) -> None
assert len(calls) == 1
async def test_automation_list_setting(hass: HomeAssistant, calls) -> None:
async def test_automation_list_setting(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Event is not a valid condition."""
assert await async_setup_component(
hass,
@ -365,7 +379,9 @@ async def test_automation_list_setting(hass: HomeAssistant, calls) -> None:
assert len(calls) == 2
async def test_automation_calling_two_actions(hass: HomeAssistant, calls) -> None:
async def test_automation_calling_two_actions(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test if we can call two actions from automation async definition."""
assert await async_setup_component(
hass,
@ -389,7 +405,7 @@ async def test_automation_calling_two_actions(hass: HomeAssistant, calls) -> Non
assert calls[1].data["position"] == 1
async def test_shared_context(hass: HomeAssistant, calls) -> None:
async def test_shared_context(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test that the shared context is passed down the chain."""
assert await async_setup_component(
hass,
@ -456,7 +472,7 @@ async def test_shared_context(hass: HomeAssistant, calls) -> None:
assert calls[0].context is second_trigger_context
async def test_services(hass: HomeAssistant, calls) -> None:
async def test_services(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test the automation services for turning entities on/off."""
entity_id = "automation.hello"
@ -539,7 +555,10 @@ async def test_services(hass: HomeAssistant, calls) -> None:
async def test_reload_config_service(
hass: HomeAssistant, calls, hass_admin_user: MockUser, hass_read_only_user: MockUser
hass: HomeAssistant,
calls: list[ServiceCall],
hass_admin_user: MockUser,
hass_read_only_user: MockUser,
) -> None:
"""Test the reload config service."""
assert await async_setup_component(
@ -618,7 +637,9 @@ async def test_reload_config_service(
assert calls[1].data.get("event") == "test_event2"
async def test_reload_config_when_invalid_config(hass: HomeAssistant, calls) -> None:
async def test_reload_config_when_invalid_config(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the reload config service handling invalid config."""
with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component(
@ -657,7 +678,9 @@ async def test_reload_config_when_invalid_config(hass: HomeAssistant, calls) ->
assert len(calls) == 1
async def test_reload_config_handles_load_fails(hass: HomeAssistant, calls) -> None:
async def test_reload_config_handles_load_fails(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the reload config service."""
assert await async_setup_component(
hass,
@ -697,7 +720,9 @@ async def test_reload_config_handles_load_fails(hass: HomeAssistant, calls) -> N
@pytest.mark.parametrize(
"service", ["turn_off_stop", "turn_off_no_stop", "reload", "reload_single"]
)
async def test_automation_stops(hass: HomeAssistant, calls, service) -> None:
async def test_automation_stops(
hass: HomeAssistant, calls: list[ServiceCall], service: str
) -> None:
"""Test that turning off / reloading stops any running actions as appropriate."""
entity_id = "automation.hello"
test_entity = "test.entity"
@ -774,7 +799,7 @@ async def test_automation_stops(hass: HomeAssistant, calls, service) -> None:
@pytest.mark.parametrize("extra_config", [{}, {"id": "sun"}])
async def test_reload_unchanged_does_not_stop(
hass: HomeAssistant, calls, extra_config
hass: HomeAssistant, calls: list[ServiceCall], extra_config: dict[str, str]
) -> None:
"""Test that reloading stops any running actions as appropriate."""
test_entity = "test.entity"
@ -820,7 +845,7 @@ async def test_reload_unchanged_does_not_stop(
async def test_reload_single_unchanged_does_not_stop(
hass: HomeAssistant, calls
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test that reloading stops any running actions as appropriate."""
test_entity = "test.entity"
@ -870,7 +895,9 @@ async def test_reload_single_unchanged_does_not_stop(
assert len(calls) == 1
async def test_reload_single_add_automation(hass: HomeAssistant, calls) -> None:
async def test_reload_single_add_automation(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test that reloading a single automation."""
config1 = {automation.DOMAIN: {}}
config2 = {
@ -904,7 +931,9 @@ async def test_reload_single_add_automation(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1
async def test_reload_single_parallel_calls(hass: HomeAssistant, calls) -> None:
async def test_reload_single_parallel_calls(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test reloading single automations in parallel."""
config1 = {automation.DOMAIN: {}}
config2 = {
@ -1017,7 +1046,9 @@ async def test_reload_single_parallel_calls(hass: HomeAssistant, calls) -> None:
assert len(calls) == 4
async def test_reload_single_remove_automation(hass: HomeAssistant, calls) -> None:
async def test_reload_single_remove_automation(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test that reloading a single automation."""
config1 = {
automation.DOMAIN: {
@ -1052,7 +1083,7 @@ async def test_reload_single_remove_automation(hass: HomeAssistant, calls) -> No
async def test_reload_moved_automation_without_alias(
hass: HomeAssistant, calls
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test that changing the order of automations without alias triggers reload."""
with patch(
@ -1107,7 +1138,7 @@ async def test_reload_moved_automation_without_alias(
async def test_reload_identical_automations_without_id(
hass: HomeAssistant, calls
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test reloading of identical automations without id."""
with patch(
@ -1282,7 +1313,7 @@ async def test_reload_identical_automations_without_id(
],
)
async def test_reload_unchanged_automation(
hass: HomeAssistant, calls, automation_config
hass: HomeAssistant, calls: list[ServiceCall], automation_config: dict[str, Any]
) -> None:
"""Test an unmodified automation is not reloaded."""
with patch(
@ -1317,7 +1348,7 @@ async def test_reload_unchanged_automation(
@pytest.mark.parametrize("extra_config", [{}, {"id": "sun"}])
async def test_reload_automation_when_blueprint_changes(
hass: HomeAssistant, calls, extra_config
hass: HomeAssistant, calls: list[ServiceCall], extra_config: dict[str, str]
) -> None:
"""Test an automation is updated at reload if the blueprint has changed."""
with patch(
@ -2409,7 +2440,9 @@ async def test_automation_this_var_always(
assert "Error rendering variables" not in caplog.text
async def test_blueprint_automation(hass: HomeAssistant, calls) -> None:
async def test_blueprint_automation(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test blueprint automation."""
assert await async_setup_component(
hass,
@ -2527,7 +2560,7 @@ async def test_blueprint_automation_fails_substitution(
) in caplog.text
async def test_trigger_service(hass: HomeAssistant, calls) -> None:
async def test_trigger_service(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test the automation trigger service."""
assert await async_setup_component(
hass,
@ -2557,7 +2590,9 @@ async def test_trigger_service(hass: HomeAssistant, calls) -> None:
assert calls[0].context.parent_id is context.id
async def test_trigger_condition_implicit_id(hass: HomeAssistant, calls) -> None:
async def test_trigger_condition_implicit_id(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test triggers."""
assert await async_setup_component(
hass,
@ -2607,7 +2642,9 @@ async def test_trigger_condition_implicit_id(hass: HomeAssistant, calls) -> None
assert calls[-1].data.get("param") == "one"
async def test_trigger_condition_explicit_id(hass: HomeAssistant, calls) -> None:
async def test_trigger_condition_explicit_id(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test triggers."""
assert await async_setup_component(
hass,