109 lines
3.0 KiB
Python
Executable File
109 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import configparser
|
|
import contextlib
|
|
import glob
|
|
import io
|
|
import itertools
|
|
import multiprocessing
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
|
|
def run_tests(src, test, fail):
|
|
stderr = io.StringIO()
|
|
stdout = io.StringIO()
|
|
with contextlib.redirect_stderr(stderr):
|
|
with contextlib.redirect_stdout(stdout):
|
|
e = pytest.main(
|
|
[
|
|
"-qq",
|
|
"--disable-pytest-warnings",
|
|
"--cov",
|
|
src.replace(".py", "").replace("/", "."),
|
|
"--cov-fail-under",
|
|
"100",
|
|
"--cov-report",
|
|
"term-missing:skip-covered",
|
|
"-o",
|
|
"faulthandler_timeout=0",
|
|
test,
|
|
]
|
|
)
|
|
|
|
if e == 0:
|
|
if fail:
|
|
print(
|
|
"FAIL DUE TO UNEXPECTED SUCCESS:",
|
|
src,
|
|
"Please remove this file from setup.cfg tool:individual_coverage/exclude.",
|
|
)
|
|
e = 42
|
|
else:
|
|
print(".")
|
|
else:
|
|
if fail:
|
|
print("Ignoring allowed fail:", src)
|
|
e = 0
|
|
else:
|
|
cov = [
|
|
line
|
|
for line in stdout.getvalue().split("\n")
|
|
if (src in line) or ("was never imported" in line)
|
|
]
|
|
if len(cov) == 1:
|
|
print("FAIL:", cov[0])
|
|
else:
|
|
print("FAIL:", src, test, stdout.getvalue(), stdout.getvalue())
|
|
print(stderr.getvalue())
|
|
print(stdout.getvalue())
|
|
|
|
sys.exit(e)
|
|
|
|
|
|
def start_pytest(src, test, fail):
|
|
# run pytest in a new process, otherwise imports and modules might conflict
|
|
proc = multiprocessing.Process(target=run_tests, args=(src, test, fail))
|
|
proc.start()
|
|
proc.join()
|
|
return (src, test, proc.exitcode)
|
|
|
|
|
|
def main():
|
|
c = configparser.ConfigParser()
|
|
c.read("setup.cfg")
|
|
fs = c["tool:individual_coverage"]["exclude"].strip().split("\n")
|
|
no_individual_cov = [f.strip() for f in fs]
|
|
|
|
excluded = [
|
|
"mitmproxy/contrib/",
|
|
"mitmproxy/test/",
|
|
"mitmproxy/tools/",
|
|
"mitmproxy/platform/",
|
|
]
|
|
src_files = glob.glob("mitmproxy/**/*.py", recursive=True)
|
|
src_files = [f for f in src_files if os.path.basename(f) != "__init__.py"]
|
|
src_files = [
|
|
f for f in src_files if not any(os.path.normpath(p) in f for p in excluded)
|
|
]
|
|
if len(sys.argv) > 1:
|
|
src_files = [f for f in src_files if sys.argv[1] in str(f)]
|
|
|
|
ps = []
|
|
for src in sorted(src_files):
|
|
test = os.path.join(
|
|
"test", os.path.dirname(src), "test_" + os.path.basename(src)
|
|
)
|
|
if os.path.isfile(test):
|
|
ps.append((src, test, src in no_individual_cov))
|
|
|
|
result = list(itertools.starmap(start_pytest, ps))
|
|
|
|
if any(e != 0 for _, _, e in result):
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|