141 lines
5.6 KiB
Python
141 lines
5.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""k1_harness CLI.
|
|||
|
|
|
|||
|
|
python -m k1_harness selftest
|
|||
|
|
python -m k1_harness run --suite app --transport com --port COM5
|
|||
|
|
python -m k1_harness run --suite app --transport ble
|
|||
|
|
python -m k1_harness run --suite ble
|
|||
|
|
python -m k1_harness run --suite keys,packs,ui --device AT32F403AC
|
|||
|
|
python -m k1_harness run --suite all --profile nightly
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
from .paths import default_report_dir
|
|||
|
|
from .report import HarnessReport
|
|||
|
|
from .suites import SUITES, ALL_ORDER
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class Ctx:
|
|||
|
|
transport: str = "ble"
|
|||
|
|
port: str = ""
|
|||
|
|
baud: int = 115200
|
|||
|
|
device: str = "AT32F403AC"
|
|||
|
|
allow_poweroff: bool = False
|
|||
|
|
dump_png: str = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _run_selftest() -> int:
|
|||
|
|
report = HarnessReport(title="K1 harness selftest（无硬件）")
|
|||
|
|
|
|||
|
|
# 1) 协议帧编解码自检（复用现有 selftest 逻辑，零硬件）
|
|||
|
|
import io
|
|||
|
|
import contextlib
|
|||
|
|
import test_protocol_app_sim as appsim
|
|||
|
|
|
|||
|
|
buf = io.StringIO()
|
|||
|
|
with contextlib.redirect_stdout(buf):
|
|||
|
|
code = appsim.selftest()
|
|||
|
|
for line in buf.getvalue().splitlines():
|
|||
|
|
line = line.strip()
|
|||
|
|
if line.startswith("[") and "]" in line:
|
|||
|
|
status = line[1:line.index("]")].strip()
|
|||
|
|
name = line[line.index("]") + 1:].split("|")[0].strip()
|
|||
|
|
report.add("selftest", name, status if status in ("PASS", "FAIL", "SKIP", "SENT") else "FAIL")
|
|||
|
|
report.add("selftest", "app_sim selftest 退出码", "PASS" if code == 0 else "FAIL", f"exit={code}")
|
|||
|
|
|
|||
|
|
# 2) pitch 分析器离线自检：喂一条合法/非法 PITCH 行
|
|||
|
|
from .transports.rtt_session import analyze_lines
|
|||
|
|
good = ("[PITCH] on ch8(bass) 52->40 d=-12 deg=3 chord=8 xp=2 xf=0 fl=0x03 FIX\n"
|
|||
|
|
"[PITCH] on ch1(chord) 64->64 d=0 deg=1 chord=1 xp=0 xf=0 fl=0x00 PASS\n")
|
|||
|
|
res = analyze_lines(good.splitlines())
|
|||
|
|
report.add("selftest", "pitch 分析器(合法样本)", "PASS" if res.fail == 0 and res.ok >= 2 else "FAIL",
|
|||
|
|
f"ok={res.ok} fail={res.fail}")
|
|||
|
|
bad = "[PITCH] on ch8(bass) 52->30 d=-22 deg=3 chord=8 xp=2 xf=0 fl=0x03 FIX\n"
|
|||
|
|
res2 = analyze_lines(bad.splitlines())
|
|||
|
|
report.add("selftest", "pitch 分析器(非法样本检出)", "PASS" if res2.fail >= 1 else "FAIL",
|
|||
|
|
f"fail={res2.fail}")
|
|||
|
|
|
|||
|
|
# 3) 报告落盘自检
|
|||
|
|
out = os.path.join(default_report_dir(), "k1_selftest.md")
|
|||
|
|
report.write_markdown(out)
|
|||
|
|
report.print_summary()
|
|||
|
|
return report.exit_code()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _run_suites(args) -> int:
|
|||
|
|
names = []
|
|||
|
|
for part in args.suite.split(","):
|
|||
|
|
part = part.strip().lower()
|
|||
|
|
if part == "all":
|
|||
|
|
names.extend(n for n in ALL_ORDER if n not in names)
|
|||
|
|
elif part in SUITES:
|
|||
|
|
if part not in names:
|
|||
|
|
names.append(part)
|
|||
|
|
else:
|
|||
|
|
print(f"未知套件: {part}（可选: {', '.join(SUITES)} / all）", file=sys.stderr)
|
|||
|
|
return 2
|
|||
|
|
|
|||
|
|
ctx = Ctx(
|
|||
|
|
transport=args.transport,
|
|||
|
|
port=args.port or "",
|
|||
|
|
baud=args.baud,
|
|||
|
|
device=args.device,
|
|||
|
|
allow_poweroff=args.allow_poweroff or args.profile == "destructive",
|
|||
|
|
dump_png=args.dump_png or "",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
report = HarnessReport(title="K1 HIL 自动化测试报告")
|
|||
|
|
report.notes.append(f"套件: {', '.join(names)}")
|
|||
|
|
report.notes.append(f"transport={ctx.transport} device={ctx.device} profile={args.profile}")
|
|||
|
|
|
|||
|
|
for name in names:
|
|||
|
|
mod, needs = SUITES[name]
|
|||
|
|
if "transport" in needs and ctx.transport == "com" and not ctx.port:
|
|||
|
|
report.add(name, "前置条件", "SKIP", "transport=com 需 --port；已跳过该套件")
|
|||
|
|
continue
|
|||
|
|
print(f"\n===== 套件 {name} =====", flush=True)
|
|||
|
|
try:
|
|||
|
|
mod.run(report, ctx)
|
|||
|
|
except SystemExit as exc:
|
|||
|
|
report.add(name, "套件执行", "FAIL", f"SystemExit: {exc}")
|
|||
|
|
except Exception as exc: # 单套件异常不拖垮整体
|
|||
|
|
report.add(name, "套件执行", "FAIL", f"{type(exc).__name__}: {exc}")
|
|||
|
|
|
|||
|
|
report.print_summary()
|
|||
|
|
ts = report.started.strftime("%Y%m%d_%H%M%S")
|
|||
|
|
out = args.report or os.path.join(default_report_dir(), f"k1_hil_{ts}.md")
|
|||
|
|
report.write_markdown(out)
|
|||
|
|
return report.exit_code()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> None:
|
|||
|
|
ap = argparse.ArgumentParser(prog="k1_harness", description="K1 吉他统一 HIL 自动化测试")
|
|||
|
|
sub = ap.add_subparsers(dest="cmd", required=True)
|
|||
|
|
|
|||
|
|
sub.add_parser("selftest", help="无硬件自检（帧编解码 + 分析器 + 报告）")
|
|||
|
|
|
|||
|
|
runp = sub.add_parser("run", help="跑测试套件")
|
|||
|
|
runp.add_argument("--suite", required=True,
|
|||
|
|
help="逗号分隔: app,ble,keys,keys1617,packs,ui 或 all")
|
|||
|
|
runp.add_argument("--transport", choices=("com", "ble"), default="ble",
|
|||
|
|
help="app 套件通道（默认 ble）")
|
|||
|
|
runp.add_argument("--port", default="", help="COM 端口（transport=com 时必填）")
|
|||
|
|
runp.add_argument("--baud", type=int, default=115200)
|
|||
|
|
runp.add_argument("--device", default="AT32F403AC", help="J-Link 器件名")
|
|||
|
|
runp.add_argument("--profile", choices=("smoke", "nightly", "destructive"), default="nightly")
|
|||
|
|
runp.add_argument("--allow-poweroff", action="store_true", help="允许 05 00 关机用例")
|
|||
|
|
runp.add_argument("--dump-png", default="", help="ui 套件截屏输出路径（需 DEBUG_LCD_DUMP 固件）")
|
|||
|
|
runp.add_argument("--report", default="", help="Markdown 报告输出路径")
|
|||
|
|
|
|||
|
|
args = ap.parse_args()
|
|||
|
|
if args.cmd == "selftest":
|
|||
|
|
raise SystemExit(_run_selftest())
|
|||
|
|
raise SystemExit(_run_suites(args))
|