摘要：
HTMLTestRunner生成如下图所示的报告：展开后，您可以查看详细信息：直接复制以下代码，将其保存为HTMLTestRunner。py，然后#coding=utf-8“”“ATestRunner用于Python单元测试框架。它生成HTML报告以显示结果
HTMLTestRunner 生成报告如下图：
展开后可查看详情信息：
直接复制以下代码，保存为HTMLTestRunner.py 即可
# coding=utf-8 """ A TestRunner for use with the Python unit testing framework. It generates a HTML report to show the result at a glance. The simplest way to use this is to invoke its main method. E.g. import unittest import HTMLTestRunner ... define your tests ... if __name__ == '__main__': HTMLTestRunner.main() For more customization options, instantiates a HTMLTestRunner object. HTMLTestRunner is a counterpart to unittest's TextTestRunner. E.g. # output to a file fp = file('my_report.html', 'wb') runner = HTMLTestRunner.HTMLTestRunner( stream=fp, title='My unit test', description='This demonstrates the report output by HTMLTestRunner.' ) # Use an external stylesheet. # See the Template_mixin class for more customizable options runner.STYLESHEET_TMPL = '<link rel="stylesheet" href="http://t.zoukankan.com/my_stylesheet.css" type="text/css">' # run the test runner.run(my_test_suite) ------------------------------------------------------------------------ Copyright (c) 2004-2007, Wai Yip Tung All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name Wai Yip Tung nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. """ # URL: http://tungwaiyip.info/software/HTMLTestRunner.html # URL: https://github.com/Gelomen/HTMLTestReportCN-ScreenShot __author__ = "Wai Yip Tung, Findyou, boafantasy, Gelomen" __version__ = "1.2.0" """ Change History Version 1.2.0 -- Gelomen * 优化用例说明显示 * 错误和失败报告里可以放入多张截图 Version 1.1.0 -- Gelomen * 优化报告截图写入方式 Version 1.0.2 -- Gelomen * 新增测试结果统计饼图 * 优化筛选时只显示预览 Version 1.0.1 -- Gelomen * 修复报告存入文件夹的bug * 优化报告的命名方式 Version 1.0.0 -- Gelomen * 修改测试报告文件夹路径的获取方式 * 修改截图获取文件夹路径的获取方式 Version 0.9.9 -- Gelomen * 优化报告文件夹命名 * 优化截图存放的目录 * 增加图片阴影边框以突出图片 * 优化 失败用例合集 和 错误用例合集 显示的颜色 Version 0.9.8 -- Gelomen * 优化回到顶部按钮的显示方式 Version 0.9.7 -- Gelomen * 优化截图显示，滚动页面会固定居中 Version 0.9.6 -- Gelomen * 新增打开图片的特效，可以直接在当前页面看截图 Version 0.9.5 -- Gelomen * heading新增 失败 和 错误 测试用例合集 Version 0.9.4 -- Gelomen * 修复失败和错误用例里对应按钮的颜色 Version 0.9.3 -- Gelomen * 修复点击失败或错误按钮后，浏览器版本和截图的列不会隐藏的bug Version 0.9.2 -- Gelomen * 美化 浏览器版本 和 截图 的显示 Version 0.9.1 -- Gelomen * 使用UI自动化测试时，增加 错误、失败 详细信息的 浏览器类型和版本 Version 0.9.0 -- Gelomen * 可通过 `need_screenshot=1` 作为开关，将报告开启截图功能 * 增加 失败 和 错误 详细信息的 截图链接 Version 0.8.4 -- Gelomen * 删除 失败模块 的显示 Version 0.8.3 -- Gelomen * 修复 测试结果 的筛选 * 优化 失败、错误 小图标的颜色 * 增加表格 最后一列 的显示，以美化表格 Version 0.8.2.1 -Findyou * 改为支持python3 Version 0.8.2.1 -Findyou * 支持中文，汉化 * 调整样式，美化（需要连入网络，使用的百度的Bootstrap.js） * 增加 通过分类显示、测试人员、通过率的展示 * 优化“详细”与“收起”状态的变换 * 增加返回顶部的锚点 Version 0.8.2 * Show output inline instead of popup window (Viorel Lupu). Version in 0.8.1 * Validated XHTML (Wolfgang Borgert). * Added description of test classes and test cases. Version in 0.8.0 * Define Template_mixin class for customization. * Workaround a IE 6 bug that it does not treat <script> block as CDATA. Version in 0.7.1 * Back port to Python 2.3 (Frank Horowitz). * Fix missing scroll bars in detail log (Podi). """ # TODO: color stderr # TODO: simplify javascript using ,ore than 1 class in the class attribute? import datetime import io import time import unittest from xml.sax import saxutils import sys import os import re # 全局变量 -- Gelomen _global_dict = {} # 让新建的报告文件夹路径存入全局变量 -- Gelomen class GlobalMsg(object): def __init__(self): global _global_dict _global_dict = {} @staticmethod def set_value(name, value): _global_dict[name] = value @staticmethod def get_value(name): try: return _global_dict[name] except KeyError: return None # ------------------------------------------------------------------------ # The redirectors below are used to capture output during testing. Output # sent to sys.stdout and sys.stderr are automatically captured. However # in some cases sys.stdout is already cached before HTMLTestRunner is # invoked (e.g. calling logging.basicConfig). In order to capture those # output, use the redirectors for the cached stream. # # e.g. # >>> logging.basicConfig(stream=HTMLTestRunner.stdout_redirector) # >>> class OutputRedirector(object): """ Wrapper to redirect stdout or stderr """ def __init__(self, fp): self.fp = fp def write(self, s): self.fp.write(s) def writelines(self, lines): self.fp.writelines(lines) def flush(self): self.fp.flush() stdout_redirector = OutputRedirector(sys.stdout) stderr_redirector = OutputRedirector(sys.stderr) # ---------------------------------------------------------------------- # Template class Template_mixin(object): """ Define a HTML template for report customerization and generation. Overall structure of an HTML report HTML +------------------------+ |<html> | | <head> | | | | STYLESHEET | | +----------------+ | | | | | | +----------------+ | | | | </head> | | | | <body> | | | | HEADING | | +----------------+ | | | | | | +----------------+ | | | | REPORT | | +----------------+ | | | | | | +----------------+ | | | | ENDING | | +----------------+ | | | | | | +----------------+ | | | | </body> | |</html> | +------------------------+ """ STATUS = { 0: '通过', 1: '失败', 2: '错误', } DEFAULT_TITLE = '测试报告' DEFAULT_DESCRIPTION = '' DEFAULT_TESTER = 'QA' # ------------------------------------------------------------------------ # HTML Template HTML_TMPL = r"""<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <title>%(title)s</title> <meta name="generator" content="%(generator)s"/> <meta http-equiv="Content-Type" content="text/html; charset=UTF-8"/> <link href="http://libs.baidu.com/bootstrap/3.0.3/css/bootstrap.min.css" rel="stylesheet"> <script src="http://libs.baidu.com/jquery/2.0.0/jquery.min.js"></script> <script src="http://libs.baidu.com/bootstrap/3.0.3/js/bootstrap.min.js"></script> <script src="https://img.hcharts.cn/highcharts/highcharts.js"></script> <script src="https://img.hcharts.cn/highcharts/modules/exporting.js"></script> %(stylesheet)s </head> <body > <script language="javascript" type="text/javascript"> $(function(){ // 修改 失败 和 错误 用例里对应按钮的颜色ClassName为动态加载 -- Gelomen $("button").each(function () { var text = $(this).text(); if(text == "失败"){ $(this).addClass("btn-danger") }else if(text == "错误") { $(this).addClass("btn-warning") } }); // 给失败和错误合集加样式 -- Gelomen var p_attribute = $("p.attribute"); p_attribute.eq(4).addClass("failCollection"); p_attribute.eq(5).addClass("errorCollection"); // 打开截图，放大，点击任何位置可以关闭图片 -- Gelomen $(".screenshot").click(function(){ var img = $(this).attr("img"); $('.pic_show img').attr('src', img); $('.pic_looper').fadeIn(200); $('.pic_show').fadeIn(200); var browserHeight = $(window).height(); var pic_boxHeight = $(".pic_box").height(); var top = (browserHeight - pic_boxHeight)/2; $('.pic_box').css("margin-top", top + "px") }); $('.pic_looper, .pic_show').click(function(){ $('.pic_looper').fadeOut(200); $('.pic_show').fadeOut(200) }); var browserWidth = $(window).width(); var margin_left = browserWidth/2 - 450; if(margin_left <= 240){ $("#container").css("margin", "auto"); }else { $("#container").css("margin-left", margin_left + "px"); } $(window).resize(function(){ // 改变窗口大小时，自动改变图片与顶部的距离 -- Gelomen var browserHeight = $(window).height(); var pic_boxHeight = $(".pic_box").height(); var top = (browserHeight - pic_boxHeight)/2; $('.pic_box').css("margin-top", top + "px"); // 改变窗口大小时，自动改变饼图的边距 -- Gelomen var browserWidth = $(window).width(); var margin_left = browserWidth/2 - 450; if(margin_left <= 240){ $("#container").css("margin", "auto"); }else { $("#container").css("margin-left", margin_left + "px"); } }); // 距离顶部超过浏览器窗口一屏时，回到顶部按钮才出现 -- Gelomen $(window).scroll(function(){ var browserHeight = $(window).height(); var top = $(window).scrollTop(); if(top >= browserHeight){ $("#toTop").css("display", "block") }else { $("#toTop").css("display", "none") } }) // 增加回到顶部过程的动画，以看上去不会那么生硬 -- Gelomen $("#toTop").click(function() { $("html,body").animate({"scrollTop":0}, 700) }) // 增加饼状图 -- Gelomen $('#container').highcharts({ chart: { plotBackgroundColor: null, plotBorderWidth: null, plotShadow: false, spacing : [0, 0, 0, 0] }, credits: { enabled: false }, navigation: { buttonOptions: { enabled: false } }, title: { floating:true, text: '测试结果占比' }, tooltip: { pointFormat: '{series.name}: <b>{point.percentage:.1f}%%</b>' }, plotOptions: { pie: { allowPointSelect: true, cursor: 'pointer', colors: ['#81ca9d', '#f16d7e', '#fdc68c'], dataLabels: { enabled: true, format: '<b>{point.name}</b>: {point.percentage:.1f} %%', style: { color: (Highcharts.theme && Highcharts.theme.contrastTextColor) || 'black' } }, point: { events: { mouseOver: function(e) { // 鼠标滑过时动态更新标题 chart.setTitle({ text: e.target.name+ ' '+ e.target.y + ' 个' }); } } } } }, series: [{ type: 'pie', innerSize: '80%%', name: '比例', data: [ ['通过', %(Pass)s], { name: '失败', y: %(fail)s, sliced: true, selected: true }, ['错误', %(error)s] ] }] }, function(c) { // 环形图圆心 var centerY = c.series[0].center[1], titleHeight = parseInt(c.title.styles.fontSize); c.setTitle({ y:centerY + titleHeight/2 }); chart = c; }); // 查看 失败 和 错误 合集链接文字切换 -- Gelomen $(".showDetail").click(function () { if($(this).html() == "点击查看"){ $(this).html("点击收起") }else { $(this).html("点击查看") } }) }); output_list = Array(); /*level 调整增加只显示通过用例的分类 --Findyou / 修复筛选显示bug --Gelomen 0:Summary //all hiddenRow 1:Failed //pt&et hiddenRow, ft none 2:Pass //pt none, ft&et hiddenRow 3:Error //pt&ft hiddenRow, et none 4:All //all none */ function showCase(level) { trs = document.getElementsByTagName("tr"); for (var i = 0; i < trs.length; i++) { tr = trs[i]; id = tr.id; if (id.substr(0,2) == 'ft') { if (level == 2 || level == 0 || level == 3) { tr.className = 'hiddenRow'; } else { tr.className = ''; // 切换筛选时只显示预览 -- Gelomen $("div[id^='div_ft']").attr("class", "collapse"); $("div[id^='div_et']").attr("class", "collapse"); } } if (id.substr(0,2) == 'pt') { if (level == 1 || level == 0 || level == 3) { tr.className = 'hiddenRow'; } else { tr.className = ''; // 切换筛选时只显示预览 -- Gelomen $("div[id^='div_ft']").attr("class", "collapse"); $("div[id^='div_et']").attr("class", "collapse"); } } if (id.substr(0,2) == 'et') { if (level == 1 || level == 0 || level == 2) { tr.className = 'hiddenRow'; } else { tr.className = ''; // 切换筛选时只显示预览 -- Gelomen $("div[id^='div_ft']").attr("class", "collapse"); $("div[id^='div_et']").attr("class", "collapse"); } } } //加入【详细】切换文字变化 --Findyou detail_class=document.getElementsByClassName('detail'); //console.log(detail_class.length) if (level == 3) { for (var i = 0; i < detail_class.length; i++){ detail_class[i].innerHTML="收起" } } else{ for (var i = 0; i < detail_class.length; i++){ detail_class[i].innerHTML="详细" } } } function showClassDetail(cid, count) { var id_list = Array(count); var toHide = 1; for (var i = 0; i < count; i++) { //ID修改 点 为 下划线 -Findyou tid0 = 't' + cid.substr(1) + '_' + (i+1); tid = 'f' + tid0; tr = document.getElementById(tid); if (!tr) { tid = 'p' + tid0; tr = document.getElementById(tid); if (!tr) { tid = 'e' + tid0; tr = document.getElementById(tid); } } id_list[i] = tid; if (tr.className) { toHide = 0; } } for (var i = 0; i < count; i++) { tid = id_list[i]; //修改点击无法收起的BUG，加入【详细】切换文字变化 --Findyou if (toHide) { document.getElementById(tid).className = 'hiddenRow'; document.getElementById(cid).innerText = "详细" } else { document.getElementById(tid).className = ''; document.getElementById(cid).innerText = "收起" } } } function html_escape(s) { s = s.replace(/&/g,'&'); s = s.replace(/</g,'<'); s = s.replace(/>/g,'>'); return s; } </script> %(heading)s %(report)s %(ending)s </body> </html> """ # variables: (title, generator, stylesheet, heading, report, ending) # ------------------------------------------------------------------------ # Stylesheet # # alternatively use a <link> for external style sheet, e.g. # <link rel="stylesheet" href="http://t.zoukankan.com/$url" type="text/css"> STYLESHEET_TMPL = """ <style type="text/css" media="screen"> body { font-family: Microsoft YaHei;padding: 20px; font-size: 100%; } table { font-size: 100%; } .table tbody tr td{ vertical-align: middle; } /* -- heading ---------------------------------------------------------------------- */ .heading .description, .attribute { clear: both; } /* --- 失败和错误合集样式 -- Gelomen --- */ .failCollection, .errorCollection { 100px; float: left; } #failCaseOl li { color: red } #errorCaseOl li { color: orange } /* --- 打开截图特效样式 -- Gelomen --- */ .data-img{ cursor:pointer } .pic_looper{ 100%; height:100%; position: fixed; left: 0; top:0; opacity: 0.6; background: #000; display: none; z-index: 100; } .pic_show{ 100%; position:fixed; left:0; top:0; right:0; bottom:0; margin:auto; text-align: center; display: none; z-index: 100; } .pic_box{ padding:10px; 90%; height:90%; margin:40px auto; text-align: center; overflow: hidden; } .pic_box img{ auto; height: 100%; -moz-box-shadow: 0px 0px 20px 0px #000; -webkit-box-shadow: 0px 0px 20px 0px #000; box-shadow: 0px 0px 20px 0px #000; } /* --- 饼状图div样式 -- Gelomen --- */ #container { 450px; height: 300px; float: left; } /* -- report ------------------------------------------------------------------------ */ #total_row { font-weight: bold; } .passCase { color: #5cb85c; } .failCase { color: #d9534f; font-weight: bold; } .errorCase { color: #f0ad4e; font-weight: bold; } .hiddenRow { display: none; } .testcase { margin-left: 2em; } .screenshot:link { text-decoration: none;color: deeppink; } .screenshot:visited { text-decoration: none;color: deeppink; } .screenshot:hover { text-decoration: none;color: darkcyan; } .screenshot:active { text-decoration: none;color: deeppink; } </style> """ # ------------------------------------------------------------------------ # Heading # # 添加显示截图 和 饼状图 的div -- Gelomen HEADING_TMPL = """<div class='pic_looper'></div> <div class='pic_show'><div class='pic_box'><img src='https://tool.4xseo.com/article/303256.html'/></div> </div> <div class='heading'> <div style=" 650px; float: left;"> <h1 style="font-family: Microsoft YaHei">%(title)s</h1> %(parameters)s <p class='description'>%(description)s</p> </div> <div id="container"></div> </div> """ # variables: (title, parameters, description) HEADING_ATTRIBUTE_TMPL = """<p class='attribute'><strong>%(name)s : </strong> %(value)s</p> """ # variables: (name, value) # ------------------------------------------------------------------------ # Report # # 汉化,加美化效果 --Findyou REPORT_TMPL = """ <div style=" 500px; clear: both;"> <p id='show_detail_line'> <a href='javascript:showCase(0)'>概要{ %(passrate)s }</a> <a href='javascript:showCase(2)'>通过{ %(Pass)s }</a> <a href='javascript:showCase(1)'>失败{ %(fail)s }</a> <a href='javascript:showCase(3)'>错误{ %(error)s }</a> <a href='javascript:showCase(4)'>所有{ %(count)s }</a> </p> </div> <table class="table table-condensed table-bordered table-hover"> <colgroup> <col /> <col /> <col /> <col /> <col /> <col /> <col /> <col /> </colgroup> <tr style="font-weight: bold;font-size: 14px;"> <td>用例集/测试用例</td> <td>说明</td> <td>总计</td> <td>通过</td> <td>失败</td> <td>错误</td> <td>耗时</td> <td>详细</td> </tr> %(test_list)s <tr class="text-center active"> <td colspan='2'>总计</td> <td>%(count)s</td> <td>%(Pass)s</td> <td>%(fail)s</td> <td>%(error)s</td> <td>%(time_usage)s</td> <td>通过率：%(passrate)s</td> </tr> </table> """ # variables: (test_list, count, Pass, fail, error ,passrate) REPORT_CLASS_TMPL = r""" <tr class='%(style)s warning'> <td>%(name)s</td> <td>%(doc)s</td> <td class="text-center">%(count)s</td> <td class="text-center">%(Pass)s</td> <td class="text-center">%(fail)s</td> <td class="text-center">%(error)s</td> <td class="text-center">%(time_usage)s</td> <td class="text-center"><a href="javascript:showClassDetail('%(cid)s',%(count)s)" id='%(cid)s'>详细</a></td> </tr> """ # variables: (style, desc, count, Pass, fail, error, cid) # 失败 的样式，去掉原来JS效果，美化展示效果 -Findyou / 美化类名上下居中，有截图列 -- Gelomen REPORT_TEST_WITH_OUTPUT_TMPL_1 = r""" <tr class='%(Class)s'> <td style="vertical-align: middle"><div class='testcase'>%(name)s</div></td> <td style="vertical-align: middle">%(doc)s</td> <td colspan='5' align='center'> <!--默认收起错误信息 -Findyou <button type="button" data-toggle="collapse" data-target='#div_%(tid)s'>%(status)s</button> <div class="collapse"> --> <!-- 默认展开错误信息 -Findyou / 修复失败按钮的颜色 -- Gelomen --> <button type="button" data-toggle="collapse" data-target='#div_%(tid)s,#div_%(tid)s_screenshot'>%(status)s</button> <div class="collapse in"> <pre style="text-align:left"> %(script)s </pre> </div> </td> <td style="vertical-align: middle"><div class="collapse in">浏览器版本：<div style="color: brown;">%(browser)s</div></br>截图：%(screenshot)s</div></td> </tr> """ # variables: (tid, Class, style, desc, status) # 失败 的样式，去掉原来JS效果，美化展示效果 -Findyou / 美化类名上下居中，无截图列 -- Gelomen REPORT_TEST_WITH_OUTPUT_TMPL_0 = r""" <tr class='%(Class)s'> <td style="vertical-align: middle"><div class='testcase'>%(name)s</div></td> <td style="vertical-align: middle">%(doc)s</td> <td colspan='5' align='center'> <!--默认收起错误信息 -Findyou <button type="button" data-toggle="collapse" data-target='#div_%(tid)s'>%(status)s</button> <div class="collapse"> --> <!-- 默认展开错误信息 -Findyou / 修复失败按钮的颜色 -- Gelomen --> <button type="button" data-toggle="collapse" data-target='#div_%(tid)s'>%(status)s</button> <div class="collapse in"> <pre style="text-align:left"> %(script)s </pre> </div> </td> <td style="vertical-align: middle"></td> </tr> """ # variables: (tid, Class, style, desc, status) # 通过 的样式，加标签效果 -Findyou / 美化类名上下居中 -- Gelomen REPORT_TEST_NO_OUTPUT_TMPL = r""" <tr class='%(Class)s'> <td style="vertical-align: middle"><div class='testcase'>%(name)s</div></td> <td style="vertical-align: left">%(doc)s</td> <td colspan='5' align='center'><span class="label label-success success">%(status)s</span></td> <td style="vertical-align: middle"></td> </tr> """ # variables: (tid, Class, style, desc, status) REPORT_TEST_OUTPUT_TMPL = r""" %(id)s: %(output)s """ # variables: (id, output) # ------------------------------------------------------------------------ # ENDING # # 增加返回顶部按钮 --Findyou ENDING_TMPL = """<div id='ending'> </div> <div style=" position:fixed;right:50px; bottom:30px; 20px; height:20px;cursor:pointer; display: none"> <a><span aria-hidden="true"> </span></a></div> """ # -------------------- The end of the Template class ------------------- TestResult = unittest.TestResult class _TestResult(TestResult): # note: _TestResult is a pure representation of results. # It lacks the output and reporting ability compares to unittest._TextTestResult. def __init__(self, verbosity=1): TestResult.__init__(self) self.stdout0 = None self.stderr0 = None self.success_count = 0 self.failure_count = 0 self.error_count = 0 self.verbosity = verbosity # result is a list of result in 4 tuple # ( # result code (0: success; 1: fail; 2: error), # TestCase object, # Test output (byte string), # stack trace, # ) self.result = [] # 增加一个测试通过率 --Findyou self.passrate = float(0) # 增加失败用例合集 self.failCase = "" # 增加错误用例合集 self.errorCase = "" def startTest(self, test): stream = sys.stderr # stdout_content = " Testing: " + str(test) # stream.write(stdout_content) # stream.flush() # stream.write(" ") TestResult.startTest(self, test) # just one buffer for both stdout and stderr self.outputBuffer = io.StringIO() stdout_redirector.fp = self.outputBuffer stderr_redirector.fp = self.outputBuffer self.stdout0 = sys.stdout self.stderr0 = sys.stderr sys.stdout = stdout_redirector sys.stderr = stderr_redirector self.test_start_time = round(time.time(), 2) def complete_output(self): """ Disconnect output redirection and return buffer. Safe to call multiple times. """ self.test_end_time = round(time.time(), 2) if self.stdout0: sys.stdout = self.stdout0 sys.stderr = self.stderr0 self.stdout0 = None self.stderr0 = None return self.outputBuffer.getvalue() def stopTest(self, test): # Usually one of addSuccess, addError or addFailure would have been called. # But there are some path in unittest that would bypass this. # We must disconnect stdout in stopTest(), which is guaranteed to be called. self.complete_output() def addSuccess(self, test): self.success_count += 1 TestResult.addSuccess(self, test) output = self.complete_output() use_time = round(self.test_end_time - self.test_start_time, 2) self.result.append((0, test, output, '', use_time)) if self.verbosity > 1: sys.stderr.write(' S ') sys.stderr.write(str(test)) sys.stderr.write(' ') else: sys.stderr.write(' S ') sys.stderr.write(' ') def addError(self, test, err): self.error_count += 1 TestResult.addError(self, test, err) _, _exc_str = self.errors[-1] output = self.complete_output() use_time = round(self.test_end_time - self.test_start_time, 2) self.result.append((2, test, output, _exc_str, use_time)) if self.verbosity > 1: sys.stderr.write(' E ') sys.stderr.write(str(test)) sys.stderr.write(' ') else: sys.stderr.write(' E ') sys.stderr.write(' ') # 添加收集错误用例名字 -- Gelomen self.errorCase += "<li>" + str(test) + "</li>" def addFailure(self, test, err): self.failure_count += 1 TestResult.addFailure(self, test, err) _, _exc_str = self.failures[-1] output = self.complete_output() use_time = round(self.test_end_time - self.test_start_time, 2) self.result.append((1, test, output, _exc_str, use_time)) if self.verbosity > 1: sys.stderr.write(' F ') sys.stderr.write(str(test)) sys.stderr.write(' ') else: sys.stderr.write(' F ') sys.stderr.write(' ') # 添加收集失败用例名字 -- Gelomen self.failCase += "<li>" + str(test) + "</li>" # 新增 need_screenshot 参数，-1为无需截图，否则需要截图 -- Gelomen class HTMLTestRunner(Template_mixin): """ """ def __init__(self, stream=sys.stdout, verbosity=2, title=None, description=None, tester=None): self.need_screenshot = 0 self.stream = stream self.verbosity = verbosity if title is None: self.title = self.DEFAULT_TITLE else: self.title = title if description is None: self.description = self.DEFAULT_DESCRIPTION else: self.description = description if tester is None: self.tester = self.DEFAULT_TESTER else: self.tester = tester self.startTime = datetime.datetime.now() def run(self, test): "Run the given test case or test suite." result = _TestResult(self.verbosity) # verbosity为1,只输出成功与否，为2会输出用例名称 test(result) self.stopTime = datetime.datetime.now() self.generateReport(test, result) # 优化测试结束后打印蓝色提示文字 -- Gelomen print("