"""
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="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.
"""

import os, re, sys, io, time, datetime, unittest, logging
from xml.sax import saxutils

_global_dict = {}


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 = 'Tester'

    # ------------------------------------------------------------------------
    # 网页模板开始
    # 网页模板，变量列表 title, generator, stylesheet, heading, report, ending
    HTML_TMPL = r"""<!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"/>
    <meta name="viewport" content="width=device-width,initial-scale=1.0"/>
    <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")
            }else if(text == "通过") {
                $(this).addClass("btn-success")
            }
        });

        // 给失败和错误合集加样式 -- 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 <= 0){
                $("#container").css("margin", "auto");
                $("#container").css("float", "left");
                $("#container").css("width", "100%%");
                $("#testinfo").css("width", "100%%");
            }else {
                $("#container").css("margin-left", margin_left + "px");
                $("#container").css("float", "right");
                $("#container").css("width", "450px");
                $("#testinfo").css("width", "30%%");
            }
        });

        // 距离顶部超过浏览器窗口一屏时，回到顶部按钮才出现  -- 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: '<text style="font-size:10px">{series.name}: {point.percentage:.1f}%%</text>'
            },
            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+ '\t'+ e.target.y + ' 个'
                                });
                            }
                        }
                    }
                }
            },
            series: [{
                type: 'pie',
                innerSize: '80%%',
                name: '比例',
                data: [
                    ['通过', %(Pass)s], ['失败', %(fail)s], ['错误', %(error)s]   
                ]
            }]
        }, function(c) {
            // 环形圆心
            var centerY = c.series[0].center[1],
                titleHeight = parseInt(c.title.styles.fontSize);
            c.setTitle({
                x:0,
                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");
                $("div[id^='div_pt']").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");
                $("div[id^='div_pt']").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");
                $("div[id^='div_pt']").attr("class", "collapse");
            }
        }
    }

    //加入【详细】切换文字变化 --Findyou
    detail_class=document.getElementsByClassName('detail');
	//console.log(detail_class.length)
	if (level == 4) {
		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,'&amp;');
    s = s.replace(/</g,'&lt;');
    s = s.replace(/>/g,'&gt;');
    return s;
}
</script>
%(heading)s
%(report)s
%(ending)s
<div style='width:auto;height:24px;line-height:24px;border:1px solid #e3e3e3;text-align:center'><a href='#' style='color:#505050'></a></div>
</body>
</html>
"""
    # 网页模板结束
    # ------------------------------------------------------------------------

    # ------------------------------------------------------------------------
    # Stylesheet
    #
    # alternatively use a <link> for external style sheet, e.g.
    #   <link rel="stylesheet" href="$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 {
        width: auto;
        float: left;
    }
    #failedCaseOl li {
        color: red
    }
    #errorsCaseOl li {
        color: orange
    }
    
    /* --- 打开截图特效样式 -- Gelomen --- */
    .data-img{
        cursor:pointer
    }
    
    .pic_looper{
        width:100%;
        height:100%;
        position: fixed;
        left: 0;
        top:0;
        opacity: 0.6;
        background: #000;
        display: none;
        z-index: 100;
    }
    
    .pic_show{
        width: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;
        width:90%;
        height:90%;
        margin:40px auto;
        text-align: center;
        overflow: hidden;
    }
    
    .pic_box img{
        width: 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 {
        max-width: 100%;
        width: 450px;
        height: 300px;
        float: right;
    }
    
    /* -- report ------------------------------------------------------------------------ */
    #total_row  { font-weight: bold; }
    .passedCase { color: #3FB83F; font-family: Menlo,Monaco,Consolas,"Courier New",monospace; font-size: 14px; font-weight: bold; }
    .failedCase { color: #D9433F; font-family: Menlo,Monaco,Consolas,"Courier New",monospace; font-size: 14px; font-weight: bold; }
    .errorsCase { color: #F0A02F; font-family: Menlo,Monaco,Consolas,"Courier New",monospace; font-size: 14px; font-weight: bold; }
    .hiddenRow  { display: none; }
    .testcase   { margin-left: 1em; word-break: break-all; white-space: pre-wrap; }
    .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>
    """

    # ------------------------------------------------------------------------
    # 头部信息开始
    # 添加显示截图和统计图div，变量列表 title, parameters, description
    HEADING_TMPL = """
    <div class='pic_looper'></div>
    <div class='pic_show'>
        <div class='pic_box'>
            <img src=''/>
        </div>
    </div>
    <div class='heading'>
        <div id="testinfo" style="max-width:515px ;width: auto; 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>
    """

    # 测试信息模板，变量列表 name, value
    HEADING_ATTRIBUTE_TMPL = """
    <p class='attribute'><strong>%(name)s : </strong> %(value)s</p>
    """
    # 头部信息结束
    # ------------------------------------------------------------------------

    # ------------------------------------------------------------------------
    # 报告模板开始
    # 变量列表 test_list, count, Pass, fail, error ,passrate
    REPORT_TMPL = """
    <div style="width: auto; clear: both;">
    <p id='show_detail_line'>
    <a class="btn btn-primary" href='javascript:showCase(0)'>概要 %(passrate)s</a>
    <a class="btn btn-success" href='javascript:showCase(2)'>通过 %(Pass)s</a>
    <a class="btn btn-danger" href='javascript:showCase(1)'>失败 %(fail)s</a>
    <a class="btn btn-warning" href='javascript:showCase(3)'>错误 %(error)s</a>
    <a class="btn btn-info" href='javascript:showCase(4)'>全部 %(count)s</a>
    </p>
    </div>
    <table id='result_table' class="table table-condensed table-bordered table-hover">
    <colgroup>
    <col align='left' style="width: 300px;"/>
    <col align='right' style="width: 285px;"/>
    <col align='right' />
    <col align='right' />
    <col align='right' />
    <col align='right' />
    <col align='right' />
    <col align='right' style="width: 200px;"/>
    </colgroup>
    <tr id='header_row' class="text-center success" 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 id='total_row' 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>
    """

    # 变量列表 style, desc, count, Pass, fail, error, cid
    REPORT_CLASS_TMPL = """
    <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)" class="detail" id='%(cid)s'>查看全部</a></td>
    </tr>
    """

    # 失败样式(有截图列)，变量列表 tid, Class, style, desc, status
    REPORT_TEST_WITH_OUTPUT_TMPL_1 = """
    <tr id='%(tid)s' class='%(Class)s'>
        <td class='%(style)s' style="vertical-align: middle"><div class='testcase'>%(name)s</div></td>
        <td style="vertical-align: middle">%(doc)s</td>
        <td colspan='5' align='center'>
        <button id='btn_%(tid)s' type="button"  class="btn btn-xs" data-toggle="collapse" data-target='#div_%(tid)s,#div_%(tid)s_screenshot'>%(status)s</button>
        <div id='div_%(tid)s' class="collapse in">
        <pre style="text-align:left;font-size:12px;color:#e52000">
        %(script)s
        </pre>
        </div>
        </td>
        <td class="text-center" style="vertical-align: middle"><div id='div_%(tid)s_screenshot' class="collapse in">浏览器：<div style="color: brown;">%(browser)s</div></br>截图：%(screenshot)s</div></td>
    </tr>
    """

    # 失败样式(无截图列)，变量列表 tid, Class, style, desc, status
    REPORT_TEST_WITH_OUTPUT_TMPL_0 = """
    <tr id='%(tid)s' class='%(Class)s'>
        <td class='%(style)s' style="vertical-align: middle"><div class='testcase'>%(name)s</div></td>
        <td style="vertical-align: middle">%(doc)s</td>
        <td colspan='5' align='center'>
        <button id='btn_%(tid)s' type="button"  class="btn btn-xs" data-toggle="collapse" data-target='#div_%(tid)s'>%(status)s</button>
        <div id='div_%(tid)s' class="collapse in">
        <pre style="text-align:left;font-size:12px;color:#e52000">%(script)s</pre>
        </div>
        </td>
        <td class='%(style)s' style="vertical-align: middle"></td>
    </tr>
    """

    # 通过样式，变量列表 tid, Class, style, desc, status
    REPORT_TEST_NO_OUTPUT_TMPL = """
    <tr id='%(tid)s' class='%(Class)s'>
        <td class='%(style)s' style="vertical-align: middle"><div class='testcase'>%(name)s</div></td>
        <td style="vertical-align: left">%(doc)s</td>
        <td colspan='5' align='center'><button type="button" class="btn btn-xs">%(status)s</button></td>
        <td class='%(style)s' style="vertical-align: middle"></td>
    </tr>
    """

    # 测试输出内容
    REPORT_TEST_OUTPUT_TMPL = '%(id)s:' + "\n" + '%(output)s'

    # 返回顶部按钮
    ENDING_TMPL = """
    <div id='ending'>&nbsp;</div>
    <div id="toTop" style="position:fixed;right:50px; bottom:30px; width:20px; height:20px;cursor:pointer; display: none">
        <a>
            <span class="glyphicon glyphicon-eject" style = "font-size:28px;color:#b0b0b0" aria-hidden="true">
            </span>
        </a>
    </div>
    """
    # 报告模板结束
    # ------------------------------------------------------------------------


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):
        super().__init__(verbosity=verbosity)
        self.verbosity = verbosity
        self.outputBuffer = None
        self.stdout0 = None
        self.stderr0 = None
        self.passed_count = 0
        self.failed_count = 0
        self.errors_count = 0

        # 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 = []
        self.passrate = float(0)

        # 增加失败用例合集
        self.failedCase = ""
        # 增加错误用例合集
        self.errorsCase = ""

        self.logger = logging.getLogger('test')

    def startTest(self, test):
        stream = sys.stderr
        # stdout_content = " Testing: " + str(test)
        # stream.write(stdout_content)
        # stream.flush()
        # stream.write("\n")
        super().startTest(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.stime = round(time.time(), 2)

        self.loggerStream = io.StringIO()
        self.ch = logging.StreamHandler(self.loggerStream)
        self.ch.setLevel(logging.DEBUG)
        self.ch.setFormatter(
            logging.Formatter('%(asctime)s - %(name)s -%(levelname)s -%(process)d -%(processName)s -  %(message)s'))
        self.logger.addHandler(self.ch)


    def complete_output(self):
        """
        Disconnect output redirection and return buffer.
        Safe to call multiple times.
        """
        self.etime = round(time.time(), 2)
        if self.stdout0:
            sys.stdout = self.stdout0
            sys.stderr = self.stderr0
            self.stdout0 = None
            self.stderr0 = None
        return self.loggerStream.getvalue() + 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()
        # 移除日志Handler
        self.logger.removeHandler(self.ch)

    def addSuccess(self, test):
        self.passed_count += 1
        super().addSuccess(test)
        output = self.complete_output()
        utime = round(self.etime - self.stime, 2)
        self.result.append((0, test, output, '', utime))
        if self.verbosity > 1:
            # sys.stderr.write('Passed: ' + str(test.__module__) + '.' + str(test.__class__.__qualname__) + '.' +
                             # str(test.__dict__["_testMethodName"])     )
            sys.stderr.write('%s %s.%s.%-12s\t|\t%s' %('Passed:', str(test.__module__), str(test.__class__.__qualname__),
                             str(test.__dict__["_testMethodName"]), str(test.__dict__["_testMethodDoc"] or "")  ))
            sys.stderr.write("\n")
            # sys.stderr.write(str(test))
            # sys.stderr.write('\n')
            # sys.stderr.write(str(test.__dict__))


    def addError(self, test, err):
        self.errors_count += 1
        super().addError(test, err)
        _, _exc_str = self.errors[-1]
        output = self.complete_output()
        utime = round(self.etime - self.stime, 2)
        self.result.append((2, test, output, _exc_str, utime))
        if self.verbosity > 1:
            sys.stderr.write('  E  ')
            sys.stderr.write(str(test))
            sys.stderr.write('\n')
        else:
            sys.stderr.write('  E  ')
            sys.stderr.write('\n')

        # 收集错误测试用例名称以在测试报告中显示
        self.errorsCase += "<li>" + str(test) + "</li>"

    def addFailure(self, test, err):
        self.failed_count += 1
        super().addFailure(test, err)
        _, _exc_str = self.failures[-1]
        output = self.complete_output()
        utime = round(self.etime - self.stime, 2)
        self.result.append((1, test, output, _exc_str, utime))
        if self.verbosity > 1:
            sys.stderr.write('  F  ')
            sys.stderr.write(str(test))
            sys.stderr.write('\n')
        else:
            sys.stderr.write('  F  ')
            sys.stderr.write('\n')

        # 收集失败测试用例名称以在测试报告中显示
        self.failedCase += "<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("\n\033[36;0m--------------------- 测试结束 ---------------------\n"
              "------------- 合计耗时: %s -------------\033[0m" % (self.stopTime - self.startTime), file=sys.stderr)
        return result

    def sortResult(self, result_list):
        # unittest does not seems to run in any particular order.
        # Here at least we want to group them together by class.
        rmap = {}
        classes = []
        for n, t, o, e, s in result_list:
            cls = t.__class__
            if cls not in rmap:
                rmap[cls] = []
                classes.append(cls)
            rmap[cls].append((n, t, o, e, s))
        r = [(cls, rmap[cls]) for cls in classes]
        return r

    # 替换测试结果status为通过率 --Findyou
    def getReportAttributes(self, result):
        """
        Return report attributes as a list of (name, value).
        Override this to add custom attributes.
        """
        startTime = str(self.startTime)[:19]
        duration = time.strftime('%H:%M:%S', time.gmtime((self.stopTime - self.startTime).seconds))
        status = []
        status.append('总共 %s' % (result.passed_count + result.failed_count + result.errors_count))
        if result.passed_count:
            status.append('通过 %s' % result.passed_count)
        if result.failed_count:
            status.append('失败 %s' % result.failed_count)
        if result.errors_count:
            status.append('错误 %s' % result.errors_count)
        if status:
            status = '，'.join(status)
            if (result.passed_count + result.failed_count + result.errors_count) > 0:
                self.passrate = str("%.2f%%" % (float(result.passed_count) / float(
                    result.passed_count + result.failed_count + result.errors_count) * 100))
            else:
                self.passrate = "0.00 %"
        else:
            status = 'none'

        if len(result.failedCase) > 0:
            failedCase = result.failedCase
        else:
            failedCase = "无"

        if len(result.errorsCase) > 0:
            errorsCase = result.errorsCase
        else:
            errorsCase = "无"

        return [
            ('测试人员', self.tester),
            ('开始时间', startTime),
            ('合计耗时', duration),
            ('测试结果', status + "，通过率 " + self.passrate),
            ('失败用例', failedCase),
            ('错误用例', errorsCase),
        ]

    def generateReport(self, test, result):
        report_attrs = self.getReportAttributes(result)
        generator = 'HTMLTestRunner'
        stylesheet = self._generate_stylesheet()
        # 添加 通过、失败 和 错误 的统计，以用于饼图  -- Gelomen
        Pass = self._generate_report(result)["Pass"]
        fail = self._generate_report(result)["fail"]
        error = self._generate_report(result)["error"]

        heading = self._generate_heading(report_attrs)
        report = self._generate_report(result)["report"]
        ending = self._generate_ending()
        output = self.HTML_TMPL % dict(
            title=saxutils.escape(self.title),
            generator=generator,
            stylesheet=stylesheet,
            Pass=Pass,
            fail=fail,
            error=error,
            heading=heading,
            report=report,
            ending=ending,
        )
        self.stream.write(output.encode('utf8'))

    def _generate_stylesheet(self):
        return self.STYLESHEET_TMPL

    # 增加Tester显示 -Findyou
    # 增加 失败用例合集 和 错误用例合集 的显示  -- Gelomen
    def _generate_heading(self, report_attrs):
        a_lines = []
        for name, value in report_attrs:
            # 如果是 失败用例 或 错误用例合集，则不进行转义 -- Gelomen
            if name == "失败用例":
                if value == "无":
                    line = self.HEADING_ATTRIBUTE_TMPL % dict(
                        name=name,
                        value=value,
                    )
                else:
                    line = self.HEADING_ATTRIBUTE_TMPL % dict(
                        name=name,
                        value="<a class='showDetail' data-toggle='collapse' href='#failedCaseOl' style='text-decoration: none;'>点击查看</a>"
                              "<ol id='failedCaseOl' class='collapse' style='float: left; font-family: Menlo,Monaco,Consolas,monospace;'>" + value + "</ol>",
                    )
            elif name == "错误用例":
                if value == "无":
                    line = self.HEADING_ATTRIBUTE_TMPL % dict(
                        name=name,
                        value=value,
                    )
                else:
                    line = self.HEADING_ATTRIBUTE_TMPL % dict(
                        name=name,
                        value="<a class='showDetail' data-toggle='collapse' href='#errorsCaseOl' style='text-decoration: none;'>点击查看</a>"
                              "<ol id='errorsCaseOl' class='collapse' style='float: left; font-family: Menlo,Monaco,Consolas,monospace;'>" + value + "</ol>",
                    )
            else:
                line = self.HEADING_ATTRIBUTE_TMPL % dict(
                    name=saxutils.escape(name),
                    value=saxutils.escape(value),
                )
            a_lines.append(line)
        heading = self.HEADING_TMPL % dict(
            title=saxutils.escape(self.title),
            parameters=''.join(a_lines),
            description=saxutils.escape(self.description),
            tester=saxutils.escape(self.tester),
        )
        return heading

    # 生成报告  --Findyou添加注释
    def _generate_report(self, result):
        rows = []
        sortedResult = self.sortResult(result.result)
        # 所有用例统计耗时初始化
        sum_ns = 0
        for cid, (cls, cls_results) in enumerate(sortedResult):
            # subtotal for a class
            np = nf = ne = ns = 0
            for n, t, o, e, s in cls_results:
                if n == 0:
                    np += 1
                elif n == 1:
                    nf += 1
                elif n == 2:
                    ne += 1
                ns += s  # 把单个class用例文件里面的多个def用例每次的耗时相加
            ns = round(ns, 2)
            sum_ns += ns  # 把所有用例的每次耗时相加
            # format class description
            # if cls.__module__ == "__main__":
            #     name = cls.__name__
            # else:
            #     name = "%s.%s" % (cls.__module__, cls.__name__)
            name = cls.__name__
            doc = cls.__doc__ and cls.__doc__.split("\n")[0] or ""
            # desc = doc and '%s - %s' % (name, doc) or name

            row = self.REPORT_CLASS_TMPL % dict(
                style=ne > 0 and 'errorClass' or nf > 0 and 'failClass' or 'passClass',
                name=name,
                doc=doc,
                count=np + nf + ne,
                Pass=np,
                fail=nf,
                error=ne,
                cid='c%s' % (cid + 1),
                time_usage=str(ns) + "秒"  # 单个用例耗时
            )
            rows.append(row)

            for tid, (n, t, o, e, s) in enumerate(cls_results):
                self._generate_report_test(rows, cid, tid, n, t, o, e)
        sum_ns = round(sum_ns, 2)
        report = self.REPORT_TMPL % dict(
            test_list=''.join(rows),
            count=str(result.passed_count + result.failed_count + result.errors_count),
            Pass=str(result.passed_count),
            fail=str(result.failed_count),
            error=str(result.errors_count),
            time_usage=str(sum_ns) + "秒",  # 所有用例耗时
            passrate=self.passrate,
        )

        # 获取 通过、失败 和 错误 的统计并return，以用于饼图  -- Gelomen
        Pass = str(result.passed_count)
        fail = str(result.failed_count)
        error = str(result.errors_count)
        return {"report": report, "Pass": Pass, "fail": fail, "error": error}

    def _generate_report_test(self, rows, cid, tid, n, t, o, e):
        # e.g. 'pt1_1', 'ft1_1', 'et1_1'etc
        has_output = bool(o or e)
        # ID修改点为下划线,支持Bootstrap折叠展开特效 - Findyou
        if n == 0:
            tid_flag = 'p'
        elif n == 1:
            tid_flag = 'f'
        elif n == 2:
            tid_flag = 'e'
        tid = tid_flag + 't%s_%s' % (cid + 1, tid + 1)
        name = t.id().split('.')[-1]
        doc = t.shortDescription() or ""
        # desc = doc and ('%s - %s' % (name, doc)) or name

        # utf-8 支持中文 - Findyou
        # o and e should be byte string because they are collected from stdout and stderr?
        if isinstance(o, str):
            # TODO: some problem with 'string_escape': it escape \n and mess up formating
            # uo = unicode(o.encode('string_escape'))
            # uo = o.decode('latin-1')
            uo = o
        else:
            uo = o
        if isinstance(e, str):
            # TODO: some problem with 'string_escape': it escape \n and mess up formating
            # ue = unicode(e.encode('string_escape'))
            # ue = e.decode('latin-1')
            ue = e
        else:
            ue = e

        script = self.REPORT_TEST_OUTPUT_TMPL % dict(
            id=tid,
            output=saxutils.escape(uo + ue),
        )

        # 截图名字通过抛出异常存放在u，通过截取字段获得截图名字  -- Gelomen
        u = uo + ue
        # 先判断是否需要截图
        self.need_screenshot = u.find("errorImg[")

        if self.need_screenshot == -1:
            tmpl = has_output and self.REPORT_TEST_WITH_OUTPUT_TMPL_0 or self.REPORT_TEST_NO_OUTPUT_TMPL

            row = tmpl % dict(
                tid=tid,
                Class=(n == 0 and 'hiddenRow' or 'none'),
                style=n == 2 and 'errorsCase' or (n == 1 and 'failedCase' or 'passedCase'),
                name=name,
                doc=doc,
                script=script,
                status=self.STATUS[n],
            )
        else:
            tmpl = has_output and self.REPORT_TEST_WITH_OUTPUT_TMPL_1 or self.REPORT_TEST_NO_OUTPUT_TMPL

            screenshot_list = re.findall("errorImg\[(.*?)\]errorImg", u)
            screenshot = ""
            for i in screenshot_list:
                screenshot += "</br><a class=\"screenshot\" href=\"javascript:void(0)\" img=\"image/" + i + "\">img_" + i + "</a>"

            # screenshot = u[u.find('errorImg[') + 9:u.find(']errorImg')]
            browser = u[u.find('browser[') + 8:u.find(']browser')]

            row = tmpl % dict(
                tid=tid,
                Class=(n == 0 and 'hiddenRow' or 'none'),
                style=n == 2 and 'errorsCase' or (n == 1 and 'failedCase' or 'passedCase'),
                name=name,
                doc=doc,
                script=script,
                status=self.STATUS[n],
                # 添加截图字段
                screenshot=screenshot,
                # 添加浏览器版本字段
                browser=browser
            )
        rows.append(row)

        if not has_output:
            return

    def _generate_ending(self):
        return self.ENDING_TMPL


# 集成创建文件夹、保存截图、获得截图名字等方法，与HTMLTestReportCN交互从而实现嵌入截图  -- Gelomen
class ReportDirectory(object):

    def __init__(self, path="../../result/"):
        self.path = path
        self.title = "Test Report"

    def create_dir(self, title=None):
        i = 1.0

        if title is not None:
            self.title = title

        dir_path = self.path + self.title + "V" + str(round(i, 1))
        # 判断文件夹是否存在，不存在则创建
        while True:
            is_dir = os.path.isdir(dir_path)
            if is_dir:
                i += 0.1
                dir_path = self.path + self.title + "V" + str(round(i, 1))
            else:
                break

        os.makedirs(dir_path)

        # 测试报告路径
        report_path = dir_path + "/" + self.title + "V" + str(round(i, 1)) + ".html"

        # 将新建的 文件夹路径 和 报告路径 存入全局变量
        GlobalMsg.set_value("dir_path", dir_path)
        GlobalMsg.set_value("report_path", report_path)

    @staticmethod
    def get_screenshot(browser):
        i = 1

        # 通过全局变量获取文件夹路径
        new_dir = GlobalMsg.get_value("dir_path")

        img_dir = new_dir + "/image"
        # 判断文件夹是否存在，不存在则创建
        is_dir = os.path.isdir(img_dir)
        if not is_dir:
            os.makedirs(img_dir)

        img_path = img_dir + "/" + str(i) + ".png"

        # 有可能同个测试步骤出错，截图名字一样导致覆盖文件，所以名字存在则增加id
        while True:
            is_file = os.path.isfile(img_path)
            if is_file:
                i += 1
                img_path = img_dir + "/" + str(i) + ".png"
            else:
                break

        browser.get_screenshot_as_file(img_path)
        img_name = str(i) + ".png"

        browser_type = browser.capabilities["browserName"]
        browser_version = browser.capabilities["browserVersion"]
        browser_msg = browser_type + "(" + browser_version + ")"

        print("errorImg[" + img_name + "]errorImg, browser[" + browser_msg + "]browser")


##############################################################################
# Facilities for running tests from the command line
##############################################################################

# Note: Reuse unittest.TestProgram to launch test. In the future we may
# build our own launcher to support more specific command line
# parameters like test title, CSS, etc.
class TestProgram(unittest.TestProgram):
    """
    A variation of the unittest.TestProgram. Please refer to the base
    class for command line parameters.
    """

    def runTests(self):
        # Pick HTMLTestRunner as the default test runner.
        # base class's testRunner parameter is not useful because it means
        # we have to instantiate HTMLTestRunner before we know self.verbosity.
        if self.testRunner is None:
            self.testRunner = HTMLTestRunner(verbosity=self.verbosity)
        unittest.TestProgram.runTests(self)


main = TestProgram

##############################################################################
# Executing this module from the command line
##############################################################################

if __name__ == "__main__":
    main(module=None)
