81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
"""
|
||
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,
|
||
}
|