116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
|
"""
|
||
|
|
Pytest wrapper for McRogueFace test scripts.
|
||
|
|
|
||
|
|
This file discovers and runs all McRogueFace test scripts in unit/, integration/,
|
||
|
|
and regression/ directories via subprocess.
|
||
|
|
|
||
|
|
Usage:
|
||
|
|
pytest tests/test_mcrogueface.py -q # Quiet output
|
||
|
|
pytest tests/test_mcrogueface.py -k "bsp" # Filter by name
|
||
|
|
pytest tests/test_mcrogueface.py --mcrf-timeout=30 # Custom timeout
|
||
|
|
pytest tests/test_mcrogueface.py -x # Stop on first failure
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import pytest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
# Paths
|
||
|
|
TESTS_DIR = Path(__file__).parent
|
||
|
|
BUILD_DIR = TESTS_DIR.parent / "build"
|
||
|
|
LIB_DIR = TESTS_DIR.parent / "__lib"
|
||
|
|
MCROGUEFACE = BUILD_DIR / "mcrogueface"
|
||
|
|
|
||
|
|
# Test directories
|
||
|
|
TEST_DIRS = ['unit', 'integration', 'regression']
|
||
|
|
|
||
|
|
# Default timeout
|
||
|
|
DEFAULT_TIMEOUT = 10
|
||
|
|
|
||
|
|
|
||
|
|
def discover_tests():
|
||
|
|
"""Find all test scripts in test directories."""
|
||
|
|
tests = []
|
||
|
|
for test_dir in TEST_DIRS:
|
||
|
|
dir_path = TESTS_DIR / test_dir
|
||
|
|
if dir_path.exists():
|
||
|
|
for test_file in sorted(dir_path.glob("*.py")):
|
||
|
|
if test_file.name != '__init__.py':
|
||
|
|
rel_path = f"{test_dir}/{test_file.name}"
|
||
|
|
tests.append(rel_path)
|
||
|
|
return tests
|
||
|
|
|
||
|
|
|
||
|
|
def get_env():
|
||
|
|
"""Get environment with LD_LIBRARY_PATH set."""
|
||
|
|
env = os.environ.copy()
|
||
|
|
existing_ld = env.get('LD_LIBRARY_PATH', '')
|
||
|
|
env['LD_LIBRARY_PATH'] = f"{LIB_DIR}:{existing_ld}" if existing_ld else str(LIB_DIR)
|
||
|
|
return env
|
||
|
|
|
||
|
|
|
||
|
|
def run_mcrf_test(script_path, timeout=DEFAULT_TIMEOUT):
|
||
|
|
"""Run a McRogueFace test script and return (passed, output)."""
|
||
|
|
full_path = TESTS_DIR / script_path
|
||
|
|
env = get_env()
|
||
|
|
|
||
|
|
try:
|
||
|
|
result = subprocess.run(
|
||
|
|
[str(MCROGUEFACE), '--headless', '--exec', str(full_path)],
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
timeout=timeout,
|
||
|
|
cwd=str(BUILD_DIR),
|
||
|
|
env=env
|
||
|
|
)
|
||
|
|
|
||
|
|
output = result.stdout + result.stderr
|
||
|
|
passed = result.returncode == 0
|
||
|
|
|
||
|
|
# Check for PASS/FAIL in output
|
||
|
|
if 'FAIL' in output and 'PASS' not in output.split('FAIL')[-1]:
|
||
|
|
passed = False
|
||
|
|
|
||
|
|
return passed, output
|
||
|
|
|
||
|
|
except subprocess.TimeoutExpired:
|
||
|
|
return False, "TIMEOUT"
|
||
|
|
except Exception as e:
|
||
|
|
return False, str(e)
|
||
|
|
|
||
|
|
|
||
|
|
# Discover tests at module load time
|
||
|
|
ALL_TESTS = discover_tests()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mcrf_timeout(request):
|
||
|
|
"""Get timeout from command line or default."""
|
||
|
|
return request.config.getoption("--mcrf-timeout", default=DEFAULT_TIMEOUT)
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_addoption(parser):
|
||
|
|
"""Add --mcrf-timeout option."""
|
||
|
|
try:
|
||
|
|
parser.addoption(
|
||
|
|
"--mcrf-timeout",
|
||
|
|
action="store",
|
||
|
|
default=DEFAULT_TIMEOUT,
|
||
|
|
type=int,
|
||
|
|
help="Timeout in seconds for McRogueFace tests"
|
||
|
|
)
|
||
|
|
except ValueError:
|
||
|
|
# Option already added
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("script_path", ALL_TESTS, ids=lambda x: x.replace('/', '::'))
|
||
|
|
def test_mcrogueface_script(script_path, request):
|
||
|
|
"""Run a McRogueFace test script."""
|
||
|
|
timeout = request.config.getoption("--mcrf-timeout", default=DEFAULT_TIMEOUT)
|
||
|
|
passed, output = run_mcrf_test(script_path, timeout=timeout)
|
||
|
|
|
||
|
|
if not passed:
|
||
|
|
# Show last 15 lines of output on failure
|
||
|
|
lines = output.strip().split('\n')[-15:]
|
||
|
|
pytest.fail('\n'.join(lines))
|