Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | # |
| 3 | # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
| 6 | |
| 7 | """Tool for enumerating the tests in a given suite. |
| 8 | |
| 9 | Given an autotest root directory and a suite name (e.g., bvt, regression), this |
| 10 | tool will print out the name of each test in that suite, one per line. |
| 11 | |
| 12 | Example: |
| 13 | $ ./site_utils/suite_enumerator.py -a /usr/local/autotest bvt 2>/dev/null |
Chris Masone | a7dbfbc | 2014-02-06 14:24:22 -0800 | [diff] [blame] | 14 | login_LoginSuccess |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 15 | logging_CrashSender |
| 16 | login_BadAuthentication |
| 17 | |
| 18 | This is intended for use only with Chrome OS test suits that leverage the |
| 19 | dynamic suite infrastructure in server/cros/dynamic_suite.py. |
| 20 | """ |
| 21 | |
J. Richard Barnette | 6b355f1 | 2014-07-22 17:35:37 -0700 | [diff] [blame] | 22 | import logging |
Dan Shi | 11d36bc | 2013-02-25 16:23:02 -0800 | [diff] [blame] | 23 | import optparse, os, sys |
J. Richard Barnette | 6b355f1 | 2014-07-22 17:35:37 -0700 | [diff] [blame] | 24 | |
| 25 | # Silence messages relating to imports of missing, unneeded |
| 26 | # modules. |
| 27 | logging.basicConfig(level=logging.INFO) |
| 28 | |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 29 | import common |
Chris Sosa | accb5ce | 2012-08-30 17:29:15 -0700 | [diff] [blame] | 30 | from autotest_lib.client.common_lib.cros import dev_server |
Chris Masone | 44e4d6c | 2012-08-15 14:25:53 -0700 | [diff] [blame] | 31 | from autotest_lib.server.cros.dynamic_suite.suite import Suite |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 32 | |
| 33 | def parse_options(): |
Dan Shi | 11d36bc | 2013-02-25 16:23:02 -0800 | [diff] [blame] | 34 | """Parse command line for arguments including autotest directory, suite |
| 35 | name, if to list stable tests only, and if to list all available suites. |
| 36 | """ |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 37 | usage = "usage: %prog [options] suite_name" |
| 38 | parser = optparse.OptionParser(usage=usage) |
| 39 | parser.add_option('-a', '--autotest_dir', dest='autotest_dir', |
| 40 | default=os.path.abspath( |
J. Richard Barnette | 6b355f1 | 2014-07-22 17:35:37 -0700 | [diff] [blame] | 41 | os.path.join(os.path.dirname(__file__), |
| 42 | os.pardir)), |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 43 | help='Directory under which to search for tests.'\ |
| 44 | ' (e.g. /usr/local/autotest)') |
Zdenek Behan | 849db05 | 2012-02-29 19:16:28 +0100 | [diff] [blame] | 45 | parser.add_option('-l', '--listall', |
| 46 | action='store_true', default=False, |
| 47 | help='Print a listing of all suites. Ignores all args.') |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 48 | options, args = parser.parse_args() |
| 49 | return parser, options, args |
| 50 | |
| 51 | |
| 52 | def main(): |
Dan Shi | 11d36bc | 2013-02-25 16:23:02 -0800 | [diff] [blame] | 53 | """Entry point to run the suite enumerator command.""" |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 54 | parser, options, args = parse_options() |
Zdenek Behan | 849db05 | 2012-02-29 19:16:28 +0100 | [diff] [blame] | 55 | if options.listall: |
| 56 | if args: |
| 57 | print 'Cannot use suite_name with --listall' |
| 58 | parser.print_help() |
| 59 | elif not args or len(args) != 1: |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 60 | parser.print_help() |
| 61 | return |
| 62 | |
Chris Masone | 44e4d6c | 2012-08-15 14:25:53 -0700 | [diff] [blame] | 63 | fs_getter = Suite.create_fs_getter(options.autotest_dir) |
Chris Sosa | accb5ce | 2012-08-30 17:29:15 -0700 | [diff] [blame] | 64 | devserver = dev_server.ImageServer('') |
Zdenek Behan | 849db05 | 2012-02-29 19:16:28 +0100 | [diff] [blame] | 65 | if options.listall: |
Chris Sosa | accb5ce | 2012-08-30 17:29:15 -0700 | [diff] [blame] | 66 | for suite in Suite.list_all_suites('', devserver, fs_getter): |
Zdenek Behan | 849db05 | 2012-02-29 19:16:28 +0100 | [diff] [blame] | 67 | print suite |
| 68 | return |
| 69 | |
Alex Miller | a091307 | 2013-06-12 10:01:51 -0700 | [diff] [blame] | 70 | suite = Suite.create_from_name(args[0], '', '', devserver, fs_getter) |
Yusuf Mohsinally | 610911d | 2012-11-08 15:59:43 -0800 | [diff] [blame] | 71 | # If in test list, print firmware_FAFTSetup before other tests |
| 72 | # NOTE: the test.name value can be *different* from the directory |
| 73 | # name that appears in test.path |
| 74 | PRETEST_LIST = ['firmware_FAFTSetup',] |
| 75 | for test in filter(lambda test: test.name in \ |
| 76 | PRETEST_LIST, suite.stable_tests()): |
| 77 | print test.path |
| 78 | for test in filter(lambda test: test.name not in \ |
| 79 | PRETEST_LIST, suite.stable_tests()): |
Chris Masone | e8a4eff | 2012-02-28 16:33:43 -0800 | [diff] [blame] | 80 | print test.path |
J. Richard Barnette | 6b355f1 | 2014-07-22 17:35:37 -0700 | [diff] [blame] | 81 | |
Dan Shi | 11d36bc | 2013-02-25 16:23:02 -0800 | [diff] [blame] | 82 | # Check if test_suites/control.suite_name exists. |
J. Richard Barnette | 6b355f1 | 2014-07-22 17:35:37 -0700 | [diff] [blame] | 83 | control_path = os.path.join(options.autotest_dir, 'test_suites', |
| 84 | 'control.' + args[0]) |
| 85 | if not os.path.exists(control_path): |
| 86 | print >> sys.stderr, ('Warning! control file is missing: %s' % |
| 87 | control_path) |
Chris Masone | d6f38c8 | 2012-02-22 14:53:42 -0800 | [diff] [blame] | 88 | |
| 89 | |
| 90 | if __name__ == "__main__": |
| 91 | sys.exit(main()) |