72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""单测进程的联网护栏。导入即安装，联网失败即使被捕获，进程也必须失败。
|
|||
|
|
|
|||
|
|
完整隔离入口：python scripts/unit_test_guard.py scripts/test_batch3_units.py。
|
|||
|
|
入口在空临时目录运行测试，只继承系统环境，不读取仓库 .env 或生产参数。
|
|||
|
|
仅供测试使用，生产模块不得导入。
|
|||
|
|
"""
|
|||
|
|
import atexit
|
|||
|
|
import os
|
|||
|
|
from pathlib import Path
|
|||
|
|
import runpy
|
|||
|
|
import sys
|
|||
|
|
import tempfile
|
|||
|
|
import traceback
|
|||
|
|
|
|||
|
|
_violations = []
|
|||
|
|
_ROOT = str(Path(__file__).resolve().parents[1]) + os.sep
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _audit(event, args):
|
|||
|
|
if event in {"socket.connect", "socket.getaddrinfo", "socket.sendto", "socket.sendmsg"}:
|
|||
|
|
# 不输出连接地址或凭据，只保留调用位置。
|
|||
|
|
_violations.append((event, traceback.extract_stack()[:-1]))
|
|||
|
|
raise RuntimeError("单测禁止联网: " + event)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _finish():
|
|||
|
|
if _violations:
|
|||
|
|
print("\nTEST ISOLATION FAILED: 检测到未替换的外部调用", file=sys.stderr)
|
|||
|
|
try:
|
|||
|
|
for event, stack in _violations[:5]:
|
|||
|
|
print(event, file=sys.stderr)
|
|||
|
|
local = [frame for frame in stack if frame.filename.startswith(_ROOT)]
|
|||
|
|
traceback.print_list(local[-12:] or stack[-12:], file=sys.stderr)
|
|||
|
|
sys.stdout.flush()
|
|||
|
|
sys.stderr.flush()
|
|||
|
|
finally:
|
|||
|
|
os._exit(1) # atexit 里抛异常不会改变进程退出码。
|
|||
|
|
|
|||
|
|
|
|||
|
|
def run_suite(path):
|
|||
|
|
path = Path(path).resolve()
|
|||
|
|
sys.path.insert(0, str(path.parent))
|
|||
|
|
keep = {k: v for k, v in os.environ.items()
|
|||
|
|
if k in {"PATH", "SYSTEMROOT", "LANG", "LC_ALL"}}
|
|||
|
|
keep.update(TZ="Asia/Shanghai", PYTHONDONTWRITEBYTECODE="1")
|
|||
|
|
os.environ.clear()
|
|||
|
|
os.environ.update(keep)
|
|||
|
|
sys.dont_write_bytecode = True
|
|||
|
|
import time
|
|||
|
|
if hasattr(time, "tzset"):
|
|||
|
|
time.tzset()
|
|||
|
|
previous = Path.cwd()
|
|||
|
|
with tempfile.TemporaryDirectory(prefix="pms-unit-") as isolated:
|
|||
|
|
try:
|
|||
|
|
os.chdir(isolated)
|
|||
|
|
sys.argv = [str(path)]
|
|||
|
|
runpy.run_path(str(path), run_name="__main__")
|
|||
|
|
finally:
|
|||
|
|
os.chdir(previous)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
if len(sys.argv) != 2:
|
|||
|
|
raise SystemExit("用法: python scripts/unit_test_guard.py 测试文件")
|
|||
|
|
# 独立模块保存护栏状态，不与 runpy 临时使用的 __main__ 共用全局字典。
|
|||
|
|
from unit_test_guard import run_suite
|
|||
|
|
run_suite(sys.argv[1])
|
|||
|
|
else:
|
|||
|
|
sys.addaudithook(_audit)
|
|||
|
|
atexit.register(_finish)
|