mirror of
https://github.com/esphome/esphome.git
synced 2026-11-09 14:45:57 +00:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
"""Integration test for Home Assistant action lambdas returning strings and numbers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from .log_utils import LineWaiter
|
|
from .state_utils import assert_device_name, require_service
|
|
from .types import APIClientConnectedFactory, RunCompiledFunction
|
|
from .wait_utils import wait_or_fail
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_string_lambda(
|
|
yaml_config: str,
|
|
run_compiled: RunCompiledFunction,
|
|
api_client_connected: APIClientConnectedFactory,
|
|
) -> None:
|
|
"""Test Home Assistant action lambdas that return different types are sent as strings."""
|
|
waiter = LineWaiter()
|
|
|
|
# Run with log monitoring
|
|
async with (
|
|
run_compiled(yaml_config, line_callback=waiter.callback),
|
|
api_client_connected() as client,
|
|
):
|
|
# Verify device info
|
|
await assert_device_name(client, "api-string-lambda-test")
|
|
|
|
# List services to find our test services
|
|
_, services = await client.list_entities_services()
|
|
|
|
# Find all test services
|
|
string_service = require_service(services, "test_string_lambda")
|
|
|
|
int_service = require_service(services, "test_int_lambda")
|
|
|
|
float_service = require_service(services, "test_float_lambda")
|
|
|
|
char_ptr_service = require_service(services, "test_char_ptr_lambda")
|
|
|
|
# Execute all four services to test different lambda return types
|
|
await client.execute_service(
|
|
string_service, {"input_string": "STRING_FROM_LAMBDA"}
|
|
)
|
|
await client.execute_service(int_service, {"input_number": 42})
|
|
await client.execute_service(float_service, {"input_float": 3.14})
|
|
await client.execute_service(
|
|
char_ptr_service, {"input_number": 123, "input_string": "test_string"}
|
|
)
|
|
|
|
# Confirms the lambdas compiled and executed
|
|
await wait_or_fail(
|
|
waiter.wait_for_each(
|
|
"Service called with string: STRING_FROM_LAMBDA",
|
|
"Service called with int: 42",
|
|
"Service called with float: 3.14",
|
|
"Service called with number for char* test: 123",
|
|
timeout=10.0,
|
|
),
|
|
5.0,
|
|
"Not every lambda service logged its call",
|
|
logged=waiter.lines,
|
|
)
|