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
|
#!/usr/bin/env python
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""List all the test cases for a google test.
See more info at http://code.google.com/p/googletest/.
"""
import optparse
import subprocess
import sys
class Failure(Exception):
pass
def fix_python_path(cmd):
"""Returns the fixed command line to call the right python executable."""
out = cmd[:]
if out[0] == 'python':
out[0] = sys.executable
elif out[0].endswith('.py'):
out.insert(0, sys.executable)
return out
def gtest_list_tests(executable):
cmd = [executable, '--gtest_list_tests']
cmd = fix_python_path(cmd)
try:
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
except OSError, e:
raise Failure('Failed to run %s\n%s' % (executable, str(e)))
out, err = p.communicate()
if p.returncode:
raise Failure('Failed to run %s\n%s' % (executable, err), p.returncode)
# pylint: disable=E1103
if err and not err.startswith('Xlib: extension "RANDR" missing on display '):
raise Failure('Unexpected spew:\n%s' % err, 1)
return out
def _starts_with(a, b, prefix):
return a.startswith(prefix) or b.startswith(prefix)
def parse_gtest_cases(out, disabled=False, fails=False, flaky=False):
"""Expected format is a concatenation of this:
TestFixture1
TestCase1
TestCase2
"""
tests = []
fixture = None
lines = out.splitlines()
while lines:
line = lines.pop(0)
if not line:
break
if not line.startswith(' '):
fixture = line
else:
case = line[2:]
if case.startswith('YOU HAVE'):
# It's a 'YOU HAVE foo bar' line. We're done.
break
assert ' ' not in case
if not disabled and _starts_with(fixture, case, 'DISABLED_'):
continue
if not fails and _starts_with(fixture, case, 'FAILS_'):
continue
if not flaky and _starts_with(fixture, case, 'FLAKY_'):
continue
tests.append(fixture + case)
return tests
def main():
"""CLI frontend to validate arguments."""
parser = optparse.OptionParser(
usage='%prog <options> [gtest]')
parser.add_option(
'-d', '--disabled',
action='store_true',
help='Include DISABLED_ tests')
parser.add_option(
'-f', '--fails',
action='store_true',
help='Include FAILS_ tests')
parser.add_option(
'-F', '--flaky',
action='store_true',
help='Include FLAKY_ tests')
options, args = parser.parse_args()
if len(args) != 1:
parser.error('Please provide the executable to run')
try:
out = gtest_list_tests(args[0])
tests = parse_gtest_cases(
out, options.disabled, options.fails, options.flaky)
for test in tests:
print test
except Failure, e:
print e.args[0]
return e.args[1]
return 0
if __name__ == '__main__':
sys.exit(main())
|
