sascha/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-10-31 07:03:55 +00:00
Code Issues Releases Wiki Activity

preen

Browse Source
This commit is contained in:
J. Nick Koston
2025-07-08 11:12:46 -06:00
parent d32db20aa0
commit 748604d374
1 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
tests/integration/test_runtime_stats.py
View File

@@ -54,35 +54,35 @@ async def test_runtime_stats(
component_name = match.group(1) component_name = match.group(1)
component_stats_found.add(component_name) component_stats_found.add(component_name)
async with run_compiled(yaml_config, line_callback=check_output): async with (
async with api_client_connected() as client: run_compiled(yaml_config, line_callback=check_output),
# Verify device is connected api_client_connected() as client,
device_info = await client.device_info() ):
assert device_info is not None # Verify device is connected
device_info = await client.device_info()
assert device_info is not None
# Wait for first "Total stats" log (should happen at 1s) # Wait for first "Total stats" log (should happen at 1s)
try: try:
await asyncio.wait_for(first_stats_future, timeout=5.0) await asyncio.wait_for(first_stats_future, timeout=5.0)
except asyncio.TimeoutError: except asyncio.TimeoutError:
pytest.fail("First 'Total stats' log not seen within 5 seconds") pytest.fail("First 'Total stats' log not seen within 5 seconds")
# Wait for second "Total stats" log (should happen at 2s) # Wait for second "Total stats" log (should happen at 2s)
try: try:
await asyncio.wait_for(second_stats_future, timeout=5.0) await asyncio.wait_for(second_stats_future, timeout=5.0)
except asyncio.TimeoutError: except asyncio.TimeoutError:
pytest.fail( pytest.fail(f"Second 'Total stats' log not seen. Total seen: {stats_count}")
f"Second 'Total stats' log not seen. Total seen: {stats_count}"
)
# Verify we got at least 2 stats logs # Verify we got at least 2 stats logs
assert stats_count >= 2, ( assert stats_count >= 2, (
f"Expected at least 2 'Total stats' logs, got {stats_count}" f"Expected at least 2 'Total stats' logs, got {stats_count}"
) )
# Verify we found stats for our components # Verify we found stats for our components
assert "template.sensor" in component_stats_found, ( assert "template.sensor" in component_stats_found, (
f"Expected template.sensor stats, found: {component_stats_found}" f"Expected template.sensor stats, found: {component_stats_found}"
) )
assert "template.switch" in component_stats_found, ( assert "template.switch" in component_stats_found, (
f"Expected template.switch stats, found: {component_stats_found}" f"Expected template.switch stats, found: {component_stats_found}"
) )