mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
firestar5683
d734bebcb2
Add control trace and platform qualification tools that exercise explicit safety modes and vehicle coverage.
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Run host tests with disposable Params and a private messaging namespace."""
|
|
|
|
import ast
|
|
import os
|
|
from pathlib import Path
|
|
import platform
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
RUNNER = ROOT / 'tools/test_runner.py'
|
|
|
|
|
|
def pytest_files(root=ROOT):
|
|
"""Select custom files containing pytest functions or plain Test classes."""
|
|
files = []
|
|
for path in sorted((root / 'openpilot/starpilot').rglob('test_*.py')):
|
|
module = ast.parse(path.read_text(), filename=str(path))
|
|
if any((isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)) and node.name.startswith('test_')) or
|
|
(isinstance(node, ast.ClassDef) and node.name.startswith('Test') and not node.bases)
|
|
for node in module.body):
|
|
files.append(path.relative_to(root).as_posix())
|
|
return files
|
|
|
|
|
|
def main(arguments=None):
|
|
arguments = sys.argv[1:] if arguments is None else arguments
|
|
pytest_mode = bool(arguments and arguments[0] == '--pytest')
|
|
if pytest_mode:
|
|
files = pytest_files()
|
|
if not files:
|
|
print('No StarPilot pytest functions were discovered.', file=sys.stderr)
|
|
return 5
|
|
shared_memory = '/tmp' if platform.system() == 'Darwin' else '/dev/shm'
|
|
with tempfile.TemporaryDirectory(prefix='starpilot-host-params-') as params, \
|
|
tempfile.TemporaryDirectory(prefix='msgq_starpilot-host-', dir=shared_memory) as messaging:
|
|
# Set these before imports in the child. UIState constructs Params and
|
|
# subscribers at import time; SCALE avoids querying a desktop monitor.
|
|
environment = dict(os.environ, PARAMS_ROOT=params, SCALE='1',
|
|
OPENPILOT_PREFIX=Path(messaging).name.removeprefix('msgq_'))
|
|
command = ([sys.executable, '-m', 'pytest', '-q', *arguments[1:], *files] if pytest_mode
|
|
else [sys.executable, str(RUNNER), *arguments])
|
|
result = subprocess.run(command, cwd=ROOT, env=environment, check=False)
|
|
return result.returncode if result.returncode >= 0 else 128 - result.returncode
|
|
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main())
|