76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
from typing import List
|
|
|
|
|
|
def build_pytest_cmd(args: argparse.Namespace) -> List[str]:
|
|
cmd = [sys.executable, "-m", "pytest"]
|
|
|
|
# Пакеты тестов по умолчанию
|
|
test_paths = [
|
|
"uploader-bot/tests",
|
|
]
|
|
cmd += test_paths
|
|
|
|
# Маркеры
|
|
if args.mark:
|
|
cmd += ["-m", args.mark]
|
|
|
|
# Параллельность (pytest-xdist)
|
|
if args.parallel:
|
|
cmd += ["-n", str(args.parallel)]
|
|
|
|
# Verbose
|
|
if args.verbose:
|
|
cmd += ["-vv"]
|
|
|
|
# Coverage
|
|
if args.coverage:
|
|
# Требуется pytest-cov
|
|
cov_targets = args.cov_targets or ["app"]
|
|
cmd += ["--cov", *cov_targets, "--cov-report", "term-missing"]
|
|
if args.cov_xml:
|
|
cmd += ["--cov-report", "xml:coverage.xml"]
|
|
if args.cov_html:
|
|
cmd += ["--cov-report", "html:coverage_html"]
|
|
|
|
# Запрет пропуска warning как ошибок
|
|
if args.strict:
|
|
cmd += ["-W", "error"]
|
|
|
|
return cmd
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Unified test runner for uploader-bot")
|
|
parser.add_argument("--mark", help="pytest -m expression (e.g. 'performance or e2e')", default=None)
|
|
parser.add_argument("--parallel", "-n", type=int, help="Number of workers for pytest-xdist", default=None)
|
|
parser.add_argument("--verbose", "-v", action="store_true", help="Verbose output")
|
|
parser.add_argument("--coverage", action="store_true", help="Enable coverage via pytest-cov")
|
|
parser.add_argument("--cov-targets", nargs="*", help="Modules for coverage (default: app)", default=None)
|
|
parser.add_argument("--cov-xml", action="store_true", help="Emit coverage.xml")
|
|
parser.add_argument("--cov-html", action="store_true", help="Emit HTML coverage report to coverage_html/")
|
|
parser.add_argument("--strict", action="store_true", help="Treat warnings as errors")
|
|
parser.add_argument("--maxfail", type=int, default=1, help="Stop after N failures")
|
|
parser.add_argument("--no-capture", action="store_true", help="Disable output capture (-s)")
|
|
args = parser.parse_args()
|
|
|
|
cmd = build_pytest_cmd(args)
|
|
# Дополнительные флаги
|
|
cmd += ["--maxfail", str(args.maxfail)]
|
|
if args.no_capture:
|
|
cmd += ["-s"]
|
|
|
|
print("[run_tests] Command:", " ".join(cmd), flush=True)
|
|
env = os.environ.copy()
|
|
try:
|
|
return subprocess.call(cmd, env=env)
|
|
except KeyboardInterrupt:
|
|
return 130
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main()) |