pwndbg/pytests_launcher.py

28 lines
608 B
Python
Raw Normal View History

import os
import pytest
import sys
print(sys.argv)
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)
# If you want to debug tests locally, add '--pdb' here
args = [test, '-vvv', '-s', '--showlocals', '--color=yes']
print('Launching pytest with args: %s' % args)
return_code = pytest.main(args)
if return_code != 0:
print('-' * 80)
print('If you want to debug tests locally, modify {} and add --pdb to its args'.format(__file__))
print('-' * 80)
sys.exit(return_code)