109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
|
#!/usr/bin/python
|
||
|
|
#-*- coding:utf-8 -*-
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
REQUIRED_PATHS = ["/usr/lib/anaconda",
|
||
|
|
"/usr/share/system-config-date"]
|
||
|
|
sys.path.extend(REQUIRED_PATHS)
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import tests
|
||
|
|
import string
|
||
|
|
from optparse import OptionParser
|
||
|
|
|
||
|
|
|
||
|
|
def getFullTestName(test, full_test_names):
|
||
|
|
tests = []
|
||
|
|
for full_test_name in full_test_names:
|
||
|
|
if full_test_name.lower().find(test) != -1:
|
||
|
|
tests.append(full_test_name)
|
||
|
|
|
||
|
|
return tests
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
usage = "usage: %prog [options] [test1 test2 ...]"
|
||
|
|
parser = OptionParser(usage)
|
||
|
|
parser.add_option("-l", "--list", action="store_true", default=False,
|
||
|
|
help="print all available tests and exit")
|
||
|
|
|
||
|
|
(options, args) = parser.parse_args(sys.argv[1:])
|
||
|
|
|
||
|
|
print "Searching for test suites"
|
||
|
|
available_suites = tests.getAvailableSuites()
|
||
|
|
test_keys = available_suites.keys()
|
||
|
|
if not test_keys:
|
||
|
|
print "No test suites available, exiting"
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
test_keys.sort()
|
||
|
|
|
||
|
|
if options.list:
|
||
|
|
print "\nAvailable tests:"
|
||
|
|
for test_name in test_keys:
|
||
|
|
print test_name
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
tests_to_run = []
|
||
|
|
|
||
|
|
if len(args) == 0:
|
||
|
|
# interactive mode
|
||
|
|
print "Running in interactive mode"
|
||
|
|
print "\nAvailable tests:"
|
||
|
|
test_num = 0
|
||
|
|
for test_name in test_keys:
|
||
|
|
print "[%3d] %s" % (test_num, test_name)
|
||
|
|
test_num += 1
|
||
|
|
print
|
||
|
|
|
||
|
|
try:
|
||
|
|
input_string = raw_input("Type in the test you want to run, "
|
||
|
|
"or \"all\" to run all listed tests: ")
|
||
|
|
except KeyboardInterrupt as e:
|
||
|
|
print "\nAborted by user"
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
for arg in input_string.split():
|
||
|
|
if arg.isdigit():
|
||
|
|
arg = int(arg)
|
||
|
|
try:
|
||
|
|
args.append(test_keys[arg])
|
||
|
|
except KeyError as e:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
args.append(arg)
|
||
|
|
|
||
|
|
args = map(string.lower, args)
|
||
|
|
if "all" in args:
|
||
|
|
tests_to_run = test_keys[:]
|
||
|
|
else:
|
||
|
|
for arg in args:
|
||
|
|
matching_tests = getFullTestName(arg, test_keys)
|
||
|
|
tests_to_run.extend(filter(lambda test: test not in tests_to_run,
|
||
|
|
matching_tests))
|
||
|
|
|
||
|
|
# run the tests
|
||
|
|
if tests_to_run:
|
||
|
|
tests_to_run.sort()
|
||
|
|
print "Running tests: %s" % tests_to_run
|
||
|
|
test_suite = unittest.TestSuite([available_suites[test]
|
||
|
|
for test in tests_to_run])
|
||
|
|
|
||
|
|
try:
|
||
|
|
result = unittest.TextTestRunner(verbosity=2).run(test_suite)
|
||
|
|
except KeyboardInterrupt as e:
|
||
|
|
print "\nAborted by user"
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if result.wasSuccessful():
|
||
|
|
print "\nAll tests OK"
|
||
|
|
sys.exit(0)
|
||
|
|
else:
|
||
|
|
print "\nTests finished with %d errors and %d failures" % (len(result.errors),
|
||
|
|
len(result.failures))
|
||
|
|
sys.exit(2)
|
||
|
|
else:
|
||
|
|
print "No test suites matching your criteria found, exiting"
|
||
|
|
sys.exit(1)
|