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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#!/usr/bin/env python
from imp import find_module
import os
import os.path
import sys
from textwrap import dedent
from argparse import ArgumentParser
from unittest.loader import TestLoader
from unittest.runner import TextTestRunner
# list autopilot depends here, with the form:
# ('python module name', 'ubuntu package name'),
DEPENDS = [
('compizconfig', 'python-compizconfig'),
('dbus', 'python-dbus'),
('gobject', 'python-gobject'),
('gtk', 'python-gtk2'),
('ibus', 'python-ibus'),
('junitxml', 'python-junitxml'),
('testscenarios', 'python-testscenarios'),
('testtools', 'python-testtools'),
('xdg', 'python-xdg'),
('Xlib', 'python-xlib'),
]
def check_depends():
"""Check for required dependancies, and print a helpful message if any are
missing.
If all required modules are present, return True, False otherwise.
"""
missing = []
for module_name, package_name in DEPENDS:
try:
find_module(module_name)
except ImportError:
missing.append(package_name)
if missing:
print dedent("""\
You are missing one or more packages required to run autopilot. They
are:
%s
Please install these packages and re-run this script.
""" % (' '.join(missing))
)
return False
return True
def ensure_autopilot_is_importable():
"""Patch sys.path with the local autopilot directory if it's not already
importable.
"""
try:
find_module("autopilot")
except ImportError:
ap_dir = os.path.join(os.path.dirname(__file__),
"../tests/autopilot")
ap_dir = os.path.realpath(ap_dir)
sys.path.append(ap_dir)
print "Patching sys.path to include local autopilot folder '%s'\n" % ap_dir
def parse_arguments():
"""Parse command-line arguments, and return an argparse arguments object."""
parser = ArgumentParser(description="Autopilot test-runner")
subparsers = parser.add_subparsers(help='Run modes', dest="mode")
parser_run = subparsers.add_parser('run', help="Run autopilot tests")
parser_run.add_argument('-o', "--output", required=False,
help='Write test result report to file. Defaults to stdout')
parser_run.add_argument('-f', "--format", choices=['text', 'xml'], default='text',
required=False, help='Specify desired output format. Default is "text".')
parser_run.add_argument('-r', '--record', action='store_true', default=False,
required=False, help="Record failing tests. Required 'recordmydesktop' app to be installed. Videos \
are stored in /tmp/autopilot.")
parser_run.add_argument("-rd", "--record-directory", required=False,
default="/tmp/autopilot", type=str, help="Directory to put recorded tests (only if -r) specified.")
parser_run.add_argument("test", nargs="*", help="Specify tests to run, as listed by the 'list' command")
parser_list = subparsers.add_parser('list', help="List autopilot tests")
args = parser.parse_args()
return args
def list_tests():
"""Print a list of tests we find inside autopilot.tests."""
num_tests = 0
from testtools import iterate_tests
loader = TestLoader()
test_suite = loader.discover('autopilot.tests')
print "Listing all autopilot tests:"
print
for test in iterate_tests(test_suite):
has_scenarios = hasattr(test, "scenarios")
if has_scenarios:
num_tests += len(test.scenarios)
print " *%d %s" % (len(test.scenarios), test.id())
else:
num_tests += 1
print test.id()
print "\n %d total tests." % (num_tests)
def run_tests(args):
"""Run tests, using input from `args`."""
import junitxml
import autopilot.globals
if args.record:
autopilot.globals.video_recording_enabled = True
autopilot.globals.video_record_directory = args.record_directory
loader = TestLoader()
if args.test:
test_suite = loader.loadTestsFromNames(args.test)
else:
test_suite = loader.discover('autopilot.tests')
if args.output == None:
results_stream = sys.stdout
else:
try:
path = os.path.dirname(args.output)
if path != '' and not os.path.exists(path):
os.makedirs(path)
results_stream = open(args.output, 'w')
except:
results_stream = sys.stdout
if args.format == "xml":
result = junitxml.JUnitXmlResult(results_stream)
result.startTestRun()
test_suite.run(result)
result.stopTestRun()
results_stream.close()
if not result.wasSuccessful:
exit(1)
elif args.format == "text":
runner = TextTestRunner(stream=results_stream)
success = runner.run(test_suite).wasSuccessful()
if not success:
exit(1)
def main():
args = parse_arguments()
if args.mode == 'list':
list_tests()
elif args.mode == 'run':
run_tests(args)
if __name__ == "__main__":
if not check_depends():
exit(1)
ensure_autopilot_is_importable()
main()
|