from tap import TAPTestRunner
import unittest
import sys
+import argparse
if __name__ == '__main__':
+ argparser = argparse.ArgumentParser()
+ argparser.add_argument(
+ '-f', '--failfast', help='Stop on first fail or error', action='store_true'
+ )
+
+ argparser.add_argument(
+ 'start_dir', help='Base directory where to search for tests', type=str
+ )
+
+ mut_exclu_group = argparser.add_mutually_exclusive_group(required=True)
+
+ mut_exclu_group.add_argument(
+ '-p',
+ '--pattern',
+ help='Glob-style pattern of test files to run ' '(e.g. test_event*.py)',
+ type=str,
+ )
+
+ mut_exclu_group.add_argument(
+ '-t',
+ '--test-case',
+ help='Run a specfic test module name, test class '
+ 'name, or test method name '
+ '(e.g. test_event.EventTestCase.test_clock_value)',
+ type=str,
+ )
+
+ args = argparser.parse_args()
+
loader = unittest.TestLoader()
- if len(sys.argv) >= 3:
- pattern = sys.argv[2]
+ start_dir = args.start_dir
+ pattern = args.pattern
+ failfast = args.failfast
+ test_case = args.test_case
+
+ if test_case:
+ sys.path.append(start_dir)
+ tests = loader.loadTestsFromName(test_case)
+ elif pattern:
+ tests = loader.discover(start_dir, pattern)
else:
- pattern = 'test*.py'
+ # This will never happen because the mutual exclusion group has the
+ # `required` parameter set to True. So one or the other must be set or
+ # else it will fail to parse.
+ sys.exit(1)
+
+ if tests.countTestCases() < 1:
+ print("No tests matching '%s' found in '%s'" % (pattern, start_dir))
+ sys.exit(1)
- tests = loader.discover(sys.argv[1], pattern)
- runner = TAPTestRunner()
+ runner = TAPTestRunner(failfast=failfast)
runner.set_stream(True)
runner.set_format('{method_name}')
sys.exit(0 if runner.run(tests).wasSuccessful() else 1)