1
0
mirror of https://github.com/esphome/esphome.git synced 2025-07-15 11:33:51 +01:00
Files
.devcontainer
.github
.vscode
docker
esphome
script
tests
component_tests
components
dashboard
integration
fixtures
README.md
__init__.py
conftest.py
const.py
test_api_conditional_memory.py
test_api_custom_services.py
test_api_message_size_batching.py
test_api_reboot_timeout.py
test_api_vv_logging.py
test_areas_and_devices.py
test_batch_delay_zero_rapid_transitions.py
test_device_id_in_state.py
test_duplicate_entities.py
test_entity_icon.py
test_host_mode_basic.py
test_host_mode_batch_delay.py
test_host_mode_empty_string_options.py
test_host_mode_entity_fields.py
test_host_mode_fan_preset.py
test_host_mode_many_entities.py
test_host_mode_many_entities_multiple_connections.py
test_host_mode_noise_encryption.py
test_host_mode_reconnect.py
test_host_mode_sensor.py
test_large_message_batching.py
test_legacy_area.py
test_light_calls.py
test_loop_disable_enable.py
test_scheduler_bulk_cleanup.py
test_scheduler_defer_cancel.py
test_scheduler_defer_cancel_regular.py
test_scheduler_defer_fifo_simple.py
test_scheduler_defer_stress.py
test_scheduler_heap_stress.py
test_scheduler_null_name.py
test_scheduler_rapid_cancellation.py
test_scheduler_recursive_timeout.py
test_scheduler_simultaneous_callbacks.py
test_scheduler_string_lifetime.py
test_scheduler_string_name_stress.py
test_scheduler_string_test.py
types.py
script
test_build_components
test_packages
unit_tests
.gitignore
README.md
custom.h
dummy_main.cpp
pnglogo.png
.clang-format
.clang-tidy
.clang-tidy.hash
.coveragerc
.dockerignore
.editorconfig
.flake8
.gitattributes
.gitignore
.pre-commit-config.yaml
.yamllint
CODEOWNERS
CODE_OF_CONDUCT.md
CONTRIBUTING.md
Doxyfile
LICENSE
MANIFEST.in
README.md
netlify.toml
platformio.ini
pyproject.toml
requirements.txt
requirements_dev.txt
requirements_test.txt
sdkconfig.defaults
esphome/tests/integration/test_host_mode_many_entities.py

73 lines
2.6 KiB
Python

"""Integration test for many entities to test API batching."""
from __future__ import annotations
import asyncio
from aioesphomeapi import EntityState, SensorState
import pytest
from .types import APIClientConnectedFactory, RunCompiledFunction
@pytest.mark.asyncio
async def test_host_mode_many_entities(
yaml_config: str,
run_compiled: RunCompiledFunction,
api_client_connected: APIClientConnectedFactory,
) -> None:
"""Test API batching with many entities of different types."""
# Write, compile and run the ESPHome device, then connect to API
loop = asyncio.get_running_loop()
async with run_compiled(yaml_config), api_client_connected() as client:
# Subscribe to state changes
states: dict[int, EntityState] = {}
sensor_count_future: asyncio.Future[int] = loop.create_future()
def on_state(state: EntityState) -> None:
states[state.key] = state
# Count sensor states specifically
sensor_states = [
s
for s in states.values()
if isinstance(s, SensorState) and isinstance(s.state, float)
]
# When we have received states from at least 50 sensors, resolve the future
if len(sensor_states) >= 50 and not sensor_count_future.done():
sensor_count_future.set_result(len(sensor_states))
client.subscribe_states(on_state)
# Wait for states from at least 50 sensors with timeout
try:
sensor_count = await asyncio.wait_for(sensor_count_future, timeout=10.0)
except asyncio.TimeoutError:
sensor_states = [
s
for s in states.values()
if isinstance(s, SensorState) and isinstance(s.state, float)
]
pytest.fail(
f"Did not receive states from at least 50 sensors within 10 seconds. "
f"Received {len(sensor_states)} sensor states out of {len(states)} total states"
)
# Verify we received a good number of entity states
assert len(states) >= 50, (
f"Expected at least 50 total states, got {len(states)}"
)
# Verify we have the expected sensor states
sensor_states = [
s
for s in states.values()
if isinstance(s, SensorState) and isinstance(s.state, float)
]
assert sensor_count >= 50, (
f"Expected at least 50 sensor states, got {sensor_count}"
)
assert len(sensor_states) >= 50, (
f"Expected at least 50 sensor states, got {len(sensor_states)}"
)