@harmonyos-arkts/d2h 0.1.11 → 0.1.12

package/skills/d2h-hypium-forge/scripts/run_parallel.py

@@ -1,121 +1,121 @@
-#!/usr/bin/env python3
-"""
-run_parallel.py — 多设备并发直跑器（N 设备 × 1 线程/设备）。
-
-用法:
-  python3 run_parallel.py --dir testcases --devices 127.0.0.1:5555,127.0.0.1:5556,127.0.0.1:5557
-  可选: --filter-prefix F01（只跑某前缀）/ --results results.json
-
-设计: 每设备一个常驻线程 + 独立 UiDriver（hypium 驱动线程安全边界内），
-     用例按 round-robin 分发；单例失败不中断；汇总 per-case 耗时与总时长。
-"""
-import argparse
-import importlib.util
-import json
-import os
-import queue
-import sys
-import threading
-import time
-import traceback
-
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from hypium import UiDriver
-
-
-def load_case(path):
-    spec = importlib.util.spec_from_file_location(os.path.basename(path)[:-3], path)
-    mod = importlib.util.module_from_spec(spec)
-    spec.loader.exec_module(mod)
-    return mod
-
-
-def run_case_on(d, mod_path, name):
-    mod = load_case(mod_path)
-    t0 = time.time()
-    try:
-        cls = None
-        for attr in dir(mod):
-            obj = getattr(mod, attr)
-            if isinstance(obj, type) and hasattr(obj, 'process') and attr != 'TestCase':
-                cls = obj
-                break
-        inst = object.__new__(cls)
-        inst.driver = d
-        inst.device1 = None
-        inst.TAG = name
-        if hasattr(cls, 'setup'):
-            cls.setup(inst)
-        try:
-            cls.process(inst)
-        finally:
-            if hasattr(cls, 'teardown'):
-                try:
-                    cls.teardown(inst)
-                except Exception:
-                    pass
-        return 'PASS', '', round(time.time() - t0, 1)
-    except Exception as e:
-        tb = traceback.format_exc().strip().split('\n')
-        return 'FAIL', f'{e} | {tb[-1][:100] if len(tb) > 1 else ""}', round(time.time() - t0, 1)
-
-
-def worker(dev_sn, case_q, results, lock):
-    try:
-        d = UiDriver.connect(device_sn=dev_sn)
-    except Exception as e:
-        with lock:
-            results.append({'case': f'WORKER-{dev_sn}', 'status': 'ERROR', 'msg': str(e)[:100], 'dur': 0})
-        return
-    while True:
-        try:
-            name, path = case_q.get_nowait()
-        except queue.Empty:
-            return
-        status, msg, dur = run_case_on(d, path, name)
-        with lock:
-            results.append({'case': name, 'status': status, 'msg': msg, 'dur': dur, 'device': dev_sn})
-            print(f'[{dev_sn}] {name}: {status} ({dur}s) {msg[:60]}', flush=True)
-        case_q.task_done()
-        time.sleep(2)
-
-
-def main():
-    ap = argparse.ArgumentParser()
-    ap.add_argument('--dir', default='testcases')
-    ap.add_argument('--devices', required=True, help='逗号分隔设备 SN')
-    ap.add_argument('--filter-prefix', default=None)
-    ap.add_argument('--results', default='results_parallel.json')
-    args = ap.parse_args()
-
-    sys.path.insert(0, os.path.abspath(args.dir))
-    files = sorted(f for f in os.listdir(args.dir) if f.endswith('.py') and not f.startswith('_'))
-    if args.filter_prefix:
-        files = [f for f in files if f.startswith(args.filter_prefix)]
-    if not files:
-        print('无用例'); sys.exit(1)
-
-    case_q = queue.Queue()
-    for f in files:
-        case_q.put((f[:-3], os.path.join(args.dir, f)))
-
-    devices = [d.strip() for d in args.devices.split(',') if d.strip()]
-    results, lock = [], threading.Lock()
-    t0 = time.time()
-    threads = [threading.Thread(target=worker, args=(dsn, case_q, results, lock)) for dsn in devices]
-    for t in threads:
-        t.start()
-    for t in threads:
-        t.join()
-    total = round(time.time() - t0, 1)
-
-    p = sum(1 for r in results if r['status'] == 'PASS')
-    f_ = sum(1 for r in results if r['status'] == 'FAIL')
-    durs = [r['dur'] for r in results if r['status'] in ('PASS', 'FAIL')]
-    print(f'\n══════ 并发汇总: PASS={p} FAIL={f_} / {len(results)} | 总时长 {total}s '
-          f'| 单例均值 {sum(durs)/len(durs):.1f}s | {len(devices)} 设备并发 ══════')
-    json.dump(results, open(args.results, 'w', encoding='utf-8'), ensure_ascii=False, indent=1)
-
-
-if __name__ == '__main__':
-    main()
+#!/usr/bin/env python3
+"""
+run_parallel.py — 多设备并发直跑器（N 设备 × 1 线程/设备）。
+
+用法:
+  python3 run_parallel.py --dir testcases --devices 127.0.0.1:5555,127.0.0.1:5556,127.0.0.1:5557
+  可选: --filter-prefix F01（只跑某前缀）/ --results results.json
+
+设计: 每设备一个常驻线程 + 独立 UiDriver（hypium 驱动线程安全边界内），
+     用例按 round-robin 分发；单例失败不中断；汇总 per-case 耗时与总时长。
+"""
+import argparse
+import importlib.util
+import json
+import os
+import queue
+import sys
+import threading
+import time
+import traceback
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from hypium import UiDriver
+
+
+def load_case(path):
+    spec = importlib.util.spec_from_file_location(os.path.basename(path)[:-3], path)
+    mod = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(mod)
+    return mod
+
+
+def run_case_on(d, mod_path, name):
+    mod = load_case(mod_path)
+    t0 = time.time()
+    try:
+        cls = None
+        for attr in dir(mod):
+            obj = getattr(mod, attr)
+            if isinstance(obj, type) and hasattr(obj, 'process') and attr != 'TestCase':
+                cls = obj
+                break
+        inst = object.__new__(cls)
+        inst.driver = d
+        inst.device1 = None
+        inst.TAG = name
+        if hasattr(cls, 'setup'):
+            cls.setup(inst)
+        try:
+            cls.process(inst)
+        finally:
+            if hasattr(cls, 'teardown'):
+                try:
+                    cls.teardown(inst)
+                except Exception:
+                    pass
+        return 'PASS', '', round(time.time() - t0, 1)
+    except Exception as e:
+        tb = traceback.format_exc().strip().split('\n')
+        return 'FAIL', f'{e} | {tb[-1][:100] if len(tb) > 1 else ""}', round(time.time() - t0, 1)
+
+
+def worker(dev_sn, case_q, results, lock):
+    try:
+        d = UiDriver.connect(device_sn=dev_sn)
+    except Exception as e:
+        with lock:
+            results.append({'case': f'WORKER-{dev_sn}', 'status': 'ERROR', 'msg': str(e)[:100], 'dur': 0})
+        return
+    while True:
+        try:
+            name, path = case_q.get_nowait()
+        except queue.Empty:
+            return
+        status, msg, dur = run_case_on(d, path, name)
+        with lock:
+            results.append({'case': name, 'status': status, 'msg': msg, 'dur': dur, 'device': dev_sn})
+            print(f'[{dev_sn}] {name}: {status} ({dur}s) {msg[:60]}', flush=True)
+        case_q.task_done()
+        time.sleep(2)
+
+
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument('--dir', default='testcases')
+    ap.add_argument('--devices', required=True, help='逗号分隔设备 SN')
+    ap.add_argument('--filter-prefix', default=None)
+    ap.add_argument('--results', default='results_parallel.json')
+    args = ap.parse_args()
+
+    sys.path.insert(0, os.path.abspath(args.dir))
+    files = sorted(f for f in os.listdir(args.dir) if f.endswith('.py') and not f.startswith('_'))
+    if args.filter_prefix:
+        files = [f for f in files if f.startswith(args.filter_prefix)]
+    if not files:
+        print('无用例'); sys.exit(1)
+
+    case_q = queue.Queue()
+    for f in files:
+        case_q.put((f[:-3], os.path.join(args.dir, f)))
+
+    devices = [d.strip() for d in args.devices.split(',') if d.strip()]
+    results, lock = [], threading.Lock()
+    t0 = time.time()
+    threads = [threading.Thread(target=worker, args=(dsn, case_q, results, lock)) for dsn in devices]
+    for t in threads:
+        t.start()
+    for t in threads:
+        t.join()
+    total = round(time.time() - t0, 1)
+
+    p = sum(1 for r in results if r['status'] == 'PASS')
+    f_ = sum(1 for r in results if r['status'] == 'FAIL')
+    durs = [r['dur'] for r in results if r['status'] in ('PASS', 'FAIL')]
+    print(f'\n══════ 并发汇总: PASS={p} FAIL={f_} / {len(results)} | 总时长 {total}s '
+          f'| 单例均值 {sum(durs)/len(durs):.1f}s | {len(devices)} 设备并发 ══════')
+    json.dump(results, open(args.results, 'w', encoding='utf-8'), ensure_ascii=False, indent=1)
+
+
+if __name__ == '__main__':
+    main()