Function main

pythonbuild/testdist.py:81–137

test-distribution.py functionality.

(pbs_source_dir: Path, raw_args: list[str])

def main(pbs_source_dir: Path, raw_args: list[str]) -> int:
    """test-distribution.py functionality."""

    parser = argparse.ArgumentParser()

    parser.add_argument(
        "--stdlib",
        action="store_true",
        help="Run the stdlib test harness",
    )
    parser.add_argument(
        "dist",
        nargs=1,
        help="Path to distribution to test",
    )
    parser.add_argument(
        "harness_args",
        nargs=argparse.REMAINDER,
        help="Raw arguments to pass to Python's test harness",
    )

    args = parser.parse_args(raw_args)

    dist_path_raw = Path(args.dist[0])

    td = None
    try:
        if dist_path_raw.is_file():
            td = tempfile.TemporaryDirectory()
            dist_path = extract_python_archive(dist_path_raw, Path(td.name))
        else:
            dist_path = dist_path_raw

        python_json = dist_path / "PYTHON.json"

        with python_json.open("r", encoding="utf-8") as fh:
            python_info = json.load(fh)

        codes = []

        codes.append(run_custom_unittests(pbs_source_dir, dist_path, python_info))

        if args.stdlib:
            codes.append(run_stdlib_tests(dist_path, python_info, args.harness_args))

        if len(codes) == 0:
            print("no tests run")
            return 1

        if any(code != 0 for code in codes):
            return 1

        return 0

    finally:
        if td:
            td.cleanup()