57 lines
1.4 KiB
Python
Executable File
57 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
if __name__ == "__main__":
|
|
default_cpu_count = os.cpu_count() or 2
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--run-podman", action="store_true", default=False, help="run podman tests (linux only)"
|
|
)
|
|
parser.add_argument(
|
|
"--num-processes",
|
|
action="store",
|
|
default=default_cpu_count,
|
|
help="number of processes to use for testing",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
# move cwd to the project root
|
|
os.chdir(Path(__file__).resolve().parents[1])
|
|
|
|
# unit tests
|
|
unit_test_args = [sys.executable, "-m", "pytest", "unit_test"]
|
|
|
|
if sys.platform.startswith("linux"):
|
|
# run the docker unit tests only on Linux
|
|
unit_test_args += ["--run-docker"]
|
|
|
|
if args.run_podman:
|
|
unit_test_args += ["--run-podman"]
|
|
|
|
subprocess.run(unit_test_args, check=True)
|
|
|
|
# integration tests
|
|
integration_test_args = [
|
|
sys.executable,
|
|
"-m",
|
|
"pytest",
|
|
f"--numprocesses={args.num_processes}",
|
|
"-x",
|
|
"--durations",
|
|
"0",
|
|
"--timeout=2400",
|
|
"test",
|
|
]
|
|
|
|
if sys.platform.startswith("linux") and args.run_podman:
|
|
integration_test_args += ["--run-podman"]
|
|
|
|
subprocess.run(integration_test_args, check=True)
|