Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-04-17 15:44:52 +01:00
Code Activity

Fix type annotations for set_or_remove_state test helper (#165843)

Browse Source
This commit is contained in:
Erik Montnemery
2026-03-18 08:03:45 +01:00
committed by GitHub
parent c4ff7fa676
commit d1d8754853
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/components/common.py
View File

@@ -176,7 +176,7 @@ def parametrize_target_entities(domain: str) -> list[tuple[dict, str, int]]:
]
class _StateDescription(TypedDict):
class StateDescription(TypedDict):
"""Test state with attributes."""
state: str | None
@@ -186,16 +186,16 @@ class _StateDescription(TypedDict):
class TriggerStateDescription(TypedDict):
"""Test state and expected service call count."""
included: _StateDescription # State for entities meant to be targeted
excluded: _StateDescription # State for entities not meant to be targeted
included: StateDescription # State for entities meant to be targeted
excluded: StateDescription # State for entities not meant to be targeted
count: int # Expected service call count
class ConditionStateDescription(TypedDict):
"""Test state and expected condition evaluation."""
included: _StateDescription # State for entities meant to be targeted
excluded: _StateDescription # State for entities not meant to be targeted
included: StateDescription # State for entities meant to be targeted
excluded: StateDescription # State for entities not meant to be targeted
condition_true: bool # If the condition is expected to evaluate to true
condition_true_first_entity: bool # If the condition is expected to evaluate to true for the first targeted entity
@@ -791,7 +791,7 @@ async def create_target_condition(
def set_or_remove_state(
hass: HomeAssistant,
entity_id: str,
state: TriggerStateDescription,
state: StateDescription,
) -> None:
"""Set or remove the state of an entity."""
if state["state"] is None: