Apply black code formatter on all Python code

[babeltrace.git] / tests / utils / python / testrunner.py

diff --git a/tests/utils/python/testrunner.py b/tests/utils/python/testrunner.py
index ef1872fa36b3703914d00c2ed5218adda3124cd5..6606246a0377e5c07d6ba30b7cff97c9d35d231a 100644 (file)
--- a/tests/utils/python/testrunner.py
+++ b/tests/utils/python/testrunner.py
@@ -23,12 +23,62 @@
 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()
-    tests = loader.discover(sys.argv[1])
-    runner = TAPTestRunner()
+
+    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:
+        # 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)
+
+    runner = TAPTestRunner(failfast=failfast)
     runner.set_stream(True)
     runner.set_format('{method_name}')
     sys.exit(0 if runner.run(tests).wasSuccessful() else 1)