1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/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)
|