# 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())