之前对于用例的失败重试，和用例的历史测试记录存储展示做了很多的描述呢，但是都是基于各个项目呢，不方便使用，为了更好的使用，我们对这里进行抽离，抽离出来一个单独的模块，集成到BSTestRunner中，以后我们使用BSTestRunner直接就可以使用里面的失败重试和展示历史记录了。

首先呢，我们对于失败重试做了一个简单的调整。

在BSTestRunner 增加之前的重试代码

TestResult = unittest.TestResult
class MyResult(TestResult):
    def __init__(self, verbosity=1, trynum=1):
        # 默认次数是0
        TestResult.__init__(self)
        self.outputBuffer = StringIO()
        self.stdout0 = None
        self.stderr0 = None
        self.success_count = 0
        self.failure_count = 0
        self.error_count = 0
        self.verbosity = verbosity
        self.trynnum = trynum
        self.result = []
        self.trys = 0  #
        self.istry = False

    def startTest(self, test):
        TestResult.startTest(self, test)
        self.stdout0 = sys.stdout
        self.stderr0 = sys.stderr

    def complete_output(self):
        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):
        # 判断是否要重试
        if self.istry is True:
            # 如果执行的次数小于重试的次数 就重试
            if self.trys < self.trynnum:
                # 删除最后一个结果
                reslut = self.result.pop(-1)
                # 判断结果，如果是错误就把错误的个数减掉
                # 如果是失败，就把失败的次数减掉
                if reslut[0] == 1:
                    self.failure_count -= 1
                else:
                    self.error_count -= 1
                sys.stderr.write('{}:用例正在重试中。。。'.format(test.id()) + '\n')
                # 深copy用例
                test = copy.copy(test)
                # 重试次数增加+1
                self.trys += 1
                # 测试
                test(self)
            else:
                self.istry = False
                self.trys = 0
        self.complete_output()

    def addSuccess(self, test):
        # 成功就不要重试
        self.istry = False
        self.success_count += 1
        TestResult.addSuccess(self, test)
        output = self.complete_output()
        self.result.append((0, test, output, ''))
        if self.verbosity > 1:
            sys.stderr.write('ok ')
            sys.stderr.write(str(test))
            sys.stderr.write('\n')
        else:
            sys.stderr.write('.')

    def addError(self, test, err):
        # 重试+1，错误次数+1
        self.istry = True
        self.error_count += 1
        TestResult.addError(self, test, err)
        _, _exc_str = self.errors[-1]
        output = self.complete_output()
        self.result.append((2, test, output, _exc_str))
        if self.verbosity > 1:
            sys.stderr.write('E  ')
            sys.stderr.write(str(test))
            sys.stderr.write('\n')
        else:
            sys.stderr.write('E')

    def addFailure(self, test, err):
        self.istry = True
        TestResult.startTestRun(self)
        self.failure_count += 1
        TestResult.addFailure(self, test, err)
        _, _exc_str = self.failures[-1]
        output = self.complete_output()
        self.result.append((1, test, output, _exc_str))
        if self.verbosity > 1:
            sys.stderr.write('F  ')
            sys.stderr.write(str(test))
            sys.stderr.write('\n')
        else:
            sys.stderr.write('F')

    def stop(self) -> None:
        pass

　　

参数默认增加了默认参数。历史数据，我们在代码中，调整中增加了存储功能，我们默认存在txt文档中。我们看下存储和读取的方法。

 

name=os.path.join(self.filepath,self.stopTime.strftime('%Y_%m_%d_%H_%M_%S')+'.txt')
with open(name,'w+') as f:
      f.write(result.success_count.__str__()+"_"+result.error_count.__str__()+"_"+result.failure_count.__str__())
      f.close()

　

读取历史执行数据源

def   _readresult(self):

    namerun=[]
    faillist=[]
    success=[]
    error=[]
    for root,dirs,files in os.walk(self.filepath):
        for file in files:
            if file.endswith(".txt"):
                namerun.append(file.split(".")[0].split("/")[-1])
                with open(os.path.join(root,file),'r') as f:
                    reslut=f.readline().split('\n')[0].split("_")
                    success.append(reslut[0])
                    error.append(reslut[1])
                    faillist.append(reslut[2])
    return namerun,faillist,success,error

我们在展示的时候使用了百度的echarts，官网如下。

            echarts.apache.org/zh/index.ht…

我们在测试报告中做了展示。（代码太长，我放在github上：BSTestRunner），我们写一段代码演示下。

import  unittest,os
from  BSTestRunner import  BSTestRunner
BASH_DIR="history"
report_path = os.path.join(BASH_DIR,"test.html")
openone = open(report_path, 'w+')
class  Clasee(unittest.TestCase):
    def setUp(self) -> None:
        pass
    def tearDown(self) -> None:
        pass
    def testoen(self):
        self.assertEqual(1,2)
if __name__=="__main__":
    besautiful = BSTestRunner(title="报告",
                              description="测试报告",
                              stream=openone,
                              trynum=2,#重试次数
                              filepath=BASH_DIR,#根目录
                              is_show=True)#是否展示。
    test_suit = unittest.TestSuite()
    test_suit.addTests([Clasee("testoen")])
    besautiful.run(test_suit)

我们看下结果。

        

        看下测试报告。

完成了我们之前的设想，单独抽离到了git项目中，我贴下仓库的链接。

        最后代码放在了github。

https://github.com/liwanlei/BSTestRunner_new
复制代码

        如果访问过慢，可以访问码云

https://gitee.com/liwanlei/BSTestRunner_new

欢迎关注我的个人公众号