| #!/usr/bin/python2 |
| # |
| # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Tool for enumerating the tests in a given suite. |
| |
| Given an autotest root directory and a suite name (e.g., bvt, regression), this |
| tool will print out the name of each test in that suite, one per line. |
| |
| Example: |
| $ ./site_utils/suite_enumerator.py -a /usr/local/autotest bvt 2>/dev/null |
| login_LoginSuccess |
| logging_CrashSender |
| login_BadAuthentication |
| |
| This is intended for use only with Chrome OS test suits that leverage the |
| dynamic suite infrastructure in server/cros/dynamic_suite.py. |
| """ |
| |
| import logging |
| import optparse, os, sys |
| |
| # Silence messages relating to imports of missing, unneeded |
| # modules. |
| logging.basicConfig(level=logging.INFO) |
| |
| import common |
| import autotest_lib.client.common_lib.cros as cros_lib |
| import autotest_lib.server.cros.dynamic_suite.suite as suite_lib |
| |
| |
| def parse_options(): |
| """Parse command line for arguments including autotest directory, suite |
| name, if to list stable tests only, and if to list all available suites. |
| """ |
| usage = "usage: %prog [options] suite_name" |
| parser = optparse.OptionParser(usage=usage) |
| parser.add_option('-a', '--autotest_dir', dest='autotest_dir', |
| default=os.path.abspath( |
| os.path.join(os.path.dirname(__file__), |
| os.pardir)), |
| help='Directory under which to search for tests.'\ |
| ' (e.g. /usr/local/autotest)') |
| parser.add_option('-l', '--listall', |
| action='store_true', default=False, |
| help='Print a listing of all suites. Ignores all args.') |
| options, args = parser.parse_args() |
| return parser, options, args |
| |
| |
| def main(): |
| """Entry point to run the suite enumerator command.""" |
| parser, options, args = parse_options() |
| if options.listall: |
| if args: |
| print 'Cannot use suite_name with --listall' |
| parser.print_help() |
| elif not args or len(args) != 1: |
| parser.print_help() |
| return |
| |
| fs_getter = suite_lib.create_fs_getter(options.autotest_dir) |
| devserver = cros_lib.dev_server.ImageServer('') |
| if options.listall: |
| for suite in suite_lib.list_all_suites('', devserver, fs_getter): |
| print suite |
| return |
| |
| suite = suite_lib.Suite.create_from_name(args[0], {}, '', devserver, |
| fs_getter) |
| # If in test list, print firmware_FAFTSetup before other tests |
| # NOTE: the test.name value can be *different* from the directory |
| # name that appears in test.path |
| PRETEST_LIST = ['firmware_FAFTSetup',] |
| for test in filter(lambda test: test.name in \ |
| PRETEST_LIST, suite.tests): |
| print test.path |
| for test in filter(lambda test: test.name not in \ |
| PRETEST_LIST, suite.tests): |
| print test.path |
| |
| # Check if test_suites/control.suite_name exists. |
| control_path = os.path.join(options.autotest_dir, 'test_suites', |
| 'control.' + args[0]) |
| if not os.path.exists(control_path): |
| print >> sys.stderr, ('Warning! control file is missing: %s' % |
| control_path) |
| |
| |
| if __name__ == "__main__": |
| sys.exit(main()) |