chore: init qmt_bridge repo (HTTP+WS bridge, MCP endpoint, docs, references)

2026-08-26 16:53:15 +08:00
commit 22a5b8ca04
210 changed files with 68176 additions and 0 deletions
@@ -0,0 +1,108 @@
+# coding: utf-8
+"""Run the full test suite from a single entry point.
+
+Groups tests by area and prints a clear per-group + total report. Optional
+live/API tests (need a running QMT + redis) are skipped by default.
+
+Usage:
+    python run_all_tests.py              # all offline tests (default)
+    python run_all_tests.py -v           # verbose
+    python run_all_tests.py --live       # also run live RPC tests (needs QMT running)
+    python run_all_tests.py --group signal_trader   # only one group
+    python run_all_tests.py --group backtest
+"""
+import argparse
+import os
+import subprocess
+import sys
+import time
+
+ROOT = os.path.dirname(os.path.abspath(__file__))
+SRC = os.path.join(ROOT, "src")
+
+# Test groups: (name, paths, requires_live)
+GROUPS = [
+    ("signal_trader", [os.path.join("tests", "bigqmt_signal_trader")], False),
+    ("backtest", [os.path.join("tests", "bigqmt_backtest")], False),
+]
+LIVE_GROUP = ("live_api", ["test_all_apis.py"], True)
+
+
+def run_group(name, paths, verbose):
+    """Run a pytest group, return (passed, failed, skipped, seconds)."""
+    cmd = [sys.executable, "-m", "pytest"] + paths + ["-q" if not verbose else "-v"]
+    t0 = time.time()
+    proc = subprocess.run(cmd, cwd=ROOT, capture_output=True, text=True)
+    elapsed = time.time() - t0
+    out = (proc.stdout or "") + (proc.stderr or "")
+    # Parse pytest summary like "290 passed in 8.5s" / "1 failed, 289 passed, 3 skipped"
+    passed = failed = skipped = 0
+    for line in out.splitlines():
+        line = line.strip()
+        if not any(tok in line for tok in ("passed", "failed", "skipped", "error")):
+            continue
+        for part in line.split(","):
+            words = part.strip().split()
+            if len(words) >= 2 and words[0].isdigit():
+                num = int(words[0])
+                if words[1].startswith("passed"):
+                    passed += num
+                elif words[1].startswith("failed") or words[1].startswith("error"):
+                    failed += num
+                elif words[1].startswith("skipped"):
+                    skipped += num
+    return passed, failed, skipped, elapsed, out, proc.returncode
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Run all bigqmt tests")
+    parser.add_argument("-v", "--verbose", action="store_true")
+    parser.add_argument("--live", action="store_true", help="also run live RPC tests (needs QMT)")
+    parser.add_argument("--group", help="run only this group (signal_trader/backtest/live_api)")
+    args = parser.parse_args()
+
+    groups = list(GROUPS)
+    if args.live:
+        groups.append(LIVE_GROUP)
+    if args.group:
+        groups = [g for g in groups if g[0] == args.group]
+        if not groups:
+            print("Unknown group: %s (available: %s)" % (args.group, ", ".join(g[0] for g in GROUPS + [LIVE_GROUP])))
+            return 1
+
+    print("=" * 70)
+    print("Big QMT Bridge - 全量测试")
+    print("=" * 70)
+
+    total_passed = total_failed = total_skipped = 0
+    total_time = 0.0
+    failed_groups = []
+    for name, paths, needs_live in groups:
+        print("\n--- %s ---" % name)
+        passed, failed, skipped, elapsed, out, rc = run_group(name, paths, args.verbose)
+        total_passed += passed
+        total_failed += failed
+        total_skipped += skipped
+        total_time += elapsed
+        status = "PASS" if failed == 0 and rc == 0 else "FAIL"
+        print("  %s: %d passed, %d failed, %d skipped (%.1fs)" % (status, passed, failed, skipped, elapsed))
+        if failed or rc != 0:
+            failed_groups.append(name)
+            if not args.verbose:
+                # print the failing part of the output for visibility
+                tail = "\n".join(out.splitlines()[-20:])
+                print(tail)
+
+    print("\n" + "=" * 70)
+    print("=== 汇总 ===")
+    print("通过 %d / 失败 %d / 跳过 %d / 总计 %d" % (total_passed, total_failed, total_skipped, total_passed + total_failed + total_skipped))
+    print("总耗时 %.1fs" % total_time)
+    if failed_groups:
+        print("失败分组: %s" % ", ".join(failed_groups))
+        return 1
+    print("全部通过 ✅")
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())