2014-05-16 14:20:26 +00:00
|
|
|
#!/usr/bin/env python2
|
2013-12-31 10:45:07 +00:00
|
|
|
#
|
|
|
|
# Copyright (c) 2013,Thibault Saunier <thibault.saunier@collabora.com>
|
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or
|
|
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
|
|
# License as published by the Free Software Foundation; either
|
|
|
|
# version 2.1 of the License, or (at your option) any later version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
# Lesser General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
|
|
# License along with this program; if not, write to the
|
|
|
|
# Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
|
|
# Boston, MA 02110-1301, USA.
|
|
|
|
|
|
|
|
""" Test Reporters implementation. """
|
|
|
|
|
|
|
|
import os
|
|
|
|
import re
|
2014-07-16 17:37:35 +00:00
|
|
|
import time
|
2013-12-31 10:45:07 +00:00
|
|
|
import codecs
|
2014-07-16 17:37:35 +00:00
|
|
|
import datetime
|
2016-03-23 10:48:10 +00:00
|
|
|
import tempfile
|
2014-01-08 17:51:14 +00:00
|
|
|
from loggable import Loggable
|
2013-12-31 10:45:07 +00:00
|
|
|
from xml.sax import saxutils
|
2016-03-23 10:48:10 +00:00
|
|
|
from utils import Result, printc, Colors
|
2013-12-31 10:45:07 +00:00
|
|
|
|
2016-03-23 10:48:10 +00:00
|
|
|
UNICODE_STRINGS = (type(unicode()) == type(str())) # noqa
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
|
|
|
|
class UnknownResult(Exception):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
CONTROL_CHARACTERS = re.compile(r"[\000-\010\013\014\016-\037]")
|
|
|
|
|
|
|
|
|
|
|
|
def xml_safe(value):
|
|
|
|
"""Replaces invalid XML characters with '?'."""
|
|
|
|
return CONTROL_CHARACTERS.sub('?', value)
|
|
|
|
|
|
|
|
|
|
|
|
def escape_cdata(cdata):
|
|
|
|
"""Escape a string for an XML CDATA section."""
|
|
|
|
return xml_safe(cdata).replace(']]>', ']]>]]><![CDATA[')
|
|
|
|
|
|
|
|
|
2014-01-08 17:51:14 +00:00
|
|
|
class Reporter(Loggable):
|
2013-12-31 10:45:07 +00:00
|
|
|
name = 'simple'
|
|
|
|
|
|
|
|
def __init__(self, options):
|
2014-01-08 17:51:14 +00:00
|
|
|
Loggable.__init__(self)
|
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
self.options = options
|
2014-07-16 17:37:35 +00:00
|
|
|
self._start_time = 0
|
2013-12-31 10:45:07 +00:00
|
|
|
self.stats = {'timeout': 0,
|
|
|
|
'failures': 0,
|
2014-01-09 10:14:19 +00:00
|
|
|
'passed': 0,
|
2013-12-31 10:45:07 +00:00
|
|
|
'skipped': 0
|
|
|
|
}
|
|
|
|
self.results = []
|
|
|
|
|
2015-01-16 17:25:56 +00:00
|
|
|
def init_timer(self):
|
|
|
|
"""Initialize a timer before starting tests."""
|
|
|
|
self._start_time = time.time()
|
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
def set_failed(self, test):
|
2014-01-09 10:14:19 +00:00
|
|
|
self.stats["failure"] += 1
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def set_passed(self, test):
|
|
|
|
self.stats["passed"] += 1
|
|
|
|
|
|
|
|
def add_results(self, test):
|
2014-01-08 17:51:14 +00:00
|
|
|
self.debug("%s", test)
|
2013-12-31 10:45:07 +00:00
|
|
|
if test.result == Result.PASSED:
|
|
|
|
self.set_passed(test)
|
|
|
|
elif test.result == Result.FAILED or \
|
|
|
|
test.result == Result.TIMEOUT:
|
|
|
|
self.set_failed(test)
|
|
|
|
else:
|
|
|
|
raise UnknownResult("%s" % test.result)
|
|
|
|
|
2015-01-16 17:42:19 +00:00
|
|
|
def after_test(self, test):
|
|
|
|
if test not in self.results:
|
|
|
|
self.results.append(test)
|
2014-07-21 16:00:42 +00:00
|
|
|
|
2015-01-16 17:42:19 +00:00
|
|
|
self.add_results(test)
|
2015-01-16 17:45:52 +00:00
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
def final_report(self):
|
2014-01-09 10:14:19 +00:00
|
|
|
print "\n"
|
|
|
|
printc("Final Report:", title=True)
|
2014-06-16 14:47:18 +00:00
|
|
|
for test in sorted(self.results, key=lambda test: test.result):
|
2013-12-31 10:45:07 +00:00
|
|
|
printc(test)
|
2014-07-23 15:49:21 +00:00
|
|
|
if test.result != Result.PASSED:
|
|
|
|
print "\n"
|
2013-12-31 10:45:07 +00:00
|
|
|
|
2014-01-09 10:14:19 +00:00
|
|
|
print "\n"
|
|
|
|
lenstat = (len("Statistics") + 1)
|
2014-10-24 12:23:52 +00:00
|
|
|
printc("Statistics:\n%s" % (lenstat * "-"), Colors.OKBLUE)
|
2014-07-16 17:37:35 +00:00
|
|
|
printc("\n%sTotal time spent: %s seconds\n" %
|
2014-10-24 12:23:52 +00:00
|
|
|
((lenstat * " "), datetime.timedelta(
|
|
|
|
seconds=(time.time() - self._start_time))),
|
2014-07-16 17:37:35 +00:00
|
|
|
Colors.OKBLUE)
|
2014-10-24 12:23:52 +00:00
|
|
|
printc("%sPassed: %d" %
|
|
|
|
(lenstat * " ", self.stats["passed"]), Colors.OKGREEN)
|
|
|
|
printc("%sFailed: %d" %
|
|
|
|
(lenstat * " ", self.stats["failures"]), Colors.FAIL)
|
|
|
|
printc("%s%s" %
|
|
|
|
(lenstat * " ", (len("Failed: 0")) * "-"), Colors.OKBLUE)
|
2014-01-09 10:14:19 +00:00
|
|
|
|
|
|
|
total = self.stats["failures"] + self.stats["passed"]
|
|
|
|
color = Colors.WARNING
|
|
|
|
if total == self.stats["passed"]:
|
|
|
|
color = Colors.OKGREEN
|
|
|
|
elif total == self.stats["failures"]:
|
|
|
|
color = Colors.FAIL
|
|
|
|
|
|
|
|
printc("%sTotal: %d" % (lenstat * " ", total), color)
|
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
class XunitReporter(Reporter):
|
2014-10-24 12:23:52 +00:00
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
"""This reporter provides test results in the standard XUnit XML format."""
|
|
|
|
name = 'xunit'
|
|
|
|
encoding = 'UTF-8'
|
|
|
|
|
|
|
|
def __init__(self, options):
|
|
|
|
super(XunitReporter, self).__init__(options)
|
2016-03-23 10:48:10 +00:00
|
|
|
|
|
|
|
self._createTmpFile()
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def final_report(self):
|
|
|
|
self.report()
|
2013-12-31 10:45:07 +00:00
|
|
|
super(XunitReporter, self).final_report()
|
2013-12-31 10:45:07 +00:00
|
|
|
|
2015-01-16 17:42:19 +00:00
|
|
|
def _get_captured(self, test):
|
2014-04-23 09:47:10 +00:00
|
|
|
captured = ""
|
2015-01-16 18:54:56 +00:00
|
|
|
if not self.options.redirect_logs:
|
2015-01-16 17:50:38 +00:00
|
|
|
value = test.get_log_content()
|
2013-12-31 10:45:07 +00:00
|
|
|
if value:
|
2014-04-23 09:47:10 +00:00
|
|
|
captured += '<system-out><![CDATA[%s' % \
|
2013-12-31 10:45:07 +00:00
|
|
|
escape_cdata(value)
|
2015-01-16 17:42:19 +00:00
|
|
|
for extralog in test.extra_logfiles:
|
2014-10-24 12:23:52 +00:00
|
|
|
captured += "\n\n===== %s =====\n\n" % escape_cdata(
|
|
|
|
os.path.basename(extralog))
|
2015-01-16 17:42:19 +00:00
|
|
|
value = test.get_extra_log_content(extralog)
|
2014-04-26 07:16:26 +00:00
|
|
|
captured += escape_cdata(value)
|
|
|
|
|
|
|
|
captured += "]]></system-out>"
|
2014-04-23 09:47:10 +00:00
|
|
|
|
|
|
|
return captured
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def _quoteattr(self, attr):
|
|
|
|
"""Escape an XML attribute. Value can be unicode."""
|
|
|
|
attr = xml_safe(attr)
|
|
|
|
if isinstance(attr, unicode) and not UNICODE_STRINGS:
|
|
|
|
attr = attr.encode(self.encoding)
|
|
|
|
return saxutils.quoteattr(attr)
|
|
|
|
|
|
|
|
def report(self):
|
|
|
|
"""Writes an Xunit-formatted XML file
|
|
|
|
|
|
|
|
The file includes a report of test errors and failures.
|
|
|
|
|
|
|
|
"""
|
2014-01-08 17:51:14 +00:00
|
|
|
self.debug("Writing XML file to: %s", self.options.xunit_file)
|
2016-03-23 10:48:10 +00:00
|
|
|
xml_file = codecs.open(self.options.xunit_file, 'w',
|
|
|
|
self.encoding, 'replace')
|
|
|
|
|
2013-12-31 10:45:07 +00:00
|
|
|
self.stats['encoding'] = self.encoding
|
2016-03-23 10:48:10 +00:00
|
|
|
self.stats['total'] = (self.stats['timeout'] + self.stats['failures'] +
|
|
|
|
self.stats['passed'] + self.stats['skipped'])
|
|
|
|
|
|
|
|
xml_file.write(u'<?xml version="1.0" encoding="%(encoding)s"?>'
|
|
|
|
u'<testsuite name="gst-validate-launcher" tests="%(total)d" '
|
|
|
|
u'errors="%(timeout)d" failures="%(failures)d" '
|
|
|
|
u'skip="%(skipped)d">' % self.stats)
|
|
|
|
|
|
|
|
tmp_xml_file = codecs.open(self.tmp_xml_file.name, 'r',
|
|
|
|
self.encoding, 'replace')
|
|
|
|
|
|
|
|
for l in tmp_xml_file:
|
|
|
|
xml_file.write(l)
|
|
|
|
|
|
|
|
xml_file.write(u'</testsuite>')
|
|
|
|
xml_file.close()
|
|
|
|
tmp_xml_file.close()
|
|
|
|
os.remove(self.tmp_xml_file.name)
|
|
|
|
|
|
|
|
self._createTmpFile()
|
|
|
|
|
|
|
|
def _createTmpFile(self):
|
|
|
|
self.tmp_xml_file = tempfile.NamedTemporaryFile(delete=False)
|
|
|
|
self.tmp_xml_file.close()
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def set_failed(self, test):
|
|
|
|
"""Add failure output to Xunit report.
|
|
|
|
"""
|
|
|
|
self.stats['failures'] += 1
|
2016-03-23 10:48:10 +00:00
|
|
|
|
|
|
|
xml_file = codecs.open(self.tmp_xml_file.name, 'a',
|
|
|
|
self.encoding, 'replace')
|
|
|
|
xml_file.write(self._forceUnicode(
|
2013-12-31 10:45:07 +00:00
|
|
|
'<testcase classname=%(cls)s name=%(name)s time="%(taken).3f">'
|
|
|
|
'<failure type=%(errtype)s message=%(message)s>'
|
|
|
|
'</failure>%(systemout)s</testcase>' %
|
2014-01-10 15:46:00 +00:00
|
|
|
{'cls': self._quoteattr(test.get_classname()),
|
|
|
|
'name': self._quoteattr(test.get_name()),
|
2013-12-31 10:45:07 +00:00
|
|
|
'taken': test.time_taken,
|
|
|
|
'errtype': self._quoteattr(test.result),
|
|
|
|
'message': self._quoteattr(test.message),
|
2015-01-16 17:42:19 +00:00
|
|
|
'systemout': self._get_captured(test),
|
2016-03-23 10:48:10 +00:00
|
|
|
}))
|
|
|
|
xml_file.close()
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def set_passed(self, test):
|
|
|
|
"""Add success output to Xunit report.
|
|
|
|
"""
|
2014-01-09 10:14:19 +00:00
|
|
|
self.stats['passed'] += 1
|
2016-03-23 10:48:10 +00:00
|
|
|
|
|
|
|
xml_file = codecs.open(self.tmp_xml_file.name, 'a',
|
|
|
|
self.encoding, 'replace')
|
|
|
|
xml_file.write(self._forceUnicode(
|
2013-12-31 10:45:07 +00:00
|
|
|
'<testcase classname=%(cls)s name=%(name)s '
|
|
|
|
'time="%(taken).3f">%(systemout)s</testcase>' %
|
2014-01-10 15:46:00 +00:00
|
|
|
{'cls': self._quoteattr(test.get_classname()),
|
|
|
|
'name': self._quoteattr(test.get_name()),
|
2013-12-31 10:45:07 +00:00
|
|
|
'taken': test.time_taken,
|
2015-01-16 17:42:19 +00:00
|
|
|
'systemout': self._get_captured(test),
|
2016-03-23 10:48:10 +00:00
|
|
|
}))
|
|
|
|
xml_file.close()
|
2013-12-31 10:45:07 +00:00
|
|
|
|
|
|
|
def _forceUnicode(self, s):
|
|
|
|
if not UNICODE_STRINGS:
|
|
|
|
if isinstance(s, str):
|
|
|
|
s = s.decode(self.encoding, 'replace')
|
|
|
|
return s
|