mirror of
https://github.com/esphome/esphome.git
synced 2026-06-24 12:33:10 +00:00
[valve] Add tests for valve.control action field combinations (#16126)
This commit is contained in:
@@ -442,6 +442,20 @@ valve:
|
|||||||
state: CLOSED
|
state: CLOSED
|
||||||
stop_action:
|
stop_action:
|
||||||
- logger.log: stop_action
|
- logger.log: stop_action
|
||||||
|
# Exercise valve.control with various field combinations so the
|
||||||
|
# ControlAction codegen paths get build coverage.
|
||||||
|
- valve.control:
|
||||||
|
id: template_valve
|
||||||
|
stop: true
|
||||||
|
- valve.control:
|
||||||
|
id: template_valve
|
||||||
|
position: 50%
|
||||||
|
- valve.control:
|
||||||
|
id: template_valve
|
||||||
|
state: OPEN
|
||||||
|
- valve.control:
|
||||||
|
id: template_valve
|
||||||
|
position: !lambda 'return 0.25f;'
|
||||||
optimistic: true
|
optimistic: true
|
||||||
|
|
||||||
text:
|
text:
|
||||||
|
|||||||
69
tests/integration/fixtures/valve_control_action.yaml
Normal file
69
tests/integration/fixtures/valve_control_action.yaml
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
esphome:
|
||||||
|
name: valve-control-action-test
|
||||||
|
host:
|
||||||
|
api:
|
||||||
|
logger:
|
||||||
|
level: DEBUG
|
||||||
|
|
||||||
|
globals:
|
||||||
|
- id: test_position
|
||||||
|
type: float
|
||||||
|
initial_value: "0.42"
|
||||||
|
|
||||||
|
valve:
|
||||||
|
- platform: template
|
||||||
|
name: "Test Valve"
|
||||||
|
id: test_valve
|
||||||
|
has_position: true
|
||||||
|
optimistic: true
|
||||||
|
assumed_state: true
|
||||||
|
open_action:
|
||||||
|
- valve.template.publish:
|
||||||
|
id: test_valve
|
||||||
|
position: 1.0
|
||||||
|
close_action:
|
||||||
|
- valve.template.publish:
|
||||||
|
id: test_valve
|
||||||
|
position: 0.0
|
||||||
|
stop_action:
|
||||||
|
- valve.template.publish:
|
||||||
|
id: test_valve
|
||||||
|
current_operation: IDLE
|
||||||
|
|
||||||
|
button:
|
||||||
|
# valve.control: position only
|
||||||
|
- platform: template
|
||||||
|
id: btn_position
|
||||||
|
name: "Set Position"
|
||||||
|
on_press:
|
||||||
|
- valve.control:
|
||||||
|
id: test_valve
|
||||||
|
position: 50%
|
||||||
|
|
||||||
|
# valve.control: state alias for position 1.0
|
||||||
|
- platform: template
|
||||||
|
id: btn_open_state
|
||||||
|
name: "Open State"
|
||||||
|
on_press:
|
||||||
|
- valve.control:
|
||||||
|
id: test_valve
|
||||||
|
state: OPEN
|
||||||
|
|
||||||
|
# valve.control: lambda position (exercises lambda path)
|
||||||
|
- platform: template
|
||||||
|
id: btn_lambda_position
|
||||||
|
name: "Lambda Position"
|
||||||
|
on_press:
|
||||||
|
- valve.control:
|
||||||
|
id: test_valve
|
||||||
|
position: !lambda "return id(test_position);"
|
||||||
|
|
||||||
|
# valve.control: stop only — template valve's stop_action publishes
|
||||||
|
# current_operation: IDLE.
|
||||||
|
- platform: template
|
||||||
|
id: btn_stop
|
||||||
|
name: "Stop Valve"
|
||||||
|
on_press:
|
||||||
|
- valve.control:
|
||||||
|
id: test_valve
|
||||||
|
stop: true
|
||||||
72
tests/integration/test_valve_control_action.py
Normal file
72
tests/integration/test_valve_control_action.py
Normal file
@@ -0,0 +1,72 @@
|
|||||||
|
"""Integration test for valve ControlAction.
|
||||||
|
|
||||||
|
Tests that valve.control automation actions work correctly across multiple
|
||||||
|
field combinations and the lambda path.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
from aioesphomeapi import ButtonInfo, EntityState, ValveInfo, ValveOperation, ValveState
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from .state_utils import InitialStateHelper, require_entity
|
||||||
|
from .types import APIClientConnectedFactory, RunCompiledFunction
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_valve_control_action(
|
||||||
|
yaml_config: str,
|
||||||
|
run_compiled: RunCompiledFunction,
|
||||||
|
api_client_connected: APIClientConnectedFactory,
|
||||||
|
) -> None:
|
||||||
|
"""Test valve ControlAction with constants and a lambda."""
|
||||||
|
loop = asyncio.get_running_loop()
|
||||||
|
async with run_compiled(yaml_config), api_client_connected() as client:
|
||||||
|
valve_state_future: asyncio.Future[ValveState] | None = None
|
||||||
|
|
||||||
|
def on_state(state: EntityState) -> None:
|
||||||
|
if (
|
||||||
|
isinstance(state, ValveState)
|
||||||
|
and valve_state_future is not None
|
||||||
|
and not valve_state_future.done()
|
||||||
|
):
|
||||||
|
valve_state_future.set_result(state)
|
||||||
|
|
||||||
|
async def wait_for_valve_state(timeout: float = 5.0) -> ValveState:
|
||||||
|
nonlocal valve_state_future
|
||||||
|
valve_state_future = loop.create_future()
|
||||||
|
try:
|
||||||
|
return await asyncio.wait_for(valve_state_future, timeout)
|
||||||
|
finally:
|
||||||
|
valve_state_future = None
|
||||||
|
|
||||||
|
entities, _ = await client.list_entities_services()
|
||||||
|
initial_state_helper = InitialStateHelper(entities)
|
||||||
|
client.subscribe_states(initial_state_helper.on_state_wrapper(on_state))
|
||||||
|
await initial_state_helper.wait_for_initial_states()
|
||||||
|
|
||||||
|
require_entity(entities, "test_valve", ValveInfo)
|
||||||
|
|
||||||
|
async def press_and_wait(name: str) -> ValveState:
|
||||||
|
btn = require_entity(entities, name.lower().replace(" ", "_"), ButtonInfo)
|
||||||
|
client.button_command(btn.key)
|
||||||
|
return await wait_for_valve_state()
|
||||||
|
|
||||||
|
# valve.control: position only
|
||||||
|
state = await press_and_wait("Set Position")
|
||||||
|
assert state.position == pytest.approx(0.5, abs=0.01)
|
||||||
|
|
||||||
|
# valve.control: state alias for position 1.0
|
||||||
|
state = await press_and_wait("Open State")
|
||||||
|
assert state.position == pytest.approx(1.0, abs=0.01)
|
||||||
|
|
||||||
|
# valve.control: lambda position (test_position global = 0.42)
|
||||||
|
state = await press_and_wait("Lambda Position")
|
||||||
|
assert state.position == pytest.approx(0.42, abs=0.01)
|
||||||
|
|
||||||
|
# valve.control: stop only — template valve's stop_action publishes
|
||||||
|
# current_operation: IDLE.
|
||||||
|
state = await press_and_wait("Stop Valve")
|
||||||
|
assert state.current_operation == ValveOperation.IDLE
|
||||||
Reference in New Issue
Block a user