API自动化测试笔记（三）：应用多线程+合并测试报告

 1 #!/usr/bin/python
 2 # -*- coding: utf-8 -*-
 3 
 4 from unittest import defaultTestLoader
 5 from common.general import General
 6 from BeautifulReport import BeautifulReport
 7 import concurrent.futures
 8 
 9 
10 class RunTestTool():
11 
12     def __init__(self, temp_report_path, report_path, report_name="report_new.html"):
13         """
14         运行测试公共组件
15         :param temp_report_path: 测试报告临时存放目录（使用多线程生成的测试报告存放目录）
16         :param report_path: 测试报告路径
17         :param report_name: 测试报告文件名称
18         """
19         self.project_path = General.get_project_path()  # 获取工程路径
20         self.temp_report_path = '{}{}'.format(self.project_path, temp_report_path)
21 
22     def get_all_test_suit(self, test_case_path,description="测试报告", pattern='*_test.py'):
23         """
24         返回目录及子目录下的所有测试用例
25         :param description: 测试报告描述
26         :param test_case_path: 测试用例目录，接受list\tuple
27         :param pattern: 匹配测试用例文件的正则表达式
28         :return:
29         """
30         all_test_suit = []
31         test_suit = defaultTestLoader.discover(start_dir=test_case_path[0],
32                                                pattern=pattern,
33                                                top_level_dir=None)
34         if len(test_case_path) > 1:
35             for case_path in test_case_path[1:]:
36                 path = "{}{}".format(self.project_path, case_path)
37                 suit = defaultTestLoader.discover(start_dir=path,
38                                                   pattern=pattern,
39                                                   top_level_dir=path)
40                 test_suit.addTest(suit)
41         num = 0
42         for i in test_suit:
43             if len(list(i)) != 0:
44                 all_test_suit.append((i, num,description))
45                 num += 1
46         return all_test_suit
47 
48     def run_test(self, test_suit):
49         description = test_suit[2]
50         result = BeautifulReport(test_suit[0])
51         result.report(filename='report{}.html'.format(test_suit[1]),
52                       description=description,
53                       report_dir=self.temp_report_path)
54 
55 
56 test_case_path_list = ["目录1",'目录2']
57 temp_report_path = ""  # 测试报告临时目录
58 report_path = "" # 测试报告目录
59 rt = RunTestTool(temp_report_path=temp_report_path, report_path=report_path, )
60 test_suit = rt.get_all_test_suit(test_case_path=test_case_path_list,
61                                      description="描述11",
62                                      pattern="*_test.py")
63 with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:   # 多线程执行用例
64         executor.map(rt.run_test, test_suit)