git.samba.org - third_party/subunit/blob - 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", help="Forward subunit stream on stdout.")
  37 (options, args) = parser.parse_args()
  38 if options.output_to is None:
  39     output_to = sys.stdout
  40 else:
  41     output_to = file(options.output_to, 'wb')
  42 try:
  43     result = JUnitXmlResult(output_to)
  44     if options.no_passthrough:
  45         passthrough_stream = DiscardStream()
  46     else:
  47         passthrough_stream = None
  48     if options.forward:
  49         forward_stream = sys.stdout
  50     else:
  51         forward_stream = None
  52     test = ProtocolTestCase(sys.stdin, passthrough=passthrough_stream,
  53        forward=forward_stream)
  54     result.startTestRun()
  55     test.run(result)
  56     result.stopTestRun()
  57 finally:
  58     if options.output_to is not None:
  59         output_to.close()
  60 if result.wasSuccessful():
  61     exit_code = 0
  62 else:
  63     exit_code = 1
  64 sys.exit(exit_code)