"""
app/test_runner.py - 在临时目录中真实执行 pytest 单元测试
"""
import os
import re
import sys
import tempfile
import subprocess


def run_python_tests(python_code: str, test_code: str) -> dict:
    """
    将业务代码写入 implementation.py，测试代码写入 test_implementation.py，
    在隔离的临时目录中用 pytest 执行，返回结构化测试结果。

    Returns:
        dict: {
            success: bool,
            passed: int,
            failed: int,
            errors: int,
            total: int,
            output: str   # pytest 完整输出
        }
    """
    with tempfile.TemporaryDirectory() as tmpdir:
        impl_path = os.path.join(tmpdir, "implementation.py")
        test_path = os.path.join(tmpdir, "test_implementation.py")
        conftest_path = os.path.join(tmpdir, "conftest.py")

        with open(impl_path, "w", encoding="utf-8") as f:
            f.write(python_code)

        with open(test_path, "w", encoding="utf-8") as f:
            f.write(test_code)

        # conftest.py 确保 tmpdir 在 sys.path 首位，解决模块导入问题
        with open(conftest_path, "w", encoding="utf-8") as f:
            f.write("import sys, os\nsys.path.insert(0, os.path.dirname(__file__))\n")

        try:
            proc = subprocess.run(
                [sys.executable, "-m", "pytest", "test_implementation.py",
                 "-v", "--tb=short", "--no-header", "--color=no"],
                cwd=tmpdir,
                capture_output=True,
                text=True,
                timeout=60,
                env={**os.environ, "PYTHONPATH": tmpdir},
            )
            output = proc.stdout
            if proc.stderr.strip():
                output += "\n--- stderr ---\n" + proc.stderr
        except subprocess.TimeoutExpired:
            return {
                "success": False,
                "passed": 0, "failed": 0, "errors": 1, "total": 0,
                "output": "❌ 测试执行超时（超过 60 秒）",
            }
        except FileNotFoundError:
            return {
                "success": False,
                "passed": 0, "failed": 0, "errors": 1, "total": 0,
                "output": "❌ 未找到 pytest，请确保已安装：pip install pytest",
            }

    passed = int(m.group(1)) if (m := re.search(r"(\d+) passed", output)) else 0
    failed = int(m.group(1)) if (m := re.search(r"(\d+) failed", output)) else 0
    errors = int(m.group(1)) if (m := re.search(r"(\d+) error", output)) else 0
    total  = passed + failed + errors
    success = (failed == 0 and errors == 0 and total > 0)

    return {
        "success": success,
        "passed": passed,
        "failed": failed,
        "errors": errors,
        "total": total,
        "output": output,
    }