Lib.test.junit_xml.py Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jython Show documentation
Show all versions of jython Show documentation
Jython is an implementation of the high-level, dynamic, object-oriented
language Python written in 100% Pure Java, and seamlessly integrated with
the Java platform. It thus allows you to run Python on any Java platform.
"""Support for writing JUnit XML test results for the regrtest"""
import os
import re
import sys
import time
import traceback
import unittest
from StringIO import StringIO
from xml.sax import saxutils
# Invalid XML characters (control chars)
EVIL_CHARACTERS_RE = re.compile(r"[\000-\010\013\014\016-\037]")
class JUnitXMLTestRunner:
"""A unittest runner that writes results to a JUnit XML file in
xml_dir
"""
def __init__(self, xml_dir):
self.xml_dir = xml_dir
def run(self, test):
result = JUnitXMLTestResult(self.xml_dir)
test(result)
result.write_xml()
return result
class JUnitXMLTestResult(unittest.TestResult):
"""JUnit XML test result writer.
The name of the file written to is determined from the full module
name of the first test ran
"""
def __init__(self, xml_dir):
unittest.TestResult.__init__(self)
self.xml_dir = xml_dir
# The module name of the first test ran
self.module_name = None
# All TestCases
self.tests = []
# Start time
self.start = None
self.old_stdout = sys.stdout
self.old_stderr = sys.stderr
sys.stdout = self.stdout = Tee(sys.stdout)
sys.stderr = self.stderr = Tee(sys.stderr)
def startTest(self, test):
unittest.TestResult.startTest(self, test)
self.ensure_module_name(test)
self.error, self.failure = None, None
self.start = time.time()
def stopTest(self, test):
took = time.time() - self.start
unittest.TestResult.stopTest(self, test)
args = [test, took]
if self.error:
args.extend(['error', self.error])
elif self.failure:
args.extend(['failure', self.failure])
self.tests.append(TestInfo.from_testcase(*args))
def addError(self, test, err):
unittest.TestResult.addError(self, test, err)
self.error = err
def addFailure(self, test, err):
unittest.TestResult.addFailure(self, test, err)
self.failure = err
def ensure_module_name(self, test):
"""Set self.module_name from test if not already set"""
if not self.module_name:
self.module_name = '.'.join(test.id().split('.')[:-1])
def write_xml(self):
if not self.module_name:
# No tests ran, nothing to write
return
took = time.time() - self.start
stdout = self.stdout.getvalue()
stderr = self.stderr.getvalue()
sys.stdout = self.old_stdout
sys.stderr = self.old_stderr
ensure_dir(self.xml_dir)
filename = os.path.join(self.xml_dir, 'TEST-%s.xml' % self.module_name)
stream = open(filename, 'w')
write_testsuite_xml(stream, len(self.tests), len(self.errors),
len(self.failures), 0, self.module_name, took)
for info in self.tests:
info.write_xml(stream)
write_stdouterr_xml(stream, stdout, stderr)
stream.write('')
stream.close()
class TestInfo(object):
"""The JUnit XML \n' % (skipped, name,
took))
def write_stdouterr_xml(stream, stdout, stderr):
"""Write the stdout/err tags"""
if stdout:
stream.write(' ')
stream.close()
def ensure_dir(dir):
"""Ensure dir exists"""
if not os.path.exists(dir):
os.mkdir(dir)
def exc_name(exc_info):
"""Determine the full name of the exception that caused exc_info"""
exc = exc_info[1]
name = getattr(exc.__class__, '__module__', '')
if name:
name += '.'
return name + exc.__class__.__name__
def exc_message(exc_info):
"""Safely return a short message passed through safe_str describing
exc_info, being careful of unicode values.
"""
exc = exc_info[1]
if exc is None:
return safe_str(exc_info[0])
if isinstance(exc, BaseException) and isinstance(exc.message, unicode):
return safe_str(exc.message)
try:
return safe_str(str(exc))
except UnicodeEncodeError:
try:
val = unicode(exc)
return safe_str(val)
except UnicodeDecodeError:
return '?'
def escape_cdata(cdata):
"""Escape a string for an XML CDATA section"""
return cdata.replace(']]>', ']]>]]>