pwndbg/pytests_launcher.py

32 lines
755 B
Python

from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
from __future__ import unicode_literals
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 tests_launcher.py and add --pdb to its args')
print('-' * 80)
sys.exit(return_code)