make perftest: for performance testing
[sfrench/samba-autobuild/.git] / selftest / filter-subunit
1 #!/usr/bin/env python
2 # Filter a subunit stream
3 # Copyright (C) 2009-2011 Jelmer Vernooij <jelmer@samba.org>
4
5 # This program is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
9
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
17
18 # NOTE: This script is a hack, meant as a placeholder until we can migrate
19 # to upstream subunit's filtering tools.
20
21 import optparse
22 import sys
23 import signal
24
25 sys.path.insert(0, "bin/python")
26
27 import subunithelper
28
29 parser = optparse.OptionParser("filter-subunit [options] < instream > outstream")
30 parser.add_option("--expected-failures", type="string",
31     help="File containing list of regexes matching tests to consider known "
32          "failures")
33 parser.add_option("--flapping", type="string",
34     help="File containing list of flapping tests, of which to ignore results.")
35 parser.add_option("--strip-passed-output", action="store_true",
36     help="Whether to strip output from tests that passed")
37 parser.add_option("--fail-immediately", action="store_true",
38     help="Whether to stop on the first error", default=False)
39 parser.add_option("--prefix", type="string", default='',
40     help="Add prefix to all test names")
41 parser.add_option("--suffix", type="string", default='',
42     help="Add suffix to all test names")
43 parser.add_option("--fail-on-empty", default=False,
44     action="store_true", help="Fail if there was no subunit output")
45 parser.add_option("--list", default=False,
46     action="store_true", help="Operate in list mode")
47 parser.add_option("--perf-test-output", default=False,
48     action="store_true", help="orientate output for performance measurement")
49 opts, args = parser.parse_args()
50
51 if opts.list:
52     for l in sys.stdin:
53          sys.stdout.write("%s%s%s\n" % (opts.prefix, l.rstrip(), opts.suffix))
54     sys.exit(0)
55
56 if opts.perf_test_output:
57     bad_options = []
58     for bad_opt in ('fail_immediately', 'strip_passed_output',
59                     'flapping', 'expected_failures'):
60         if getattr(opts, bad_opt):
61             bad_options.append(bad_opt)
62     if bad_options:
63         print >>sys.stderr, ("--perf-test-output is incompatible with --%s" %
64                              (', --'.join(x.replace('_', '-')
65                                           for x in bad_options)))
66         sys.exit(1)
67
68 if opts.expected_failures:
69     expected_failures = subunithelper.read_test_regexes(opts.expected_failures)
70 else:
71     expected_failures = {}
72
73
74 if opts.flapping:
75     flapping = subunithelper.read_test_regexes(opts.flapping)
76 else:
77     flapping = {}
78
79 statistics = {
80     'TESTS_UNEXPECTED_OK': 0,
81     'TESTS_EXPECTED_OK': 0,
82     'TESTS_UNEXPECTED_FAIL': 0,
83     'TESTS_EXPECTED_FAIL': 0,
84     'TESTS_ERROR': 0,
85     'TESTS_SKIP': 0,
86 }
87
88 def handle_sigint(sig, stack):
89     sys.exit(0)
90 signal.signal(signal.SIGINT, handle_sigint)
91
92 out = subunithelper.SubunitOps(sys.stdout)
93
94 if opts.perf_test_output:
95     msg_ops = subunithelper.PerfFilterOps(out, opts.prefix, opts.suffix)
96 else:
97     msg_ops = subunithelper.FilterOps(out, opts.prefix, opts.suffix,
98                                       expected_failures,
99                                       opts.strip_passed_output,
100                                       fail_immediately=opts.fail_immediately,
101                                       flapping=flapping)
102
103 try:
104     ret = subunithelper.parse_results(msg_ops, statistics, sys.stdin)
105 except subunithelper.ImmediateFail:
106     sys.stdout.flush()
107     sys.exit(1)
108
109 if opts.fail_on_empty and not msg_ops.seen_output:
110     sys.exit(1)
111 else:
112     sys.exit(ret)