Max Gramser
49 lines
2.3 KiB
Python
49 lines
2.3 KiB
Python
"""Run the firmware's render test through its encrypted API; no HA actions."""
|
|
import argparse
|
|
import asyncio
|
|
import re
|
|
from pathlib import Path
|
|
import yaml
|
|
from aioesphomeapi import APIClient, LogLevel
|
|
|
|
async def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--host', default='cyd-2432s028.local')
|
|
parser.add_argument('--name', default='cyd-2432s028', help='Verwachte DEVICE_NAME')
|
|
parser.add_argument('--secrets', type=Path, default=Path(__file__).resolve().parents[1] / 'secrets.yaml')
|
|
args = parser.parse_args()
|
|
secrets = yaml.safe_load(args.secrets.read_text())
|
|
client = APIClient(args.host, 6053, noise_psk=secrets['api_encryption_key'], client_info='CYD UI diagnostics', expected_name=args.name)
|
|
await client.connect(login=True)
|
|
try:
|
|
info = await client.device_info()
|
|
print(f'Connected: {info.name}, ESPHome {info.esphome_version}, compiled {info.compilation_time}', flush=True)
|
|
_, services = await client.list_entities_services()
|
|
test = next(s for s in services if s.name == 'ui_self_test')
|
|
done = asyncio.Event()
|
|
failures = []
|
|
passes = []
|
|
frames = []
|
|
def log(msg):
|
|
text = msg.message.decode(errors='replace') if isinstance(msg.message, bytes) else msg.message
|
|
if any(tag in text for tag in ('ui_test', 'UI_TEST', '[health', 'took a long')):
|
|
print(text, flush=True)
|
|
if 'page2=PASS' in text or 'page_check=PASS' in text:
|
|
passes.append(text)
|
|
match = re.search(r'frames=(\d+)', text)
|
|
if match: frames.append(int(match[1]))
|
|
if 'FAIL' in text: failures.append(text)
|
|
if 'UI_TEST COMPLETE' in text: done.set()
|
|
client.subscribe_logs(log, log_level=LogLevel.LOG_LEVEL_DEBUG, dump_config=False)
|
|
await client.execute_service(test, {})
|
|
await asyncio.wait_for(done.wait(), timeout=45)
|
|
assert len(passes) == 10, f'Expected 10 page checks, got {len(passes)}'
|
|
assert not failures, failures
|
|
assert len(frames) == 10 and frames[-1] - frames[0] >= 40, f'Render callbacks stalled: {frames}'
|
|
print('PASS: 10 page checks and 50 overlay render cycles completed', flush=True)
|
|
finally:
|
|
await client.disconnect()
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|