self_tests/run_all_tests.py

#!/usr/bin/python
# Copyright (c) 2013 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.

# TODO(byronk): Consider using /autotest/files/utils/unittest_suite.py
# instead of this test runner.

from optparse import OptionParser
import logging
import sys
import unittest


def parse_command_line(argv):

    parser = OptionParser()
    parser.add_option("-n", "--with-hardware", dest="with_hardware",
                      action="store_true", default=False,
                      help="Normally fakes are used. This tests against"
                      "real hardware")
    parser.add_option("-l", "--with_logs",
                      action="store_true", dest="with_logs", default=False,
                      help="print log messages to stdout")
    parser.add_option("-v", "--verbose",
                      action="store_true", dest="verbose", default=False,
                      help="print log messages to stdout")
    parser.add_option('-d', '--test_dir', dest='test_dir', default='.',
                      help='Recursively run tests under this dir ')

    (options, args) = parser.parse_args(argv)
    return options, args


def main(argv):
    options, _ = parse_command_line(argv)
    if not options.with_logs:
        print 'Disabling all loggers....'
        # Disable all loggers to make the test output easier to read.
        logging.disable(100)

    # _with_hardware is used in test file names when they need actual hardware.
    test_files_pattern = 'test*py'
    if options.with_hardware:
        test_files_pattern = '*test_with_hardware.py'

    verbosity = 2
    if options.verbose:
        verbosity = 3
        print 'Setting verbosity to 3'
        print "Looking %s for files matching %s " % (options.test_dir,
                                                     test_files_pattern)

    test_loader = unittest.defaultTestLoader.discover(options.test_dir,
        pattern=test_files_pattern)
    test_runner = unittest.TextTestRunner(verbosity=verbosity, buffer=True)
    test_runner.run(test_loader)

if __name__ == '__main__':
    # Munge the path so we can run this file from inside the wireless_automation
    # dir, rather then having to cd .. and then run it. Do it here to avoid
    # any global path munging that would affect a file that imported this file.
    # All wireless_automation imports have to come below this path change.
    sys.path.append('..')
    main(sys.argv)