HTMLTestRunner添加logging信息

如果要在报告中加入每一个测试用例执行的logging信息,则需要改HTMLTestRunner的源码

1、初始化定义logging信息,

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.logger = logging.getLogger('mylog') #已封装的log名称

  2、在startTest函数中初始化logging.Handler,记录到内存中

def startTest(self, test):
        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
        #----add logging output----fengf233,这个类加入以下部分
        self.log_cap = io.StringIO()
        self.ch = logging.StreamHandler(self.log_cap)
        self.ch.setLevel(logging.DEBUG)
        formatter = logging.Formatter('[%(levelname)s][%(asctime)s] [%(filename)s]->[%(funcName)s] line:%(lineno)d ---> %(message)s')
        self.ch.setFormatter(formatter)
        self.logger.addHandler(self.ch)

  

3、在complete_output函数的返回值中加入logging存在内存中的输出,用换行符隔开

def complete_output(self):
        """
        Disconnect output redirection and return buffer.
        Safe to call multiple times.
        """
        if self.stdout0:
            sys.stdout = self.stdout0
            sys.stderr = self.stderr0
            self.stdout0 = None
            self.stderr0 = None
        #add log out put ---fengf233修改
        return self.outputBuffer.getvalue()+'\n'+self.log_cap.getvalue()

  

4、每个用例执行完后,最好清除handler,在stopTest函数中加入

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.
        a = self.complete_output()
        #清除log的handle---fengf修改
        self.logger.removeHandler(self.ch)
        return a

  

posted @ 2022-03-08 15:50  究极不吃香菜  阅读(101)  评论(0编辑  收藏  举报