dosep can run tests in a specific subfolder

[vharron]

Summary

For example:

./dosep.py -o "$DOTEST_OPTS" lang/c/

[clayborg]

Looks good.

test/dosep.py

 out_q = None
 
 def process_dir_worker(arg_tuple):
     """Worker thread main loop when in multithreaded mode.
     Takes one directory specification at a time and works on it."""
     (root, files, test_root, dotest_options) = arg_tuple
     return process_dir(root, files, test_root, dotest_options)
 
-def walk_and_invoke(test_root, dotest_options, num_threads):
+def walk_and_invoke(test_directory, test_subdir, dotest_options, num_threads):
     """Look for matched files and invoke test driver on each one.
     In single-threaded mode, each test driver is invoked directly.
     In multi-threaded mode, submit each test driver to a worker
-    queue, and then wait for all to complete."""
+    queue, and then wait for all to complete.
+
+    test_directory - lldb/test/ directory
+    test_subdir - lldb/test/ or a subfolder with the tests we're interested in running
+    """
 
     # Collect the test files that we'll run.
     test_work_items = []
-    for root, dirs, files in os.walk(test_root, topdown=False):
-        test_work_items.append((root, files, test_root, dotest_options))
+    for root, dirs, files in os.walk(test_subdir, topdown=False):
+        test_work_items.append((root, files, test_directory, dotest_options))
 
     # Run the items, either in a pool (for multicore speedup) or
     # calling each individually.
     if num_threads > 1:
         pool = multiprocessing.Pool(num_threads)
         test_results = pool.map(process_dir_worker, test_work_items)
     else:
         test_results = []
for test_result in test_results:
         failed += dir_failed
         passed += dir_passed
 
     return (timed_out, failed, passed)
 
 def main():
     # We can't use sys.path[0] to determine the script directory
     # because it doesn't work under a debugger
-    test_root = os.path.dirname(os.path.realpath(__file__))
-
+    test_directory = os.path.dirname(os.path.realpath(__file__))
     parser = OptionParser(usage="""\
 Run lldb test suite using a separate process for each test file.
 
        Each test will run with a time limit of 10 minutes by default.
 
        Override the default time limit of 10 minutes by setting
        the environment variable LLDB_TEST_TIMEOUT.
 
""")
     parser.add_option('-t', '--threads',
                       type='int',
                       dest='num_threads',
                       help="""The number of threads to use when running tests separately.""")
 
     opts, args = parser.parse_args()
     dotest_options = opts.dotest_options
 
+    # The root directory was specified on the command line
+    if len(args) == 0:
+        test_subdir = test_directory
+    else:
+        test_subdir = os.path.join(test_directory, args[0])
+
     if opts.num_threads:
         num_threads = opts.num_threads
     else:
         num_threads_str = os.environ.get("LLDB_TEST_THREADS")
         if num_threads_str:
             num_threads = int(num_threads_str)
         else:
             num_threads = multiprocessing.cpu_count()
     if num_threads < 1:
         num_threads = 1
 
     system_info = " ".join(platform.uname())
-    (timed_out, failed, passed) = walk_and_invoke(test_root, dotest_options,
+    (timed_out, failed, passed) = walk_and_invoke(test_directory, test_subdir, dotest_options,
                                                   num_threads)
     timed_out = set(timed_out)
     num_tests = len(failed) + len(passed)
 
     print "Ran %d tests." % num_tests
     if len(failed) > 0:
         failed.sort()
         print "Failing Tests (%d)" % len(failed)