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

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/components/arcam_fmj/test_device_trigger.py
View File

@ -5,7 +5,7 @@ import pytest
from homeassistant.components import automation from homeassistant.components import automation
from homeassistant.components.arcam_fmj.const import DOMAIN from homeassistant.components.arcam_fmj.const import DOMAIN
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -22,7 +22,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -67,7 +67,11 @@ async def test_get_triggers(
async def test_if_fires_on_turn_on_request( async def test_if_fires_on_turn_on_request(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls, player_setup, state hass: HomeAssistant,
entity_registry: er.EntityRegistry,
calls: list[ServiceCall],
player_setup,
state,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
entry = entity_registry.async_get(player_setup) entry = entity_registry.async_get(player_setup)
@ -113,7 +117,11 @@ async def test_if_fires_on_turn_on_request(
async def test_if_fires_on_turn_on_request_legacy( async def test_if_fires_on_turn_on_request_legacy(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls, player_setup, state hass: HomeAssistant,
entity_registry: er.EntityRegistry,
calls: list[ServiceCall],
player_setup,
state,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
entry = entity_registry.async_get(player_setup) entry = entity_registry.async_get(player_setup)

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

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

6
tests/components/automation/test_recorder.py
View File

@ -15,7 +15,7 @@ from homeassistant.components.automation import (
from homeassistant.components.recorder import Recorder from homeassistant.components.recorder import Recorder
from homeassistant.components.recorder.history import get_significant_states from homeassistant.components.recorder.history import get_significant_states
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.core import HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
@ -24,13 +24,13 @@ from tests.components.recorder.common import async_wait_recording_done
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
async def test_exclude_attributes( async def test_exclude_attributes(
recorder_mock: Recorder, hass: HomeAssistant, calls recorder_mock: Recorder, hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test automation registered attributes to be excluded.""" """Test automation registered attributes to be excluded."""
now = dt_util.utcnow() now = dt_util.utcnow()

10
tests/components/binary_sensor/test_device_condition.py
View File

@ -11,7 +11,7 @@ from homeassistant.components.binary_sensor import DOMAIN, BinarySensorDeviceCla
from homeassistant.components.binary_sensor.device_condition import ENTITY_CONDITIONS from homeassistant.components.binary_sensor.device_condition import ENTITY_CONDITIONS
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -33,7 +33,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -239,7 +239,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
@ -327,7 +327,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
@ -387,7 +387,7 @@ async def test_if_fires_on_for_condition(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for firing if condition is on with delay.""" """Test for firing if condition is on with delay."""

10
tests/components/binary_sensor/test_device_trigger.py
View File

@ -10,7 +10,7 @@ from homeassistant.components.binary_sensor import DOMAIN, BinarySensorDeviceCla
from homeassistant.components.binary_sensor.device_trigger import ENTITY_TRIGGERS from homeassistant.components.binary_sensor.device_trigger import ENTITY_TRIGGERS
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -33,7 +33,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -240,7 +240,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for on and off triggers firing.""" """Test for on and off triggers firing."""
@ -335,7 +335,7 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""
@ -407,7 +407,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_binary_sensor_entities: dict[str, MockBinarySensor], mock_binary_sensor_entities: dict[str, MockBinarySensor],
) -> None: ) -> None:
"""Test for triggers firing.""" """Test for triggers firing."""

8
tests/components/bthome/test_device_trigger.py
View File

@ -7,7 +7,7 @@ from homeassistant.components.bluetooth.const import DOMAIN as BLUETOOTH_DOMAIN
from homeassistant.components.bthome.const import CONF_SUBTYPE, DOMAIN from homeassistant.components.bthome.const import CONF_SUBTYPE, DOMAIN
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM, CONF_TYPE from homeassistant.const import CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM, CONF_TYPE
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.helpers.device_registry import ( from homeassistant.helpers.device_registry import (
CONNECTION_NETWORK_MAC, CONNECTION_NETWORK_MAC,
async_get as async_get_dev_reg, async_get as async_get_dev_reg,
@ -32,7 +32,7 @@ def get_device_id(mac: str) -> tuple[str, str]:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -229,7 +229,9 @@ async def test_get_triggers_for_invalid_device_id(hass: HomeAssistant) -> None:
await hass.async_block_till_done() await hass.async_block_till_done()
async def test_if_fires_on_motion_detected(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_motion_detected(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for motion event trigger firing.""" """Test for motion event trigger firing."""
mac = "DE:70:E8:B2:39:0C" mac = "DE:70:E8:B2:39:0C"
entry = await _async_setup_bthome_device(hass, mac) entry = await _async_setup_bthome_device(hass, mac)

4
tests/components/button/test_device_trigger.py
View File

@ -109,7 +109,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -169,7 +169,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

8
tests/components/climate/test_device_condition.py
View File

@ -8,7 +8,7 @@ from homeassistant.components import automation
from homeassistant.components.climate import DOMAIN, HVACMode, const, device_condition from homeassistant.components.climate import DOMAIN, HVACMode, const, device_condition
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -30,7 +30,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -151,7 +151,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -272,7 +272,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

8
tests/components/climate/test_device_trigger.py
View File

@ -14,7 +14,7 @@ from homeassistant.components.climate import (
) )
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import EntityCategory, UnitOfTemperature from homeassistant.const import EntityCategory, UnitOfTemperature
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -36,7 +36,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -151,7 +151,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -272,7 +272,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

27
tests/components/conversation/test_trigger.py
View File

@ -7,7 +7,7 @@ import voluptuous as vol
from homeassistant.components.conversation import default_agent from homeassistant.components.conversation import default_agent
from homeassistant.components.conversation.models import ConversationInput from homeassistant.components.conversation.models import ConversationInput
from homeassistant.core import Context, HomeAssistant from homeassistant.core import Context, HomeAssistant, ServiceCall
from homeassistant.helpers import trigger from homeassistant.helpers import trigger
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -16,19 +16,21 @@ from tests.typing import WebSocketGenerator
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def setup_comp(hass): async def setup_comp(hass: HomeAssistant) -> None:
"""Initialize components.""" """Initialize components."""
assert await async_setup_component(hass, "homeassistant", {}) assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, "conversation", {}) assert await async_setup_component(hass, "conversation", {})
async def test_if_fires_on_event(hass: HomeAssistant, calls, setup_comp) -> None: async def test_if_fires_on_event(
hass: HomeAssistant, calls: list[ServiceCall], setup_comp: None
) -> None:
"""Test the firing of events.""" """Test the firing of events."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -134,7 +136,9 @@ async def test_empty_response(hass: HomeAssistant, setup_comp) -> None:
assert service_response["response"]["speech"]["plain"]["speech"] == "" assert service_response["response"]["speech"]["plain"]["speech"] == ""
async def test_response_same_sentence(hass: HomeAssistant, calls, setup_comp) -> None: async def test_response_same_sentence(
hass: HomeAssistant, calls: list[ServiceCall], setup_comp: None
) -> None:
"""Test the conversation response action with multiple triggers using the same sentence.""" """Test the conversation response action with multiple triggers using the same sentence."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -196,7 +200,10 @@ async def test_response_same_sentence(hass: HomeAssistant, calls, setup_comp) ->
async def test_response_same_sentence_with_error( async def test_response_same_sentence_with_error(
hass: HomeAssistant, calls, setup_comp, caplog: pytest.LogCaptureFixture hass: HomeAssistant,
calls: list[ServiceCall],
setup_comp: None,
caplog: pytest.LogCaptureFixture,
) -> None: ) -> None:
"""Test the conversation response action with multiple triggers using the same sentence and an error.""" """Test the conversation response action with multiple triggers using the same sentence and an error."""
caplog.set_level(logging.ERROR) caplog.set_level(logging.ERROR)
@ -303,7 +310,7 @@ async def test_subscribe_trigger_does_not_interfere_with_responses(
async def test_same_trigger_multiple_sentences( async def test_same_trigger_multiple_sentences(
hass: HomeAssistant, calls, setup_comp hass: HomeAssistant, calls: list[ServiceCall], setup_comp: None
) -> None: ) -> None:
"""Test matching of multiple sentences from the same trigger.""" """Test matching of multiple sentences from the same trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -348,7 +355,7 @@ async def test_same_trigger_multiple_sentences(
async def test_same_sentence_multiple_triggers( async def test_same_sentence_multiple_triggers(
hass: HomeAssistant, calls, setup_comp hass: HomeAssistant, calls: list[ServiceCall], setup_comp: None
) -> None: ) -> None:
"""Test use of the same sentence in multiple triggers.""" """Test use of the same sentence in multiple triggers."""
assert await async_setup_component( assert await async_setup_component(
@ -467,7 +474,9 @@ async def test_fails_on_no_sentences(hass: HomeAssistant) -> None:
) )
async def test_wildcards(hass: HomeAssistant, calls, setup_comp) -> None: async def test_wildcards(
hass: HomeAssistant, calls: list[ServiceCall], setup_comp: None
) -> None:
"""Test wildcards in trigger sentences.""" """Test wildcards in trigger sentences."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,

12
tests/components/cover/test_device_condition.py
View File

@ -15,7 +15,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
EntityCategory, EntityCategory,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -36,7 +36,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -358,7 +358,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -501,7 +501,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -557,7 +557,7 @@ async def test_if_position(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
caplog: pytest.LogCaptureFixture, caplog: pytest.LogCaptureFixture,
mock_cover_entities: list[MockCover], mock_cover_entities: list[MockCover],
) -> None: ) -> None:
@ -717,7 +717,7 @@ async def test_if_tilt_position(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
caplog: pytest.LogCaptureFixture, caplog: pytest.LogCaptureFixture,
mock_cover_entities: list[MockCover], mock_cover_entities: list[MockCover],
) -> None: ) -> None:

14
tests/components/cover/test_device_trigger.py
View File

@ -16,7 +16,7 @@ from homeassistant.const import (
STATE_OPENING, STATE_OPENING,
EntityCategory, EntityCategory,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -39,7 +39,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -380,7 +380,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for state triggers firing.""" """Test for state triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -533,7 +533,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for state triggers firing.""" """Test for state triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -593,7 +593,7 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -659,7 +659,7 @@ async def test_if_fires_on_position(
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
mock_cover_entities: list[MockCover], mock_cover_entities: list[MockCover],
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for position triggers.""" """Test for position triggers."""
setup_test_component_platform(hass, DOMAIN, mock_cover_entities) setup_test_component_platform(hass, DOMAIN, mock_cover_entities)
@ -811,7 +811,7 @@ async def test_if_fires_on_tilt_position(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_cover_entities: list[MockCover], mock_cover_entities: list[MockCover],
) -> None: ) -> None:
"""Test for tilt position triggers.""" """Test for tilt position triggers."""

6
tests/components/device_automation/test_init.py
View File

@ -16,7 +16,7 @@ from homeassistant.components.device_automation import (
from homeassistant.components.websocket_api.const import TYPE_RESULT from homeassistant.components.websocket_api.const import TYPE_RESULT
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import STATE_OFF, STATE_ON from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
from homeassistant.loader import IntegrationNotFound from homeassistant.loader import IntegrationNotFound
@ -1385,14 +1385,14 @@ async def test_automation_with_bad_condition(
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
async def test_automation_with_sub_condition( async def test_automation_with_sub_condition(
hass: HomeAssistant, hass: HomeAssistant,
calls, calls: list[ServiceCall],
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
) -> None: ) -> None:

10
tests/components/device_automation/test_toggle_entity.py
View File

@ -6,7 +6,7 @@ import pytest
from homeassistant.components import automation from homeassistant.components import automation
from homeassistant.const import STATE_OFF, STATE_ON from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -20,7 +20,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -29,7 +29,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing. """Test for turn_on and turn_off triggers firing.
@ -145,8 +145,8 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
trigger, trigger: str,
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

8
tests/components/device_tracker/test_device_condition.py
View File

@ -7,7 +7,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.device_tracker import DOMAIN from homeassistant.components.device_tracker import DOMAIN
from homeassistant.const import STATE_HOME, EntityCategory from homeassistant.const import STATE_HOME, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -25,7 +25,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -114,7 +114,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -199,7 +199,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

8
tests/components/device_tracker/test_device_trigger.py
View File

@ -8,7 +8,7 @@ from homeassistant.components import automation, zone
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.device_tracker import DOMAIN, device_trigger from homeassistant.components.device_tracker import DOMAIN, device_trigger
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -37,7 +37,7 @@ HOME_LONGITUDE = -117.237561
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -145,7 +145,7 @@ async def test_if_fires_on_zone_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for enter and leave triggers firing.""" """Test for enter and leave triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -252,7 +252,7 @@ async def test_if_fires_on_zone_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for enter and leave triggers firing.""" """Test for enter and leave triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

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

@ -9,7 +9,7 @@ import pytest
from homeassistant import config_entries from homeassistant import config_entries
from homeassistant.components import dialogflow, intent_script from homeassistant.components import dialogflow, intent_script
from homeassistant.config import async_process_ha_core_config from homeassistant.config import async_process_ha_core_config
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, ServiceCall, callback
from homeassistant.data_entry_flow import FlowResultType from homeassistant.data_entry_flow import FlowResultType
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -22,12 +22,12 @@ CONTEXT_NAME = "78a5db95-b7d6-4d50-9c9b-2fc73a5e34c3_id_dialog_context"
@pytest.fixture @pytest.fixture
async def calls(hass, fixture): async def calls(hass: HomeAssistant, fixture) -> list[ServiceCall]:
"""Return a list of Dialogflow calls triggered.""" """Return a list of Dialogflow calls triggered."""
calls = [] calls: list[ServiceCall] = []
@callback @callback
def mock_service(call): def mock_service(call: ServiceCall) -> None:
"""Mock action call.""" """Mock action call."""
calls.append(call) calls.append(call)
@ -343,7 +343,9 @@ async def test_intent_request_without_slots_v2(hass: HomeAssistant, fixture) ->
assert text == "You are both home, you silly" assert text == "You are both home, you silly"
async def test_intent_request_calling_service_v1(fixture, calls) -> None: async def test_intent_request_calling_service_v1(
fixture, calls: list[ServiceCall]
) -> None:
"""Test a request for calling a service. """Test a request for calling a service.
If this request is done async the test could finish before the action If this request is done async the test could finish before the action
@ -365,7 +367,9 @@ async def test_intent_request_calling_service_v1(fixture, calls) -> None:
assert call.data.get("hello") == "virgo" assert call.data.get("hello") == "virgo"
async def test_intent_request_calling_service_v2(fixture, calls) -> None: async def test_intent_request_calling_service_v2(
fixture, calls: list[ServiceCall]
) -> None:
"""Test a request for calling a service. """Test a request for calling a service.
If this request is done async the test could finish before the action If this request is done async the test could finish before the action

8
tests/components/fan/test_device_condition.py
View File

@ -7,7 +7,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.fan import DOMAIN from homeassistant.components.fan import DOMAIN
from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -25,7 +25,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -114,7 +114,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -199,7 +199,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

10
tests/components/fan/test_device_trigger.py
View File

@ -9,7 +9,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.fan import DOMAIN from homeassistant.components.fan import DOMAIN
from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -30,7 +30,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -180,7 +180,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -293,7 +293,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -353,7 +353,7 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

38
tests/components/geo_location/test_trigger.py
View File

@ -11,7 +11,7 @@ from homeassistant.const import (
SERVICE_TURN_OFF, SERVICE_TURN_OFF,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
) )
from homeassistant.core import Context, HomeAssistant from homeassistant.core import Context, HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import async_mock_service, mock_component from tests.common import async_mock_service, mock_component
@ -23,7 +23,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -48,7 +48,9 @@ def setup_comp(hass):
) )
async def test_if_fires_on_zone_enter(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_enter(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on zone enter.""" """Test for firing on zone enter."""
context = Context() context = Context()
hass.states.async_set( hass.states.async_set(
@ -126,7 +128,9 @@ async def test_if_fires_on_zone_enter(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_not_fires_for_enter_on_zone_leave(hass: HomeAssistant, calls) -> None: async def test_if_not_fires_for_enter_on_zone_leave(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for not firing on zone leave.""" """Test for not firing on zone leave."""
hass.states.async_set( hass.states.async_set(
"geo_location.entity", "geo_location.entity",
@ -161,7 +165,9 @@ async def test_if_not_fires_for_enter_on_zone_leave(hass: HomeAssistant, calls)
assert len(calls) == 0 assert len(calls) == 0
async def test_if_fires_on_zone_leave(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_leave(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on zone leave.""" """Test for firing on zone leave."""
hass.states.async_set( hass.states.async_set(
"geo_location.entity", "geo_location.entity",
@ -196,7 +202,9 @@ async def test_if_fires_on_zone_leave(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_zone_leave_2(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_leave_2(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on zone leave for unavailable entity.""" """Test for firing on zone leave for unavailable entity."""
hass.states.async_set( hass.states.async_set(
"geo_location.entity", "geo_location.entity",
@ -231,7 +239,9 @@ async def test_if_fires_on_zone_leave_2(hass: HomeAssistant, calls) -> None:
assert len(calls) == 0 assert len(calls) == 0
async def test_if_not_fires_for_leave_on_zone_enter(hass: HomeAssistant, calls) -> None: async def test_if_not_fires_for_leave_on_zone_enter(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for not firing on zone enter.""" """Test for not firing on zone enter."""
hass.states.async_set( hass.states.async_set(
"geo_location.entity", "geo_location.entity",
@ -266,7 +276,9 @@ async def test_if_not_fires_for_leave_on_zone_enter(hass: HomeAssistant, calls)
assert len(calls) == 0 assert len(calls) == 0
async def test_if_fires_on_zone_appear(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_appear(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing if entity appears in zone.""" """Test for firing if entity appears in zone."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -312,7 +324,9 @@ async def test_if_fires_on_zone_appear(hass: HomeAssistant, calls) -> None:
) )
async def test_if_fires_on_zone_appear_2(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_appear_2(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing if entity appears in zone.""" """Test for firing if entity appears in zone."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -367,7 +381,9 @@ async def test_if_fires_on_zone_appear_2(hass: HomeAssistant, calls) -> None:
) )
async def test_if_fires_on_zone_disappear(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_zone_disappear(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing if entity disappears from zone.""" """Test for firing if entity disappears from zone."""
hass.states.async_set( hass.states.async_set(
"geo_location.entity", "geo_location.entity",
@ -414,7 +430,7 @@ async def test_if_fires_on_zone_disappear(hass: HomeAssistant, calls) -> None:
async def test_zone_undefined( async def test_zone_undefined(
hass: HomeAssistant, calls, caplog: pytest.LogCaptureFixture hass: HomeAssistant, calls: list[ServiceCall], caplog: pytest.LogCaptureFixture
) -> None: ) -> None:
"""Test for undefined zone.""" """Test for undefined zone."""
hass.states.async_set( hass.states.async_set(

2
tests/components/google_translate/test_tts.py
View File

@ -39,7 +39,7 @@ def mock_tts_cache_dir_autouse(mock_tts_cache_dir):
@pytest.fixture @pytest.fixture
async def calls(hass: HomeAssistant) -> list[ServiceCall]: def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Mock media player calls.""" """Mock media player calls."""
return async_mock_service(hass, DOMAIN_MP, SERVICE_PLAY_MEDIA) return async_mock_service(hass, DOMAIN_MP, SERVICE_PLAY_MEDIA)

51
tests/components/homeassistant/triggers/test_event.py
View File

@ -4,14 +4,14 @@ import pytest
from homeassistant.components import automation from homeassistant.components import automation
from homeassistant.const import ATTR_ENTITY_ID, ENTITY_MATCH_ALL, SERVICE_TURN_OFF from homeassistant.const import ATTR_ENTITY_ID, ENTITY_MATCH_ALL, SERVICE_TURN_OFF
from homeassistant.core import Context, HomeAssistant from homeassistant.core import Context, HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import async_mock_service, mock_component from tests.common import async_mock_service, mock_component
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -28,7 +28,7 @@ def setup_comp(hass):
mock_component(hass, "group") mock_component(hass, "group")
async def test_if_fires_on_event(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_event(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test the firing of events.""" """Test the firing of events."""
context = Context() context = Context()
@ -64,7 +64,9 @@ async def test_if_fires_on_event(hass: HomeAssistant, calls) -> None:
assert calls[0].data["id"] == 0 assert calls[0].data["id"] == 0
async def test_if_fires_on_templated_event(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_templated_event(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the firing of events.""" """Test the firing of events."""
context = Context() context = Context()
@ -97,7 +99,9 @@ async def test_if_fires_on_templated_event(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_multiple_events(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_multiple_events(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the firing of events.""" """Test the firing of events."""
context = Context() context = Context()
@ -125,7 +129,7 @@ async def test_if_fires_on_multiple_events(hass: HomeAssistant, calls) -> None:
async def test_if_fires_on_event_extra_data( async def test_if_fires_on_event_extra_data(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test the firing of events still matches with event data and context.""" """Test the firing of events still matches with event data and context."""
assert await async_setup_component( assert await async_setup_component(
@ -157,7 +161,7 @@ async def test_if_fires_on_event_extra_data(
async def test_if_fires_on_event_with_data_and_context( async def test_if_fires_on_event_with_data_and_context(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test the firing of events with data and context.""" """Test the firing of events with data and context."""
assert await async_setup_component( assert await async_setup_component(
@ -204,7 +208,7 @@ async def test_if_fires_on_event_with_data_and_context(
async def test_if_fires_on_event_with_templated_data_and_context( async def test_if_fires_on_event_with_templated_data_and_context(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test the firing of events with templated data and context.""" """Test the firing of events with templated data and context."""
assert await async_setup_component( assert await async_setup_component(
@ -256,7 +260,7 @@ async def test_if_fires_on_event_with_templated_data_and_context(
async def test_if_fires_on_event_with_empty_data_and_context_config( async def test_if_fires_on_event_with_empty_data_and_context_config(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test the firing of events with empty data and context config. """Test the firing of events with empty data and context config.
@ -288,7 +292,9 @@ async def test_if_fires_on_event_with_empty_data_and_context_config(
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_event_with_nested_data(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_event_with_nested_data(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the firing of events with nested data. """Test the firing of events with nested data.
This test exercises the slow path of using vol.Schema to validate This test exercises the slow path of using vol.Schema to validate
@ -316,7 +322,9 @@ async def test_if_fires_on_event_with_nested_data(hass: HomeAssistant, calls) ->
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_event_with_empty_data(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_event_with_empty_data(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the firing of events with empty data. """Test the firing of events with empty data.
This test exercises the fast path to validate matching event data. This test exercises the fast path to validate matching event data.
@ -340,7 +348,9 @@ async def test_if_fires_on_event_with_empty_data(hass: HomeAssistant, calls) ->
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_sample_zha_event(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_sample_zha_event(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the firing of events with a sample zha event. """Test the firing of events with a sample zha event.
This test exercises the fast path to validate matching event data. This test exercises the fast path to validate matching event data.
@ -398,7 +408,7 @@ async def test_if_fires_on_sample_zha_event(hass: HomeAssistant, calls) -> None:
async def test_if_not_fires_if_event_data_not_matches( async def test_if_not_fires_if_event_data_not_matches(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test firing of event if no data match.""" """Test firing of event if no data match."""
assert await async_setup_component( assert await async_setup_component(
@ -422,7 +432,7 @@ async def test_if_not_fires_if_event_data_not_matches(
async def test_if_not_fires_if_event_context_not_matches( async def test_if_not_fires_if_event_context_not_matches(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test firing of event if no context match.""" """Test firing of event if no context match."""
assert await async_setup_component( assert await async_setup_component(
@ -446,7 +456,7 @@ async def test_if_not_fires_if_event_context_not_matches(
async def test_if_fires_on_multiple_user_ids( async def test_if_fires_on_multiple_user_ids(
hass: HomeAssistant, calls, context_with_user hass: HomeAssistant, calls: list[ServiceCall], context_with_user: Context
) -> None: ) -> None:
"""Test the firing of event when the trigger has multiple user ids. """Test the firing of event when the trigger has multiple user ids.
@ -474,7 +484,9 @@ async def test_if_fires_on_multiple_user_ids(
assert len(calls) == 1 assert len(calls) == 1
async def test_event_data_with_list(hass: HomeAssistant, calls) -> None: async def test_event_data_with_list(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test the (non)firing of event when the data schema has lists.""" """Test the (non)firing of event when the data schema has lists."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -511,7 +523,10 @@ async def test_event_data_with_list(hass: HomeAssistant, calls) -> None:
"event_type", ["state_reported", ["test_event", "state_reported"]] "event_type", ["state_reported", ["test_event", "state_reported"]]
) )
async def test_state_reported_event( async def test_state_reported_event(
hass: HomeAssistant, calls, caplog, event_type: list[str] hass: HomeAssistant,
calls: list[ServiceCall],
caplog: pytest.LogCaptureFixture,
event_type: str | list[str],
) -> None: ) -> None:
"""Test triggering on state reported event.""" """Test triggering on state reported event."""
context = Context() context = Context()
@ -541,7 +556,7 @@ async def test_state_reported_event(
async def test_templated_state_reported_event( async def test_templated_state_reported_event(
hass: HomeAssistant, calls, caplog hass: HomeAssistant, calls: list[ServiceCall], caplog: pytest.LogCaptureFixture
) -> None: ) -> None:
"""Test triggering on state reported event.""" """Test triggering on state reported event."""
context = Context() context = Context()

145
tests/components/homeassistant/triggers/test_numeric_state.py
View File

@ -18,7 +18,7 @@ from homeassistant.const import (
SERVICE_TURN_OFF, SERVICE_TURN_OFF,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
) )
from homeassistant.core import Context, HomeAssistant from homeassistant.core import Context, HomeAssistant, ServiceCall
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
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -32,7 +32,7 @@ from tests.common import (
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -63,7 +63,7 @@ async def setup_comp(hass):
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_not_fires_on_entity_removal( async def test_if_not_fires_on_entity_removal(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with removed entity.""" """Test the firing with removed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -93,7 +93,7 @@ async def test_if_not_fires_on_entity_removal(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_fires_on_entity_change_below( async def test_if_fires_on_entity_change_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -142,7 +142,10 @@ async def test_if_fires_on_entity_change_below(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_fires_on_entity_change_below_uuid( async def test_if_fires_on_entity_change_below_uuid(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls, below hass: HomeAssistant,
entity_registry: er.EntityRegistry,
calls: list[ServiceCall],
below: int | str,
) -> None: ) -> None:
"""Test the firing with changed entity specified by registry entry id.""" """Test the firing with changed entity specified by registry entry id."""
entry = entity_registry.async_get_or_create( entry = entity_registry.async_get_or_create(
@ -196,7 +199,7 @@ async def test_if_fires_on_entity_change_below_uuid(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_fires_on_entity_change_over_to_below( async def test_if_fires_on_entity_change_over_to_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -227,7 +230,7 @@ async def test_if_fires_on_entity_change_over_to_below(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_fires_on_entities_change_over_to_below( async def test_if_fires_on_entities_change_over_to_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with changed entities.""" """Test the firing with changed entities."""
hass.states.async_set("test.entity_1", 11) hass.states.async_set("test.entity_1", 11)
@ -262,7 +265,7 @@ async def test_if_fires_on_entities_change_over_to_below(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_not_fires_on_entity_change_below_to_below( async def test_if_not_fires_on_entity_change_below_to_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
context = Context() context = Context()
@ -305,7 +308,7 @@ async def test_if_not_fires_on_entity_change_below_to_below(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_not_below_fires_on_entity_change_to_equal( async def test_if_not_below_fires_on_entity_change_to_equal(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -336,7 +339,7 @@ async def test_if_not_below_fires_on_entity_change_to_equal(
"below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "below", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_not_fires_on_initial_entity_below( async def test_if_not_fires_on_initial_entity_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test the firing when starting with a match.""" """Test the firing when starting with a match."""
hass.states.async_set("test.entity", 9) hass.states.async_set("test.entity", 9)
@ -367,7 +370,7 @@ async def test_if_not_fires_on_initial_entity_below(
"above", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "above", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_not_fires_on_initial_entity_above( async def test_if_not_fires_on_initial_entity_above(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test the firing when starting with a match.""" """Test the firing when starting with a match."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -398,7 +401,7 @@ async def test_if_not_fires_on_initial_entity_above(
"above", [10, "input_number.value_10", "number.value_10", "sensor.value_10"] "above", [10, "input_number.value_10", "number.value_10", "sensor.value_10"]
) )
async def test_if_fires_on_entity_change_above( async def test_if_fires_on_entity_change_above(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 9) hass.states.async_set("test.entity", 9)
@ -425,7 +428,7 @@ async def test_if_fires_on_entity_change_above(
async def test_if_fires_on_entity_unavailable_at_startup( async def test_if_fires_on_entity_unavailable_at_startup(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test the firing with changed entity at startup.""" """Test the firing with changed entity at startup."""
assert await async_setup_component( assert await async_setup_component(
@ -450,7 +453,7 @@ async def test_if_fires_on_entity_unavailable_at_startup(
@pytest.mark.parametrize("above", [10, "input_number.value_10"]) @pytest.mark.parametrize("above", [10, "input_number.value_10"])
async def test_if_fires_on_entity_change_below_to_above( async def test_if_fires_on_entity_change_below_to_above(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
# set initial state # set initial state
@ -480,7 +483,7 @@ async def test_if_fires_on_entity_change_below_to_above(
@pytest.mark.parametrize("above", [10, "input_number.value_10"]) @pytest.mark.parametrize("above", [10, "input_number.value_10"])
async def test_if_not_fires_on_entity_change_above_to_above( async def test_if_not_fires_on_entity_change_above_to_above(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
# set initial state # set initial state
@ -515,7 +518,7 @@ async def test_if_not_fires_on_entity_change_above_to_above(
@pytest.mark.parametrize("above", [10, "input_number.value_10"]) @pytest.mark.parametrize("above", [10, "input_number.value_10"])
async def test_if_not_above_fires_on_entity_change_to_equal( async def test_if_not_above_fires_on_entity_change_to_equal(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
# set initial state # set initial state
@ -553,7 +556,7 @@ async def test_if_not_above_fires_on_entity_change_to_equal(
], ],
) )
async def test_if_fires_on_entity_change_below_range( async def test_if_fires_on_entity_change_below_range(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -590,7 +593,7 @@ async def test_if_fires_on_entity_change_below_range(
], ],
) )
async def test_if_fires_on_entity_change_below_above_range( async def test_if_fires_on_entity_change_below_above_range(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
assert await async_setup_component( assert await async_setup_component(
@ -624,7 +627,7 @@ async def test_if_fires_on_entity_change_below_above_range(
], ],
) )
async def test_if_fires_on_entity_change_over_to_below_range( async def test_if_fires_on_entity_change_over_to_below_range(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -662,7 +665,7 @@ async def test_if_fires_on_entity_change_over_to_below_range(
], ],
) )
async def test_if_fires_on_entity_change_over_to_below_above_range( async def test_if_fires_on_entity_change_over_to_below_above_range(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test the firing with changed entity.""" """Test the firing with changed entity."""
hass.states.async_set("test.entity", 11) hass.states.async_set("test.entity", 11)
@ -692,7 +695,7 @@ async def test_if_fires_on_entity_change_over_to_below_above_range(
@pytest.mark.parametrize("below", [100, "input_number.value_100"]) @pytest.mark.parametrize("below", [100, "input_number.value_100"])
async def test_if_not_fires_if_entity_not_match( async def test_if_not_fires_if_entity_not_match(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test if not fired with non matching entity.""" """Test if not fired with non matching entity."""
assert await async_setup_component( assert await async_setup_component(
@ -716,7 +719,7 @@ async def test_if_not_fires_if_entity_not_match(
async def test_if_not_fires_and_warns_if_below_entity_unknown( async def test_if_not_fires_and_warns_if_below_entity_unknown(
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls hass: HomeAssistant, caplog: pytest.LogCaptureFixture, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test if warns with unknown below entity.""" """Test if warns with unknown below entity."""
assert await async_setup_component( assert await async_setup_component(
@ -747,7 +750,7 @@ async def test_if_not_fires_and_warns_if_below_entity_unknown(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_fires_on_entity_change_below_with_attribute( async def test_if_fires_on_entity_change_below_with_attribute(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
hass.states.async_set("test.entity", 11, {"test_attribute": 11}) hass.states.async_set("test.entity", 11, {"test_attribute": 11})
@ -775,7 +778,7 @@ async def test_if_fires_on_entity_change_below_with_attribute(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_not_fires_on_entity_change_not_below_with_attribute( async def test_if_not_fires_on_entity_change_not_below_with_attribute(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes.""" """Test attributes."""
assert await async_setup_component( assert await async_setup_component(
@ -800,7 +803,7 @@ async def test_if_not_fires_on_entity_change_not_below_with_attribute(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_fires_on_attribute_change_with_attribute_below( async def test_if_fires_on_attribute_change_with_attribute_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
hass.states.async_set("test.entity", "entity", {"test_attribute": 11}) hass.states.async_set("test.entity", "entity", {"test_attribute": 11})
@ -829,7 +832,7 @@ async def test_if_fires_on_attribute_change_with_attribute_below(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_not_fires_on_attribute_change_with_attribute_not_below( async def test_if_not_fires_on_attribute_change_with_attribute_not_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
assert await async_setup_component( assert await async_setup_component(
@ -855,7 +858,7 @@ async def test_if_not_fires_on_attribute_change_with_attribute_not_below(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_not_fires_on_entity_change_with_attribute_below( async def test_if_not_fires_on_entity_change_with_attribute_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
assert await async_setup_component( assert await async_setup_component(
@ -881,7 +884,7 @@ async def test_if_not_fires_on_entity_change_with_attribute_below(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_if_not_fires_on_entity_change_with_not_attribute_below( async def test_if_not_fires_on_entity_change_with_not_attribute_below(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
assert await async_setup_component( assert await async_setup_component(
@ -907,7 +910,7 @@ async def test_if_not_fires_on_entity_change_with_not_attribute_below(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_fires_on_attr_change_with_attribute_below_and_multiple_attr( async def test_fires_on_attr_change_with_attribute_below_and_multiple_attr(
hass: HomeAssistant, calls, below hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None: ) -> None:
"""Test attributes change.""" """Test attributes change."""
hass.states.async_set( hass.states.async_set(
@ -938,7 +941,9 @@ async def test_fires_on_attr_change_with_attribute_below_and_multiple_attr(
@pytest.mark.parametrize("below", [10, "input_number.value_10"]) @pytest.mark.parametrize("below", [10, "input_number.value_10"])
async def test_template_list(hass: HomeAssistant, calls, below) -> None: async def test_template_list(
hass: HomeAssistant, calls: list[ServiceCall], below: int | str
) -> None:
"""Test template list.""" """Test template list."""
hass.states.async_set("test.entity", "entity", {"test_attribute": [11, 15, 11]}) hass.states.async_set("test.entity", "entity", {"test_attribute": [11, 15, 11]})
await hass.async_block_till_done() await hass.async_block_till_done()
@ -964,7 +969,9 @@ async def test_template_list(hass: HomeAssistant, calls, below) -> None:
@pytest.mark.parametrize("below", [10.0, "input_number.value_10"]) @pytest.mark.parametrize("below", [10.0, "input_number.value_10"])
async def test_template_string(hass: HomeAssistant, calls, below) -> None: async def test_template_string(
hass: HomeAssistant, calls: list[ServiceCall], below: float | str
) -> None:
"""Test template string.""" """Test template string."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -1005,7 +1012,7 @@ async def test_template_string(hass: HomeAssistant, calls, below) -> None:
async def test_not_fires_on_attr_change_with_attr_not_below_multiple_attr( async def test_not_fires_on_attr_change_with_attr_not_below_multiple_attr(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test if not fired changed attributes.""" """Test if not fired changed attributes."""
assert await async_setup_component( assert await async_setup_component(
@ -1040,7 +1047,9 @@ async def test_not_fires_on_attr_change_with_attr_not_below_multiple_attr(
("input_number.value_8", "input_number.value_12"), ("input_number.value_8", "input_number.value_12"),
], ],
) )
async def test_if_action(hass: HomeAssistant, calls, above, below) -> None: async def test_if_action(
hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None:
"""Test if action.""" """Test if action."""
entity_id = "domain.test_entity" entity_id = "domain.test_entity"
assert await async_setup_component( assert await async_setup_component(
@ -1088,7 +1097,9 @@ async def test_if_action(hass: HomeAssistant, calls, above, below) -> None:
("input_number.value_8", "input_number.value_12"), ("input_number.value_8", "input_number.value_12"),
], ],
) )
async def test_if_fails_setup_bad_for(hass: HomeAssistant, calls, above, below) -> None: async def test_if_fails_setup_bad_for(
hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None:
"""Test for setup failure for bad for.""" """Test for setup failure for bad for."""
hass.states.async_set("test.entity", 5) hass.states.async_set("test.entity", 5)
await hass.async_block_till_done() await hass.async_block_till_done()
@ -1114,7 +1125,7 @@ async def test_if_fails_setup_bad_for(hass: HomeAssistant, calls, above, below)
async def test_if_fails_setup_for_without_above_below( async def test_if_fails_setup_for_without_above_below(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for setup failures for missing above or below.""" """Test for setup failures for missing above or below."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
@ -1145,7 +1156,11 @@ async def test_if_fails_setup_for_without_above_below(
], ],
) )
async def test_if_not_fires_on_entity_change_with_for( async def test_if_not_fires_on_entity_change_with_for(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int | str,
below: int | str,
) -> None: ) -> None:
"""Test for not firing on entity change with for.""" """Test for not firing on entity change with for."""
assert await async_setup_component( assert await async_setup_component(
@ -1185,7 +1200,7 @@ async def test_if_not_fires_on_entity_change_with_for(
], ],
) )
async def test_if_not_fires_on_entities_change_with_for_after_stop( async def test_if_not_fires_on_entities_change_with_for_after_stop(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test for not firing on entities change with for after stop.""" """Test for not firing on entities change with for after stop."""
hass.states.async_set("test.entity_1", 0) hass.states.async_set("test.entity_1", 0)
@ -1246,7 +1261,11 @@ async def test_if_not_fires_on_entities_change_with_for_after_stop(
], ],
) )
async def test_if_fires_on_entity_change_with_for_attribute_change( async def test_if_fires_on_entity_change_with_for_attribute_change(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int | str,
below: int | str,
) -> None: ) -> None:
"""Test for firing on entity change with for and attribute change.""" """Test for firing on entity change with for and attribute change."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1292,7 +1311,7 @@ async def test_if_fires_on_entity_change_with_for_attribute_change(
], ],
) )
async def test_if_fires_on_entity_change_with_for( async def test_if_fires_on_entity_change_with_for(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test for firing on entity change with for.""" """Test for firing on entity change with for."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1323,7 +1342,9 @@ async def test_if_fires_on_entity_change_with_for(
@pytest.mark.parametrize("above", [10, "input_number.value_10"]) @pytest.mark.parametrize("above", [10, "input_number.value_10"])
async def test_wait_template_with_trigger(hass: HomeAssistant, calls, above) -> None: async def test_wait_template_with_trigger(
hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None:
"""Test using wait template with 'trigger.entity_id'.""" """Test using wait template with 'trigger.entity_id'."""
hass.states.async_set("test.entity", "0") hass.states.async_set("test.entity", "0")
await hass.async_block_till_done() await hass.async_block_till_done()
@ -1374,7 +1395,11 @@ async def test_wait_template_with_trigger(hass: HomeAssistant, calls, above) ->
], ],
) )
async def test_if_fires_on_entities_change_no_overlap( async def test_if_fires_on_entities_change_no_overlap(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int | str,
below: int | str,
) -> None: ) -> None:
"""Test for firing on entities change with no overlap.""" """Test for firing on entities change with no overlap."""
hass.states.async_set("test.entity_1", 0) hass.states.async_set("test.entity_1", 0)
@ -1429,7 +1454,11 @@ async def test_if_fires_on_entities_change_no_overlap(
], ],
) )
async def test_if_fires_on_entities_change_overlap( async def test_if_fires_on_entities_change_overlap(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int | str,
below: int | str,
) -> None: ) -> None:
"""Test for firing on entities change with overlap.""" """Test for firing on entities change with overlap."""
hass.states.async_set("test.entity_1", 0) hass.states.async_set("test.entity_1", 0)
@ -1495,7 +1524,7 @@ async def test_if_fires_on_entities_change_overlap(
], ],
) )
async def test_if_fires_on_change_with_for_template_1( async def test_if_fires_on_change_with_for_template_1(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1536,7 +1565,7 @@ async def test_if_fires_on_change_with_for_template_1(
], ],
) )
async def test_if_fires_on_change_with_for_template_2( async def test_if_fires_on_change_with_for_template_2(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1577,7 +1606,7 @@ async def test_if_fires_on_change_with_for_template_2(
], ],
) )
async def test_if_fires_on_change_with_for_template_3( async def test_if_fires_on_change_with_for_template_3(
hass: HomeAssistant, calls, above, below hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1609,7 +1638,7 @@ async def test_if_fires_on_change_with_for_template_3(
async def test_if_not_fires_on_error_with_for_template( async def test_if_not_fires_on_error_with_for_template(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing on error with for template.""" """Test for not firing on error with for template."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
@ -1655,7 +1684,9 @@ async def test_if_not_fires_on_error_with_for_template(
("input_number.value_8", "input_number.value_12"), ("input_number.value_8", "input_number.value_12"),
], ],
) )
async def test_invalid_for_template(hass: HomeAssistant, calls, above, below) -> None: async def test_invalid_for_template(
hass: HomeAssistant, calls: list[ServiceCall], above: int | str, below: int | str
) -> None:
"""Test for invalid for template.""" """Test for invalid for template."""
hass.states.async_set("test.entity", 0) hass.states.async_set("test.entity", 0)
await hass.async_block_till_done() await hass.async_block_till_done()
@ -1693,7 +1724,11 @@ async def test_invalid_for_template(hass: HomeAssistant, calls, above, below) ->
], ],
) )
async def test_if_fires_on_entities_change_overlap_for_template( async def test_if_fires_on_entities_change_overlap_for_template(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int | str,
below: int | str,
) -> None: ) -> None:
"""Test for firing on entities change with overlap and for template.""" """Test for firing on entities change with overlap and for template."""
hass.states.async_set("test.entity_1", 0) hass.states.async_set("test.entity_1", 0)
@ -1788,7 +1823,7 @@ async def test_schema_unacceptable_entities(hass: HomeAssistant) -> None:
@pytest.mark.parametrize("above", [3, "input_number.value_3"]) @pytest.mark.parametrize("above", [3, "input_number.value_3"])
async def test_attribute_if_fires_on_entity_change_with_both_filters( async def test_attribute_if_fires_on_entity_change_with_both_filters(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test for firing if both filters are match attribute.""" """Test for firing if both filters are match attribute."""
hass.states.async_set("test.entity", "bla", {"test-measurement": 1}) hass.states.async_set("test.entity", "bla", {"test-measurement": 1})
@ -1817,7 +1852,7 @@ async def test_attribute_if_fires_on_entity_change_with_both_filters(
@pytest.mark.parametrize("above", [3, "input_number.value_3"]) @pytest.mark.parametrize("above", [3, "input_number.value_3"])
async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop( async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop(
hass: HomeAssistant, calls, above hass: HomeAssistant, calls: list[ServiceCall], above: int | str
) -> None: ) -> None:
"""Test for not firing on entity change with for after stop trigger.""" """Test for not firing on entity change with for after stop trigger."""
hass.states.async_set("test.entity", "bla", {"test-measurement": 1}) hass.states.async_set("test.entity", "bla", {"test-measurement": 1})
@ -1856,7 +1891,11 @@ async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop(
[(8, 12)], [(8, 12)],
) )
async def test_variables_priority( async def test_variables_priority(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, above, below hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
above: int,
below: int,
) -> None: ) -> None:
"""Test an externally defined trigger variable is overridden.""" """Test an externally defined trigger variable is overridden."""
hass.states.async_set("test.entity_1", 0) hass.states.async_set("test.entity_1", 0)
@ -1911,7 +1950,9 @@ async def test_variables_priority(
@pytest.mark.parametrize("multiplier", [1, 5]) @pytest.mark.parametrize("multiplier", [1, 5])
async def test_template_variable(hass: HomeAssistant, calls, multiplier) -> None: async def test_template_variable(
hass: HomeAssistant, calls: list[ServiceCall], multiplier: int
) -> None:
"""Test template variable.""" """Test template variable."""
hass.states.async_set("test.entity", "entity", {"test_attribute": [11, 15, 11]}) hass.states.async_set("test.entity", "entity", {"test_attribute": [11, 15, 11]})
await hass.async_block_till_done() await hass.async_block_till_done()

118
tests/components/homeassistant/triggers/test_state.py
View File

@ -40,7 +40,9 @@ def setup_comp(hass):
hass.states.async_set("test.entity", "hello") hass.states.async_set("test.entity", "hello")
async def test_if_fires_on_entity_change(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_entity_change(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on entity change.""" """Test for firing on entity change."""
context = Context() context = Context()
hass.states.async_set("test.entity", "hello") hass.states.async_set("test.entity", "hello")
@ -88,7 +90,7 @@ async def test_if_fires_on_entity_change(hass: HomeAssistant, calls) -> None:
async def test_if_fires_on_entity_change_uuid( async def test_if_fires_on_entity_change_uuid(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls hass: HomeAssistant, entity_registry: er.EntityRegistry, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change.""" """Test for firing on entity change."""
context = Context() context = Context()
@ -144,7 +146,7 @@ async def test_if_fires_on_entity_change_uuid(
async def test_if_fires_on_entity_change_with_from_filter( async def test_if_fires_on_entity_change_with_from_filter(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with filter.""" """Test for firing on entity change with filter."""
assert await async_setup_component( assert await async_setup_component(
@ -199,7 +201,7 @@ async def test_if_fires_on_entity_change_with_not_from_filter(
async def test_if_fires_on_entity_change_with_to_filter( async def test_if_fires_on_entity_change_with_to_filter(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with to filter.""" """Test for firing on entity change with to filter."""
assert await async_setup_component( assert await async_setup_component(
@ -254,7 +256,7 @@ async def test_if_fires_on_entity_change_with_not_to_filter(
async def test_if_fires_on_entity_change_with_from_filter_all( async def test_if_fires_on_entity_change_with_from_filter_all(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with filter.""" """Test for firing on entity change with filter."""
assert await async_setup_component( assert await async_setup_component(
@ -280,7 +282,7 @@ async def test_if_fires_on_entity_change_with_from_filter_all(
async def test_if_fires_on_entity_change_with_to_filter_all( async def test_if_fires_on_entity_change_with_to_filter_all(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with to filter.""" """Test for firing on entity change with to filter."""
assert await async_setup_component( assert await async_setup_component(
@ -306,7 +308,7 @@ async def test_if_fires_on_entity_change_with_to_filter_all(
async def test_if_fires_on_attribute_change_with_to_filter( async def test_if_fires_on_attribute_change_with_to_filter(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing on attribute change.""" """Test for not firing on attribute change."""
assert await async_setup_component( assert await async_setup_component(
@ -332,7 +334,7 @@ async def test_if_fires_on_attribute_change_with_to_filter(
async def test_if_fires_on_entity_change_with_both_filters( async def test_if_fires_on_entity_change_with_both_filters(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if both filters are a non match.""" """Test for firing if both filters are a non match."""
assert await async_setup_component( assert await async_setup_component(
@ -451,7 +453,9 @@ async def test_if_fires_on_entity_change_with_from_not_to(
assert len(calls) == 2 assert len(calls) == 2
async def test_if_not_fires_if_to_filter_not_match(hass: HomeAssistant, calls) -> None: async def test_if_not_fires_if_to_filter_not_match(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for not firing if to filter is not a match.""" """Test for not firing if to filter is not a match."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -476,7 +480,7 @@ async def test_if_not_fires_if_to_filter_not_match(hass: HomeAssistant, calls) -
async def test_if_not_fires_if_from_filter_not_match( async def test_if_not_fires_if_from_filter_not_match(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing if from filter is not a match.""" """Test for not firing if from filter is not a match."""
hass.states.async_set("test.entity", "bye") hass.states.async_set("test.entity", "bye")
@ -503,7 +507,9 @@ async def test_if_not_fires_if_from_filter_not_match(
assert len(calls) == 0 assert len(calls) == 0
async def test_if_not_fires_if_entity_not_match(hass: HomeAssistant, calls) -> None: async def test_if_not_fires_if_entity_not_match(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for not firing if entity is not matching.""" """Test for not firing if entity is not matching."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -522,7 +528,7 @@ async def test_if_not_fires_if_entity_not_match(hass: HomeAssistant, calls) -> N
assert len(calls) == 0 assert len(calls) == 0
async def test_if_action(hass: HomeAssistant, calls) -> None: async def test_if_action(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test for to action.""" """Test for to action."""
entity_id = "domain.test_entity" entity_id = "domain.test_entity"
test_state = "new_state" test_state = "new_state"
@ -554,7 +560,9 @@ async def test_if_action(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fails_setup_if_to_boolean_value(hass: HomeAssistant, calls) -> None: async def test_if_fails_setup_if_to_boolean_value(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for setup failure for boolean to.""" """Test for setup failure for boolean to."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component( assert await async_setup_component(
@ -574,7 +582,9 @@ async def test_if_fails_setup_if_to_boolean_value(hass: HomeAssistant, calls) ->
assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE
async def test_if_fails_setup_if_from_boolean_value(hass: HomeAssistant, calls) -> None: async def test_if_fails_setup_if_from_boolean_value(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for setup failure for boolean from.""" """Test for setup failure for boolean from."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component( assert await async_setup_component(
@ -594,7 +604,9 @@ async def test_if_fails_setup_if_from_boolean_value(hass: HomeAssistant, calls)
assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE
async def test_if_fails_setup_bad_for(hass: HomeAssistant, calls) -> None: async def test_if_fails_setup_bad_for(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for setup failure for bad for.""" """Test for setup failure for bad for."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component( assert await async_setup_component(
@ -616,7 +628,7 @@ async def test_if_fails_setup_bad_for(hass: HomeAssistant, calls) -> None:
async def test_if_not_fires_on_entity_change_with_for( async def test_if_not_fires_on_entity_change_with_for(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing on entity change with for.""" """Test for not firing on entity change with for."""
assert await async_setup_component( assert await async_setup_component(
@ -646,7 +658,7 @@ async def test_if_not_fires_on_entity_change_with_for(
async def test_if_not_fires_on_entities_change_with_for_after_stop( async def test_if_not_fires_on_entities_change_with_for_after_stop(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing on entity change with for after stop trigger.""" """Test for not firing on entity change with for after stop trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -695,7 +707,7 @@ async def test_if_not_fires_on_entities_change_with_for_after_stop(
async def test_if_fires_on_entity_change_with_for_attribute_change( async def test_if_fires_on_entity_change_with_for_attribute_change(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with for and attribute change.""" """Test for firing on entity change with for and attribute change."""
assert await async_setup_component( assert await async_setup_component(
@ -731,7 +743,7 @@ async def test_if_fires_on_entity_change_with_for_attribute_change(
async def test_if_fires_on_entity_change_with_for_multiple_force_update( async def test_if_fires_on_entity_change_with_for_multiple_force_update(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with for and force update.""" """Test for firing on entity change with for and force update."""
assert await async_setup_component( assert await async_setup_component(
@ -765,7 +777,9 @@ async def test_if_fires_on_entity_change_with_for_multiple_force_update(
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_entity_change_with_for(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_entity_change_with_for(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on entity change with for.""" """Test for firing on entity change with for."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -792,7 +806,7 @@ async def test_if_fires_on_entity_change_with_for(hass: HomeAssistant, calls) ->
async def test_if_fires_on_entity_change_with_for_without_to( async def test_if_fires_on_entity_change_with_for_without_to(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with for.""" """Test for firing on entity change with for."""
assert await async_setup_component( assert await async_setup_component(
@ -831,7 +845,7 @@ async def test_if_fires_on_entity_change_with_for_without_to(
async def test_if_does_not_fires_on_entity_change_with_for_without_to_2( async def test_if_does_not_fires_on_entity_change_with_for_without_to_2(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity change with for.""" """Test for firing on entity change with for."""
assert await async_setup_component( assert await async_setup_component(
@ -861,7 +875,7 @@ async def test_if_does_not_fires_on_entity_change_with_for_without_to_2(
async def test_if_fires_on_entity_creation_and_removal( async def test_if_fires_on_entity_creation_and_removal(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entity creation and removal, with to/from constraints.""" """Test for firing on entity creation and removal, with to/from constraints."""
# set automations for multiple combinations to/from # set automations for multiple combinations to/from
@ -927,7 +941,9 @@ async def test_if_fires_on_entity_creation_and_removal(
assert calls[3].context.parent_id == context_0.id assert calls[3].context.parent_id == context_0.id
async def test_if_fires_on_for_condition(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_for_condition(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing if condition is on.""" """Test for firing if condition is on."""
point1 = dt_util.utcnow() point1 = dt_util.utcnow()
point2 = point1 + timedelta(seconds=10) point2 = point1 + timedelta(seconds=10)
@ -965,7 +981,7 @@ async def test_if_fires_on_for_condition(hass: HomeAssistant, calls) -> None:
async def test_if_fires_on_for_condition_attribute_change( async def test_if_fires_on_for_condition_attribute_change(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if condition is on with attribute change.""" """Test for firing if condition is on with attribute change."""
point1 = dt_util.utcnow() point1 = dt_util.utcnow()
@ -1013,7 +1029,9 @@ async def test_if_fires_on_for_condition_attribute_change(
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fails_setup_for_without_time(hass: HomeAssistant, calls) -> None: async def test_if_fails_setup_for_without_time(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for setup failure if no time is provided.""" """Test for setup failure if no time is provided."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component( assert await async_setup_component(
@ -1035,7 +1053,9 @@ async def test_if_fails_setup_for_without_time(hass: HomeAssistant, calls) -> No
assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE
async def test_if_fails_setup_for_without_entity(hass: HomeAssistant, calls) -> None: async def test_if_fails_setup_for_without_entity(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for setup failure if no entity is provided.""" """Test for setup failure if no entity is provided."""
with assert_setup_component(1, automation.DOMAIN): with assert_setup_component(1, automation.DOMAIN):
assert await async_setup_component( assert await async_setup_component(
@ -1056,7 +1076,9 @@ async def test_if_fails_setup_for_without_entity(hass: HomeAssistant, calls) ->
assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE assert hass.states.get("automation.automation_0").state == STATE_UNAVAILABLE
async def test_wait_template_with_trigger(hass: HomeAssistant, calls) -> None: async def test_wait_template_with_trigger(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test using wait template with 'trigger.entity_id'.""" """Test using wait template with 'trigger.entity_id'."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -1096,7 +1118,7 @@ async def test_wait_template_with_trigger(hass: HomeAssistant, calls) -> None:
async def test_if_fires_on_entities_change_no_overlap( async def test_if_fires_on_entities_change_no_overlap(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entities change with no overlap.""" """Test for firing on entities change with no overlap."""
assert await async_setup_component( assert await async_setup_component(
@ -1137,7 +1159,7 @@ async def test_if_fires_on_entities_change_no_overlap(
async def test_if_fires_on_entities_change_overlap( async def test_if_fires_on_entities_change_overlap(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entities change with overlap.""" """Test for firing on entities change with overlap."""
assert await async_setup_component( assert await async_setup_component(
@ -1189,7 +1211,7 @@ async def test_if_fires_on_entities_change_overlap(
async def test_if_fires_on_change_with_for_template_1( async def test_if_fires_on_change_with_for_template_1(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
assert await async_setup_component( assert await async_setup_component(
@ -1217,7 +1239,7 @@ async def test_if_fires_on_change_with_for_template_1(
async def test_if_fires_on_change_with_for_template_2( async def test_if_fires_on_change_with_for_template_2(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
assert await async_setup_component( assert await async_setup_component(
@ -1245,7 +1267,7 @@ async def test_if_fires_on_change_with_for_template_2(
async def test_if_fires_on_change_with_for_template_3( async def test_if_fires_on_change_with_for_template_3(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
assert await async_setup_component( assert await async_setup_component(
@ -1273,7 +1295,7 @@ async def test_if_fires_on_change_with_for_template_3(
async def test_if_fires_on_change_with_for_template_4( async def test_if_fires_on_change_with_for_template_4(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on change with for template.""" """Test for firing on change with for template."""
assert await async_setup_component( assert await async_setup_component(
@ -1301,7 +1323,9 @@ async def test_if_fires_on_change_with_for_template_4(
assert len(calls) == 1 assert len(calls) == 1
async def test_if_fires_on_change_from_with_for(hass: HomeAssistant, calls) -> None: async def test_if_fires_on_change_from_with_for(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on change with from/for.""" """Test for firing on change with from/for."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -1330,7 +1354,9 @@ async def test_if_fires_on_change_from_with_for(hass: HomeAssistant, calls) -> N
assert len(calls) == 1 assert len(calls) == 1
async def test_if_not_fires_on_change_from_with_for(hass: HomeAssistant, calls) -> None: async def test_if_not_fires_on_change_from_with_for(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for firing on change with from/for.""" """Test for firing on change with from/for."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -1359,7 +1385,9 @@ async def test_if_not_fires_on_change_from_with_for(hass: HomeAssistant, calls)
assert len(calls) == 0 assert len(calls) == 0
async def test_invalid_for_template_1(hass: HomeAssistant, calls) -> None: async def test_invalid_for_template_1(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for invalid for template.""" """Test for invalid for template."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -1384,7 +1412,7 @@ async def test_invalid_for_template_1(hass: HomeAssistant, calls) -> None:
async def test_if_fires_on_entities_change_overlap_for_template( async def test_if_fires_on_entities_change_overlap_for_template(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing on entities change with overlap and for template.""" """Test for firing on entities change with overlap and for template."""
assert await async_setup_component( assert await async_setup_component(
@ -1443,7 +1471,7 @@ async def test_if_fires_on_entities_change_overlap_for_template(
async def test_attribute_if_fires_on_entity_change_with_both_filters( async def test_attribute_if_fires_on_entity_change_with_both_filters(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if both filters are match attribute.""" """Test for firing if both filters are match attribute."""
hass.states.async_set("test.entity", "bla", {"name": "hello"}) hass.states.async_set("test.entity", "bla", {"name": "hello"})
@ -1472,7 +1500,7 @@ async def test_attribute_if_fires_on_entity_change_with_both_filters(
async def test_attribute_if_fires_on_entity_where_attr_stays_constant( async def test_attribute_if_fires_on_entity_where_attr_stays_constant(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if attribute stays the same.""" """Test for firing if attribute stays the same."""
hass.states.async_set("test.entity", "bla", {"name": "hello", "other": "old_value"}) hass.states.async_set("test.entity", "bla", {"name": "hello", "other": "old_value"})
@ -1510,7 +1538,7 @@ async def test_attribute_if_fires_on_entity_where_attr_stays_constant(
async def test_attribute_if_fires_on_entity_where_attr_stays_constant_filter( async def test_attribute_if_fires_on_entity_where_attr_stays_constant_filter(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if attribute stays the same.""" """Test for firing if attribute stays the same."""
hass.states.async_set("test.entity", "bla", {"name": "other_name"}) hass.states.async_set("test.entity", "bla", {"name": "other_name"})
@ -1555,7 +1583,7 @@ async def test_attribute_if_fires_on_entity_where_attr_stays_constant_filter(
async def test_attribute_if_fires_on_entity_where_attr_stays_constant_all( async def test_attribute_if_fires_on_entity_where_attr_stays_constant_all(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if attribute stays the same.""" """Test for firing if attribute stays the same."""
hass.states.async_set("test.entity", "bla", {"name": "hello", "other": "old_value"}) hass.states.async_set("test.entity", "bla", {"name": "hello", "other": "old_value"})
@ -1600,7 +1628,7 @@ async def test_attribute_if_fires_on_entity_where_attr_stays_constant_all(
async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop( async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for not firing on entity change with for after stop trigger.""" """Test for not firing on entity change with for after stop trigger."""
hass.states.async_set("test.entity", "bla", {"name": "hello"}) hass.states.async_set("test.entity", "bla", {"name": "hello"})
@ -1656,7 +1684,7 @@ async def test_attribute_if_not_fires_on_entities_change_with_for_after_stop(
async def test_attribute_if_fires_on_entity_change_with_both_filters_boolean( async def test_attribute_if_fires_on_entity_change_with_both_filters_boolean(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if both filters are match attribute.""" """Test for firing if both filters are match attribute."""
hass.states.async_set("test.entity", "bla", {"happening": False}) hass.states.async_set("test.entity", "bla", {"happening": False})
@ -1685,7 +1713,7 @@ async def test_attribute_if_fires_on_entity_change_with_both_filters_boolean(
async def test_variables_priority( async def test_variables_priority(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test an externally defined trigger variable is overridden.""" """Test an externally defined trigger variable is overridden."""
assert await async_setup_component( assert await async_setup_component(

34
tests/components/homeassistant/triggers/test_time.py
View File

@ -16,7 +16,7 @@ from homeassistant.const import (
SERVICE_TURN_OFF, SERVICE_TURN_OFF,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -29,7 +29,7 @@ from tests.common import (
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -41,7 +41,7 @@ def setup_comp(hass):
async def test_if_fires_using_at( async def test_if_fires_using_at(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing at.""" """Test for firing at."""
now = dt_util.now() now = dt_util.now()
@ -80,7 +80,11 @@ async def test_if_fires_using_at(
("has_date", "has_time"), [(True, True), (True, False), (False, True)] ("has_date", "has_time"), [(True, True), (True, False), (False, True)]
) )
async def test_if_fires_using_at_input_datetime( async def test_if_fires_using_at_input_datetime(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls, has_date, has_time hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
calls: list[ServiceCall],
has_date,
has_time,
) -> None: ) -> None:
"""Test for firing at input_datetime.""" """Test for firing at input_datetime."""
await async_setup_component( await async_setup_component(
@ -161,7 +165,7 @@ async def test_if_fires_using_at_input_datetime(
async def test_if_fires_using_multiple_at( async def test_if_fires_using_multiple_at(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing at.""" """Test for firing at."""
@ -202,7 +206,7 @@ async def test_if_fires_using_multiple_at(
async def test_if_not_fires_using_wrong_at( async def test_if_not_fires_using_wrong_at(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""YAML translates time values to total seconds. """YAML translates time values to total seconds.
@ -241,7 +245,7 @@ async def test_if_not_fires_using_wrong_at(
assert len(calls) == 0 assert len(calls) == 0
async def test_if_action_before(hass: HomeAssistant, calls) -> None: async def test_if_action_before(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test for if action before.""" """Test for if action before."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -272,7 +276,7 @@ async def test_if_action_before(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_action_after(hass: HomeAssistant, calls) -> None: async def test_if_action_after(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test for if action after.""" """Test for if action after."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -303,7 +307,9 @@ async def test_if_action_after(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_action_one_weekday(hass: HomeAssistant, calls) -> None: async def test_if_action_one_weekday(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for if action with one weekday.""" """Test for if action with one weekday."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -335,7 +341,9 @@ async def test_if_action_one_weekday(hass: HomeAssistant, calls) -> None:
assert len(calls) == 1 assert len(calls) == 1
async def test_if_action_list_weekday(hass: HomeAssistant, calls) -> None: async def test_if_action_list_weekday(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for action with a list of weekdays.""" """Test for action with a list of weekdays."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -408,7 +416,7 @@ async def test_untrack_time_change(hass: HomeAssistant) -> None:
async def test_if_fires_using_at_sensor( async def test_if_fires_using_at_sensor(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing at sensor time.""" """Test for firing at sensor time."""
now = dt_util.now() now = dt_util.now()
@ -535,7 +543,9 @@ def test_schema_invalid(conf) -> None:
time.TRIGGER_SCHEMA(conf) time.TRIGGER_SCHEMA(conf)
async def test_datetime_in_past_on_load(hass: HomeAssistant, calls) -> None: async def test_datetime_in_past_on_load(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test time trigger works if input_datetime is in past.""" """Test time trigger works if input_datetime is in past."""
await async_setup_component( await async_setup_component(
hass, hass,

24
tests/components/homeassistant/triggers/test_time_pattern.py
View File

@ -9,7 +9,7 @@ import voluptuous as vol
from homeassistant.components import automation from homeassistant.components import automation
from homeassistant.components.homeassistant.triggers import time_pattern from homeassistant.components.homeassistant.triggers import time_pattern
from homeassistant.const import ATTR_ENTITY_ID, ENTITY_MATCH_ALL, SERVICE_TURN_OFF from homeassistant.const import ATTR_ENTITY_ID, ENTITY_MATCH_ALL, SERVICE_TURN_OFF
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -17,7 +17,7 @@ from tests.common import async_fire_time_changed, async_mock_service, mock_compo
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -29,7 +29,7 @@ def setup_comp(hass):
async def test_if_fires_when_hour_matches( async def test_if_fires_when_hour_matches(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if hour is matching.""" """Test for firing if hour is matching."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -74,7 +74,7 @@ async def test_if_fires_when_hour_matches(
async def test_if_fires_when_minute_matches( async def test_if_fires_when_minute_matches(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if minutes are matching.""" """Test for firing if minutes are matching."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -105,7 +105,7 @@ async def test_if_fires_when_minute_matches(
async def test_if_fires_when_second_matches( async def test_if_fires_when_second_matches(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if seconds are matching.""" """Test for firing if seconds are matching."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -136,7 +136,7 @@ async def test_if_fires_when_second_matches(
async def test_if_fires_when_second_as_string_matches( async def test_if_fires_when_second_as_string_matches(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if seconds are matching.""" """Test for firing if seconds are matching."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -169,7 +169,7 @@ async def test_if_fires_when_second_as_string_matches(
async def test_if_fires_when_all_matches( async def test_if_fires_when_all_matches(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing if everything matches.""" """Test for firing if everything matches."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -202,7 +202,7 @@ async def test_if_fires_when_all_matches(
async def test_if_fires_periodic_seconds( async def test_if_fires_periodic_seconds(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing periodically every second.""" """Test for firing periodically every second."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -235,7 +235,7 @@ async def test_if_fires_periodic_seconds(
async def test_if_fires_periodic_minutes( async def test_if_fires_periodic_minutes(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing periodically every minute.""" """Test for firing periodically every minute."""
@ -269,7 +269,7 @@ async def test_if_fires_periodic_minutes(
async def test_if_fires_periodic_hours( async def test_if_fires_periodic_hours(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing periodically every hour.""" """Test for firing periodically every hour."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -302,7 +302,7 @@ async def test_if_fires_periodic_hours(
async def test_default_values( async def test_default_values(
hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls hass: HomeAssistant, freezer: FrozenDateTimeFactory, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for firing at 2 minutes every hour.""" """Test for firing at 2 minutes every hour."""
now = dt_util.utcnow() now = dt_util.utcnow()
@ -343,7 +343,7 @@ async def test_default_values(
assert len(calls) == 2 assert len(calls) == 2
async def test_invalid_schemas(hass: HomeAssistant, calls) -> None: async def test_invalid_schemas(hass: HomeAssistant, calls: list[ServiceCall]) -> None:
"""Test invalid schemas.""" """Test invalid schemas."""
schemas = ( schemas = (
None, None,

8
tests/components/homekit_controller/test_device_trigger.py
View File

@ -9,7 +9,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.homekit_controller.const import DOMAIN from homeassistant.components.homekit_controller.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -24,7 +24,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -239,7 +239,7 @@ async def test_handle_events(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test that events are handled.""" """Test that events are handled."""
helper = await setup_test_component(hass, create_remote) helper = await setup_test_component(hass, create_remote)
@ -359,7 +359,7 @@ async def test_handle_events_late_setup(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test that events are handled when setup happens after startup.""" """Test that events are handled when setup happens after startup."""
helper = await setup_test_component(hass, create_remote) helper = await setup_test_component(hass, create_remote)

3
tests/components/hue/conftest.py
View File

@ -15,6 +15,7 @@ import pytest
from homeassistant.components import hue from homeassistant.components import hue
from homeassistant.components.hue.v1 import sensor_base as hue_sensor_base from homeassistant.components.hue.v1 import sensor_base as hue_sensor_base
from homeassistant.components.hue.v2.device import async_setup_devices from homeassistant.components.hue.v2.device import async_setup_devices
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import ( from tests.common import (
@ -288,6 +289,6 @@ def get_device_reg(hass):
@pytest.fixture(name="calls") @pytest.fixture(name="calls")
def track_calls(hass): def track_calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")

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

@ -5,8 +5,8 @@ from pytest_unordered import unordered
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.core import HomeAssistant, ServiceCall
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from .conftest import setup_platform from .conftest import setup_platform
@ -18,7 +18,10 @@ REMOTES_RESPONSE = {"7": HUE_TAP_REMOTE_1, "8": HUE_DIMMER_REMOTE_1}
async def test_get_triggers( async def test_get_triggers(
hass: HomeAssistant, entity_registry: er.EntityRegistry, mock_bridge_v1, device_reg hass: HomeAssistant,
entity_registry: er.EntityRegistry,
mock_bridge_v1,
device_reg: dr.DeviceRegistry,
) -> None: ) -> 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)
@ -86,7 +89,10 @@ async def test_get_triggers(
async def test_if_fires_on_state_change( async def test_if_fires_on_state_change(
hass: HomeAssistant, mock_bridge_v1, device_reg, calls hass: HomeAssistant,
mock_bridge_v1,
device_reg: dr.DeviceRegistry,
calls: list[ServiceCall],
) -> None: ) -> 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)

8
tests/components/humidifier/test_device_condition.py
View File

@ -8,7 +8,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.humidifier import DOMAIN, const, device_condition from homeassistant.components.humidifier import DOMAIN, const, device_condition
from homeassistant.const import ATTR_MODE, STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import ATTR_MODE, STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -30,7 +30,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -153,7 +153,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -273,7 +273,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

10
tests/components/humidifier/test_device_trigger.py
View File

@ -16,7 +16,7 @@ from homeassistant.const import (
STATE_ON, STATE_ON,
EntityCategory, EntityCategory,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -40,7 +40,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -166,7 +166,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -429,7 +429,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -484,7 +484,7 @@ async def test_if_fires_on_state_change_legacy(
async def test_invalid_config( async def test_invalid_config(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls hass: HomeAssistant, entity_registry: er.EntityRegistry, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
entry = entity_registry.async_get_or_create(DOMAIN, "test", "5678") entry = entity_registry.async_get_or_create(DOMAIN, "test", "5678")

14
tests/components/kodi/test_device_trigger.py
View File

@ -6,7 +6,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.kodi import DOMAIN from homeassistant.components.kodi import DOMAIN
from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.media_player import DOMAIN as MP_DOMAIN
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -25,7 +25,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -75,7 +75,10 @@ async def test_get_triggers(
async def test_if_fires_on_state_change( async def test_if_fires_on_state_change(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls, kodi_media_player hass: HomeAssistant,
entity_registry: er.EntityRegistry,
calls: list[ServiceCall],
kodi_media_player,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
entry = entity_registry.async_get(kodi_media_player) entry = entity_registry.async_get(kodi_media_player)
@ -148,7 +151,10 @@ async def test_if_fires_on_state_change(
async def test_if_fires_on_state_change_legacy( async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, entity_registry: er.EntityRegistry, calls, kodi_media_player hass: HomeAssistant,
entity_registry: er.EntityRegistry,
calls: list[ServiceCall],
kodi_media_player,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
entry = entity_registry.async_get(kodi_media_player) entry = entity_registry.async_get(kodi_media_player)

3
tests/components/lcn/conftest.py
View File

@ -12,6 +12,7 @@ import pytest
from homeassistant.components.lcn.const import DOMAIN from homeassistant.components.lcn.const import DOMAIN
from homeassistant.components.lcn.helpers import generate_unique_id from homeassistant.components.lcn.helpers import generate_unique_id
from homeassistant.const import CONF_HOST from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import device_registry as dr from homeassistant.helpers import device_registry as dr
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -78,7 +79,7 @@ def create_config_entry(name):
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")

12
tests/components/lcn/test_device_trigger.py
View File

@ -11,7 +11,7 @@ from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.lcn import device_trigger from homeassistant.components.lcn import device_trigger
from homeassistant.components.lcn.const import DOMAIN, KEY_ACTIONS, SENDKEYS from homeassistant.components.lcn.const import DOMAIN, KEY_ACTIONS, SENDKEYS
from homeassistant.const import CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM, CONF_TYPE from homeassistant.const import CONF_DEVICE_ID, CONF_DOMAIN, CONF_PLATFORM, CONF_TYPE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import config_validation as cv, device_registry as dr from homeassistant.helpers import config_validation as cv, device_registry as dr
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -72,7 +72,7 @@ async def test_get_triggers_non_module_device(
async def test_if_fires_on_transponder_event( async def test_if_fires_on_transponder_event(
hass: HomeAssistant, calls, entry, lcn_connection hass: HomeAssistant, calls: list[ServiceCall], entry, lcn_connection
) -> None: ) -> None:
"""Test for transponder event triggers firing.""" """Test for transponder event triggers firing."""
address = (0, 7, False) address = (0, 7, False)
@ -119,7 +119,7 @@ async def test_if_fires_on_transponder_event(
async def test_if_fires_on_fingerprint_event( async def test_if_fires_on_fingerprint_event(
hass: HomeAssistant, calls, entry, lcn_connection hass: HomeAssistant, calls: list[ServiceCall], entry, lcn_connection
) -> None: ) -> None:
"""Test for fingerprint event triggers firing.""" """Test for fingerprint event triggers firing."""
address = (0, 7, False) address = (0, 7, False)
@ -166,7 +166,7 @@ async def test_if_fires_on_fingerprint_event(
async def test_if_fires_on_codelock_event( async def test_if_fires_on_codelock_event(
hass: HomeAssistant, calls, entry, lcn_connection hass: HomeAssistant, calls: list[ServiceCall], entry, lcn_connection
) -> None: ) -> None:
"""Test for codelock event triggers firing.""" """Test for codelock event triggers firing."""
address = (0, 7, False) address = (0, 7, False)
@ -213,7 +213,7 @@ async def test_if_fires_on_codelock_event(
async def test_if_fires_on_transmitter_event( async def test_if_fires_on_transmitter_event(
hass: HomeAssistant, calls, entry, lcn_connection hass: HomeAssistant, calls: list[ServiceCall], entry, lcn_connection
) -> None: ) -> None:
"""Test for transmitter event triggers firing.""" """Test for transmitter event triggers firing."""
address = (0, 7, False) address = (0, 7, False)
@ -269,7 +269,7 @@ async def test_if_fires_on_transmitter_event(
async def test_if_fires_on_send_keys_event( async def test_if_fires_on_send_keys_event(
hass: HomeAssistant, calls, entry, lcn_connection hass: HomeAssistant, calls: list[ServiceCall], entry, lcn_connection
) -> None: ) -> None:
"""Test for send_keys event triggers firing.""" """Test for send_keys event triggers firing."""
address = (0, 7, False) address = (0, 7, False)

4
tests/components/lg_netcast/conftest.py
View File

@ -2,10 +2,12 @@
import pytest import pytest
from homeassistant.core import HomeAssistant, ServiceCall
from tests.common import async_mock_service from tests.common import async_mock_service
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")

4
tests/components/lg_netcast/test_trigger.py
View File

@ -77,7 +77,9 @@ async def test_lg_netcast_turn_on_trigger_device_id(
assert len(calls) == 0 assert len(calls) == 0
async def test_lg_netcast_turn_on_trigger_entity_id(hass: HomeAssistant, calls): async def test_lg_netcast_turn_on_trigger_entity_id(
hass: HomeAssistant, calls: list[ServiceCall]
):
"""Test for turn_on triggers by entity firing.""" """Test for turn_on triggers by entity firing."""
await setup_lgnetcast(hass) await setup_lgnetcast(hass)

8
tests/components/light/test_device_action.py
View File

@ -14,7 +14,7 @@ from homeassistant.components.light import (
LightEntityFeature, LightEntityFeature,
) )
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -33,7 +33,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -470,7 +470,7 @@ async def test_action(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off actions.""" """Test for turn_on and turn_off actions."""
@ -635,7 +635,7 @@ async def test_action_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off actions.""" """Test for turn_on and turn_off actions."""

10
tests/components/light/test_device_condition.py
View File

@ -10,7 +10,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.light import DOMAIN from homeassistant.components.light import DOMAIN
from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import CONF_PLATFORM, STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -32,7 +32,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -184,7 +184,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
@ -271,7 +271,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
@ -330,7 +330,7 @@ async def test_if_fires_on_for_condition(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
mock_light_entities: list[MockLight], mock_light_entities: list[MockLight],
) -> None: ) -> None:
"""Test for firing if condition is on with delay.""" """Test for firing if condition is on with delay."""

10
tests/components/light/test_device_trigger.py
View File

@ -9,7 +9,7 @@ from homeassistant.components import automation
from homeassistant.components.device_automation import DeviceAutomationType from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.components.light import DOMAIN from homeassistant.components.light import DOMAIN
from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory from homeassistant.const import STATE_OFF, STATE_ON, EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -38,7 +38,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -188,7 +188,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
@ -281,7 +281,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
@ -335,7 +335,7 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
enable_custom_integrations: None, enable_custom_integrations: None,
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""

42
tests/components/litejet/test_trigger.py
View File

@ -9,7 +9,7 @@ import pytest
from homeassistant import setup from homeassistant import setup
from homeassistant.components import automation from homeassistant.components import automation
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from . import async_init_integration from . import async_init_integration
@ -31,7 +31,7 @@ ENTITY_OTHER_SWITCH_NUMBER = 2
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -100,7 +100,9 @@ async def setup_automation(hass, trigger):
await hass.async_block_till_done() await hass.async_block_till_done()
async def test_simple(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_simple(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test the simplest form of a LiteJet trigger.""" """Test the simplest form of a LiteJet trigger."""
await setup_automation( await setup_automation(
hass, {"platform": "litejet", "number": ENTITY_OTHER_SWITCH_NUMBER} hass, {"platform": "litejet", "number": ENTITY_OTHER_SWITCH_NUMBER}
@ -113,7 +115,9 @@ async def test_simple(hass: HomeAssistant, calls, mock_litejet) -> None:
assert calls[0].data["id"] == 0 assert calls[0].data["id"] == 0
async def test_only_release(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_only_release(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test the simplest form of a LiteJet trigger.""" """Test the simplest form of a LiteJet trigger."""
await setup_automation( await setup_automation(
hass, {"platform": "litejet", "number": ENTITY_OTHER_SWITCH_NUMBER} hass, {"platform": "litejet", "number": ENTITY_OTHER_SWITCH_NUMBER}
@ -124,7 +128,9 @@ async def test_only_release(hass: HomeAssistant, calls, mock_litejet) -> None:
assert len(calls) == 0 assert len(calls) == 0
async def test_held_more_than_short(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_more_than_short(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test a too short hold.""" """Test a too short hold."""
await setup_automation( await setup_automation(
hass, hass,
@ -141,7 +147,9 @@ async def test_held_more_than_short(hass: HomeAssistant, calls, mock_litejet) ->
assert len(calls) == 0 assert len(calls) == 0
async def test_held_more_than_long(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_more_than_long(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test a hold that is long enough.""" """Test a hold that is long enough."""
await setup_automation( await setup_automation(
hass, hass,
@ -161,7 +169,9 @@ async def test_held_more_than_long(hass: HomeAssistant, calls, mock_litejet) ->
assert len(calls) == 1 assert len(calls) == 1
async def test_held_less_than_short(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_less_than_short(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test a hold that is short enough.""" """Test a hold that is short enough."""
await setup_automation( await setup_automation(
hass, hass,
@ -180,7 +190,9 @@ async def test_held_less_than_short(hass: HomeAssistant, calls, mock_litejet) ->
assert calls[0].data["id"] == 0 assert calls[0].data["id"] == 0
async def test_held_less_than_long(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_less_than_long(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test a hold that is too long.""" """Test a hold that is too long."""
await setup_automation( await setup_automation(
hass, hass,
@ -199,7 +211,9 @@ async def test_held_less_than_long(hass: HomeAssistant, calls, mock_litejet) ->
assert len(calls) == 0 assert len(calls) == 0
async def test_held_in_range_short(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_in_range_short(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test an in-range trigger with a too short hold.""" """Test an in-range trigger with a too short hold."""
await setup_automation( await setup_automation(
hass, hass,
@ -218,7 +232,7 @@ async def test_held_in_range_short(hass: HomeAssistant, calls, mock_litejet) ->
async def test_held_in_range_just_right( async def test_held_in_range_just_right(
hass: HomeAssistant, calls, mock_litejet hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None: ) -> None:
"""Test an in-range trigger with a just right hold.""" """Test an in-range trigger with a just right hold."""
await setup_automation( await setup_automation(
@ -240,7 +254,9 @@ async def test_held_in_range_just_right(
assert calls[0].data["id"] == 0 assert calls[0].data["id"] == 0
async def test_held_in_range_long(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_held_in_range_long(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test an in-range trigger with a too long hold.""" """Test an in-range trigger with a too long hold."""
await setup_automation( await setup_automation(
hass, hass,
@ -260,7 +276,9 @@ async def test_held_in_range_long(hass: HomeAssistant, calls, mock_litejet) -> N
assert len(calls) == 0 assert len(calls) == 0
async def test_reload(hass: HomeAssistant, calls, mock_litejet) -> None: async def test_reload(
hass: HomeAssistant, calls: list[ServiceCall], mock_litejet
) -> None:
"""Test reloading automation.""" """Test reloading automation."""
await setup_automation( await setup_automation(
hass, hass,

8
tests/components/lock/test_device_condition.py
View File

@ -16,7 +16,7 @@ from homeassistant.const import (
STATE_UNLOCKING, STATE_UNLOCKING,
EntityCategory, EntityCategory,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -34,7 +34,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass: HomeAssistant): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -139,7 +139,7 @@ async def test_if_state(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -336,7 +336,7 @@ async def test_if_state_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off conditions.""" """Test for turn_on and turn_off conditions."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

10
tests/components/lock/test_device_trigger.py
View File

@ -18,7 +18,7 @@ from homeassistant.const import (
STATE_UNLOCKING, STATE_UNLOCKING,
EntityCategory, EntityCategory,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
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.helpers.entity_registry import RegistryEntryHider from homeassistant.helpers.entity_registry import RegistryEntryHider
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -39,7 +39,7 @@ def stub_blueprint_populate_autouse(stub_blueprint_populate: None) -> None:
@pytest.fixture @pytest.fixture
def calls(hass: HomeAssistant): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -212,7 +212,7 @@ async def test_if_fires_on_state_change(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -325,7 +325,7 @@ async def test_if_fires_on_state_change_legacy(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for turn_on and turn_off triggers firing.""" """Test for turn_on and turn_off triggers firing."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})
@ -382,7 +382,7 @@ async def test_if_fires_on_state_change_with_for(
hass: HomeAssistant, hass: HomeAssistant,
device_registry: dr.DeviceRegistry, device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry, entity_registry: er.EntityRegistry,
calls, calls: list[ServiceCall],
) -> None: ) -> None:
"""Test for triggers firing with delay.""" """Test for triggers firing with delay."""
config_entry = MockConfigEntry(domain="test", data={}) config_entry = MockConfigEntry(domain="test", data={})

16
tests/components/lutron_caseta/test_device_trigger.py
View File

@ -33,7 +33,7 @@ from homeassistant.const import (
CONF_PLATFORM, CONF_PLATFORM,
CONF_TYPE, CONF_TYPE,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import device_registry as dr from homeassistant.helpers import device_registry as dr
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
@ -103,7 +103,7 @@ MOCK_BUTTON_DEVICES = [
@pytest.fixture @pytest.fixture
def calls(hass): def calls(hass: HomeAssistant) -> list[ServiceCall]:
"""Track calls to a mock service.""" """Track calls to a mock service."""
return async_mock_service(hass, "test", "automation") return async_mock_service(hass, "test", "automation")
@ -220,7 +220,7 @@ async def test_none_serial_keypad(
async def test_if_fires_on_button_event( async def test_if_fires_on_button_event(
hass: HomeAssistant, calls, device_registry: dr.DeviceRegistry hass: HomeAssistant, calls: list[ServiceCall], device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Test for press trigger firing.""" """Test for press trigger firing."""
await _async_setup_lutron_with_picos(hass) await _async_setup_lutron_with_picos(hass)
@ -271,7 +271,7 @@ async def test_if_fires_on_button_event(
async def test_if_fires_on_button_event_without_lip( async def test_if_fires_on_button_event_without_lip(
hass: HomeAssistant, calls, device_registry: dr.DeviceRegistry hass: HomeAssistant, calls: list[ServiceCall], device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Test for press trigger firing on a device that does not support lip.""" """Test for press trigger firing on a device that does not support lip."""
await _async_setup_lutron_with_picos(hass) await _async_setup_lutron_with_picos(hass)
@ -319,7 +319,9 @@ async def test_if_fires_on_button_event_without_lip(
assert calls[0].data["some"] == "test_trigger_button_press" assert calls[0].data["some"] == "test_trigger_button_press"
async def test_validate_trigger_config_no_device(hass: HomeAssistant, calls) -> None: async def test_validate_trigger_config_no_device(
hass: HomeAssistant, calls: list[ServiceCall]
) -> None:
"""Test for no press with no device.""" """Test for no press with no device."""
assert await async_setup_component( assert await async_setup_component(
@ -358,7 +360,7 @@ async def test_validate_trigger_config_no_device(hass: HomeAssistant, calls) ->
async def test_validate_trigger_config_unknown_device( async def test_validate_trigger_config_unknown_device(
hass: HomeAssistant, calls hass: HomeAssistant, calls: list[ServiceCall]
) -> None: ) -> None:
"""Test for no press with an unknown device.""" """Test for no press with an unknown device."""
@ -442,7 +444,7 @@ async def test_validate_trigger_invalid_triggers(
async def test_if_fires_on_button_event_late_setup( async def test_if_fires_on_button_event_late_setup(
hass: HomeAssistant, calls, device_registry: dr.DeviceRegistry hass: HomeAssistant, calls: list[ServiceCall], device_registry: dr.DeviceRegistry
) -> None: ) -> None:
"""Test for press trigger firing with integration getting setup late.""" """Test for press trigger firing with integration getting setup late."""
config_entry_id = await _async_setup_lutron_with_picos(hass) config_entry_id = await _async_setup_lutron_with_picos(hass)