s4:auth Move struct auth_usersupplied_info to a common location
[samba.git] / lib / subunit / filters / subunit2junitxml
1 #!/usr/bin/env python
2 #  subunit: extensions to python unittest to get test results from subprocesses.
3 #  Copyright (C) 2009  Robert Collins <robertc@robertcollins.net>
4 #
5 #  Licensed under either the Apache License, Version 2.0 or the BSD 3-clause
6 #  license at the users choice. A copy of both licenses are available in the
7 #  project source as Apache-2.0 and BSD. You may not use this file except in
8 #  compliance with one of these two licences.
9 #  
10 #  Unless required by applicable law or agreed to in writing, software
11 #  distributed under these licenses is distributed on an "AS IS" BASIS, WITHOUT
12 #  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
13 #  license you chose for the specific language governing permissions and
14 #  limitations under that license.
15 #
16
17 """Filter a subunit stream to get aggregate statistics."""
18
19 from optparse import OptionParser
20 import sys
21 import unittest
22
23 from subunit import DiscardStream, ProtocolTestCase
24 try:
25     from junitxml import JUnitXmlResult
26 except ImportError:
27     sys.stderr.write("python-junitxml (https://launchpad.net/pyjunitxml or "
28         "http://pypi.python.org/pypi/junitxml) is required for this filter.")
29     raise
30
31 parser = OptionParser(description=__doc__)
32 parser.add_option("--no-passthrough", action="store_true",
33     help="Hide all non subunit input.", default=False, dest="no_passthrough")
34 parser.add_option("-o", "--output-to",
35     help="Output the XML to this path rather than stdout.")
36 parser.add_option("-f", "--forward", action="store_true", default=False,
37     help="Forward subunit stream on stdout.")
38 (options, args) = parser.parse_args()
39 if options.output_to is None:
40     output_to = sys.stdout
41 else:
42     output_to = file(options.output_to, 'wb')
43 try:
44     result = JUnitXmlResult(output_to)
45     if options.no_passthrough:
46         passthrough_stream = DiscardStream()
47     else:
48         passthrough_stream = None
49     if options.forward:
50         forward_stream = sys.stdout
51     else:
52         forward_stream = None
53     test = ProtocolTestCase(sys.stdin, passthrough=passthrough_stream,
54        forward=forward_stream)
55     result.startTestRun()
56     test.run(result)
57     result.stopTestRun()
58 finally:
59     if options.output_to is not None:
60         output_to.close()
61 if result.wasSuccessful():
62     exit_code = 0
63 else:
64     exit_code = 1
65 sys.exit(exit_code)