This commit is contained in:
J. Nick Koston 2025-07-08 11:12:46 -06:00
parent d32db20aa0
commit 748604d374
No known key found for this signature in database

View File

@ -54,8 +54,10 @@ async def test_runtime_stats(
component_name = match.group(1)
component_stats_found.add(component_name)
async with run_compiled(yaml_config, line_callback=check_output):
async with api_client_connected() as client:
async with (
run_compiled(yaml_config, line_callback=check_output),
api_client_connected() as client,
):
# Verify device is connected
device_info = await client.device_info()
assert device_info is not None
@ -70,9 +72,7 @@ async def test_runtime_stats(
try:
await asyncio.wait_for(second_stats_future, timeout=5.0)
except asyncio.TimeoutError:
pytest.fail(
f"Second 'Total stats' log not seen. Total seen: {stats_count}"
)
pytest.fail(f"Second 'Total stats' log not seen. Total seen: {stats_count}")
# Verify we got at least 2 stats logs
assert stats_count >= 2, (