mirror of
https://github.com/esphome/esphome.git
synced 2025-09-02 11:22:24 +01:00
118 lines
4.3 KiB
Python
118 lines
4.3 KiB
Python
"""Simple test that defer() maintains FIFO order."""
|
|
|
|
import asyncio
|
|
|
|
from aioesphomeapi import EntityState, Event, EventInfo, UserService
|
|
import pytest
|
|
|
|
from .types import APIClientConnectedFactory, RunCompiledFunction
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_scheduler_defer_fifo_simple(
|
|
yaml_config: str,
|
|
run_compiled: RunCompiledFunction,
|
|
api_client_connected: APIClientConnectedFactory,
|
|
) -> None:
|
|
"""Test that defer() maintains FIFO order with a simple test."""
|
|
|
|
async with run_compiled(yaml_config), api_client_connected() as client:
|
|
# Verify we can connect
|
|
device_info = await client.device_info()
|
|
assert device_info is not None
|
|
assert device_info.name == "scheduler-defer-fifo-simple"
|
|
|
|
# List entities and services
|
|
entity_info, services = await asyncio.wait_for(
|
|
client.list_entities_services(), timeout=5.0
|
|
)
|
|
|
|
# Find our test entities
|
|
test_complete_entity: EventInfo | None = None
|
|
test_result_entity: EventInfo | None = None
|
|
|
|
for entity in entity_info:
|
|
if isinstance(entity, EventInfo):
|
|
if entity.object_id == "test_complete":
|
|
test_complete_entity = entity
|
|
elif entity.object_id == "test_result":
|
|
test_result_entity = entity
|
|
|
|
assert test_complete_entity is not None, "test_complete event not found"
|
|
assert test_result_entity is not None, "test_result event not found"
|
|
|
|
# Find our test services
|
|
test_set_timeout_service: UserService | None = None
|
|
test_defer_service: UserService | None = None
|
|
for service in services:
|
|
if service.name == "test_set_timeout":
|
|
test_set_timeout_service = service
|
|
elif service.name == "test_defer":
|
|
test_defer_service = service
|
|
|
|
assert test_set_timeout_service is not None, (
|
|
"test_set_timeout service not found"
|
|
)
|
|
assert test_defer_service is not None, "test_defer service not found"
|
|
|
|
# Get the event loop
|
|
loop = asyncio.get_running_loop()
|
|
|
|
# Subscribe to states
|
|
# (events are delivered as EventStates through subscribe_states)
|
|
test_complete_future: asyncio.Future[bool] = loop.create_future()
|
|
test_result_future: asyncio.Future[bool] = loop.create_future()
|
|
|
|
def on_state(state: EntityState) -> None:
|
|
if not isinstance(state, Event):
|
|
return
|
|
|
|
if (
|
|
state.key == test_complete_entity.key
|
|
and state.event_type == "test_finished"
|
|
and not test_complete_future.done()
|
|
):
|
|
test_complete_future.set_result(True)
|
|
return
|
|
|
|
if state.key == test_result_entity.key and not test_result_future.done():
|
|
if state.event_type == "passed":
|
|
test_result_future.set_result(True)
|
|
elif state.event_type == "failed":
|
|
test_result_future.set_result(False)
|
|
|
|
client.subscribe_states(on_state)
|
|
|
|
# Test 1: Test set_timeout(0)
|
|
client.execute_service(test_set_timeout_service, {})
|
|
|
|
# Wait for first test completion
|
|
try:
|
|
await asyncio.wait_for(test_complete_future, timeout=5.0)
|
|
test1_passed = await asyncio.wait_for(test_result_future, timeout=1.0)
|
|
except asyncio.TimeoutError:
|
|
pytest.fail("Test set_timeout(0) did not complete within 5 seconds")
|
|
|
|
assert test1_passed is True, (
|
|
"set_timeout(0) FIFO test failed - items executed out of order"
|
|
)
|
|
|
|
# Reset futures for second test
|
|
test_complete_future = loop.create_future()
|
|
test_result_future = loop.create_future()
|
|
|
|
# Test 2: Test defer()
|
|
client.execute_service(test_defer_service, {})
|
|
|
|
# Wait for second test completion
|
|
try:
|
|
await asyncio.wait_for(test_complete_future, timeout=5.0)
|
|
test2_passed = await asyncio.wait_for(test_result_future, timeout=1.0)
|
|
except asyncio.TimeoutError:
|
|
pytest.fail("Test defer() did not complete within 5 seconds")
|
|
|
|
# Verify the test passed
|
|
assert test2_passed is True, (
|
|
"defer() FIFO test failed - items executed out of order"
|
|
)
|