3 # Pretty-format subunit output
4 # Copyright (C) 2008-2010 Jelmer Vernooij <jelmer@samba.org>
5 # Published under the GNU GPL, v3 or later
14 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../lib/subunit/python"))
15 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../lib/testtools"))
17 class PlainFormatter(object):
19 def __init__(self, summaryfile, verbose, immediate, statistics,
21 self.verbose = verbose
22 self.immediate = immediate
23 self.statistics = statistics
24 self.start_time = None
26 self.suitesfailed = []
29 self.summaryfile = summaryfile
32 self._progress_level = 0
33 self.totalsuites = totaltests
35 def progress(self, offset, whence):
36 if whence == subunit.PROGRESS_POP:
37 self._progress_level -= 1
38 elif whence == subunit.PROGRESS_PUSH:
39 self._progress_level += 1
40 elif whence == subunit.PROGRESS_SET:
41 if self._progress_level == 0:
42 self.totalsuites = offset
43 elif whence == subunit.PROGRESS_CUR:
44 raise NotImplementedError
46 def report_time(self, time):
47 if self.start_time is None:
48 self.start_time = time
51 def start_testsuite(self, name):
54 testsuite_start_time = self.last_time
56 duration = testsuite_start_time - self.start_time
59 self.test_output[name] = ""
61 out = "[%d" % self.index
62 if self.totalsuites is not None:
63 out += "/%d" % self.totalsuites
64 out += " in %ds" % duration
66 out += ", %d errors" % (len(self.suitesfailed),)
69 sys.stdout.write(out + "\n")
71 sys.stdout.write(out + ": ")
73 def output_msg(self, output):
75 sys.stdout.write(output)
76 elif self.name is not None:
77 self.test_output[self.name] += output
79 sys.stdout.write(output)
81 def control_msg(self, output):
82 #$self->output_msg($output)
85 def end_testsuite(self, name, result, reason):
89 if not name in self.test_output:
90 print "no output for name[%s]" % name
92 if result in ("success", "xfail"):
95 self.output_msg("ERROR: Testsuite[%s]\n" % name)
96 if reason is not None:
97 self.output_msg("REASON: %s\n" % (reason,))
98 self.suitesfailed.append(name)
99 if self.immediate and not self.verbose:
100 out += self.test_output[name]
103 if not self.immediate:
107 out += " " + result.upper() + "\n"
109 sys.stdout.write(out)
111 def start_test(self, testname):
114 def end_test(self, testname, result, unexpected, reason=None):
116 self.test_output[self.name] = ""
117 if not self.immediate:
122 'success': '.'}.get(result, "?(%s)" % result))
125 self.test_output[self.name] += "UNEXPECTED(%s): %s\n" % (result, testname)
126 if reason is not None:
127 self.test_output[self.name] += "REASON: %s\n" % (reason.strip(),)
129 if self.immediate and not self.verbose:
130 print self.test_output[self.name]
131 self.test_output[self.name] = ""
133 if not self.immediate:
137 'success': 'S'}.get(result, "?"))
140 f = open(self.summaryfile, 'w+')
142 if self.suitesfailed:
143 f.write("= Failed tests =\n")
145 for suite in self.suitesfailed:
146 f.write("== %s ==\n" % suite)
147 f.write(self.test_output[suite]+"\n\n")
151 if not self.immediate and not self.verbose:
152 for suite in self.suitesfailed:
154 print "FAIL: %s" % suite
155 print self.test_output[suite]
158 f.write("= Skipped tests =\n")
159 for reason in self.skips.keys():
160 f.write(reason + "\n")
161 for name in self.skips[reason]:
162 f.write("\t%s\n" % name)
166 print "\nA summary with detailed information can be found in:"
167 print " %s" % self.summaryfile
169 if not self.suitesfailed:
170 ok = (self.statistics['TESTS_EXPECTED_OK'] +
171 self.statistics['TESTS_EXPECTED_FAIL'])
172 print "\nALL OK (%d tests in %d testsuites)" % (ok, self.suites_ok)
174 print "\nFAILED (%d failures and %d errors in %d testsuites)" % (
175 self.statistics['TESTS_UNEXPECTED_FAIL'],
176 self.statistics['TESTS_ERROR'],
177 len(self.suitesfailed))
179 def skip_testsuite(self, name, reason="UNKNOWN"):
180 self.skips.setdefault(reason, []).append(name)
184 parser = optparse.OptionParser("format-subunit [options]")
185 parser.add_option("--verbose", action="store_true",
187 parser.add_option("--immediate", action="store_true",
188 help="Show failures immediately, don't wait until test run has finished")
189 parser.add_option("--prefix", type="string", default=".",
190 help="Prefix to write summary to")
192 opts, args = parser.parse_args()
196 'TESTS_UNEXPECTED_OK': 0,
197 'TESTS_EXPECTED_OK': 0,
198 'TESTS_UNEXPECTED_FAIL': 0,
199 'TESTS_EXPECTED_FAIL': 0,
204 def handle_sigint(sig, stack):
206 signal.signal(signal.SIGINT, handle_sigint)
208 msg_ops = PlainFormatter(os.path.join(opts.prefix, "summary"), opts.verbose,
209 opts.immediate, statistics)
211 expected_ret = subunithelper.parse_results(msg_ops, statistics, sys.stdin)
215 sys.exit(expected_ret)