mirror of https://github.com/pwndbg/pwndbg.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
811 B
Python
38 lines
811 B
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
|
|
import coverage
|
|
import pytest
|
|
|
|
use_pdb = os.environ.get("USE_PDB") == "1"
|
|
|
|
sys._pwndbg_unittest_run = True
|
|
|
|
CURRENT_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
test = os.environ["PWNDBG_LAUNCH_TEST"]
|
|
|
|
test = os.path.join(CURRENT_DIR, test)
|
|
|
|
args = [test, "-vvv", "-s", "--showlocals", "--color=yes"]
|
|
|
|
if use_pdb:
|
|
args.append("--pdb")
|
|
|
|
print(f"Launching pytest with args: {args}")
|
|
|
|
return_code = pytest.main(args)
|
|
|
|
if return_code != 0:
|
|
print("-" * 80)
|
|
print("If you want to debug tests locally, run ./tests.sh with the --pdb flag")
|
|
print("-" * 80)
|
|
|
|
# We must call this to ensure the code coverage file writes get flushed
|
|
# https://github.com/nedbat/coveragepy/issues/310
|
|
coverage._atexit()
|
|
sys.stdout.flush()
|
|
os._exit(return_code)
|