职场经验：Python自动化生成漂亮的测试报告【干货】

运用Python的Unittest、数据驱动测试（DDT）、Excel、Jinja2和HTML技术，构建一个能够自动生成精美可视化测试报告的自动化测试框架

思路流程

封装读取数据，让所有数据都能够再excel中填写，不再填写任何一行逻辑代码

通过unittest框架的discover()找到匹配的测试用例，由HTMLTestRunner的run()方法执行测试用例并生成最新的测试报告。

集成TestResult，增加一些获取数据逻辑，将数据封存输出，结合jinja2模板引擎，生成报告

将报告路径及收集的结果参数传给发送邮件，钉钉通知，企业微信通知的方法，再run用例结束后，自动发送数据到所需的位置

这里不展开对具体每一步得详细介绍，直接结果导向开始讲解。

1. 数据驱动测试（DDT）：以数据为驱动的智慧

为了匹配框架使用，重新写了类似ddt的装饰器，结合用例名称生成测试函数名称，这样就兼容了unittest的框架特性。核心代码：

def list_data(datas):""":param datas: Test data:return:"""def wrapper(func):setattr(func, "PARAMS", datas)return funcreturn wrapperdef yaml_data(file_path):""":param file_path:YAML file path:return:"""def wrapper(func):try:with open(file_path, "r", encoding="utf-8") as f:datas = yaml.load(f, Loader=yaml.FullLoader)except:with open(file_path, "r", encoding="gbk") as f:datas = yaml.load(f, Loader=yaml.FullLoader)setattr(func, "PARAMS", datas)return funcreturn wrapper

2.unittest框架妙用：高度封装精简脚本代码

将上述的装饰器装饰测试方法，通过多集成，、封装及反射捣鼓一波，最终的执行脚本代码，只有么关键是十来行了：

test_file = Config.TEST_CASE  # 获取 excel 文件路径
excel = DoExcel(test_file)test_case, databases, initialize_data, host = excel.get_excel_init_and_cases()@ddt
class TestProjectApi(unittest.TestCase):maxDiff = Noneaction = Action(initialize_data, databases)@classmethoddef setUpClass(cls) -> None:cls.action.load_modules_from_folder(extensions)def setUp(self) -> None:pass@list_data(test_case)def test_api(self, item):sheet, iid, condition, st, name, desc, method, expected = self.action.base_info(item)if self.action.is_run(condition):self.skipTest("这个测试用例听说泡面比较好吃，所以放弃执行了！！")regex, keys, deps, jp_dict, ex_request_data = self.action.extractor_info(item)self.action.pause_execution(st)self.action.exc_sql(item)if self.action.is_only_sql(method):self.skipTest("这条测试用例被 SQL 吃了，所以放弃执行了！！")# prepost_script = f"prepost_script_{sheet}_{iid}.py"# item = self.action.load_and_execute_script(Config.SCRIPTS_DIR, prepost_script, "setup", item)self.action.send_request(host, method, ex_request_data)self.action.analysis_response(sheet, iid, name, desc, regex, keys, deps, jp_dict)self.action.execute_validation(excel, sheet, iid, name, desc, expected)@classmethoddef tearDownClass(cls) -> None:excel.close_excel()

3.Excel王者：用例编写快速简单易过滤筛选

所有测试数据，填写在excel中，需要评审的时候，直接将excel丢给开发，一劳永逸，excel的强大，可不是三言两语能够解释清楚的，用例如下编写

"""封装读取excel“""
sheets = eval(self.get_excel_init().get(FieldNames.SHEETS))for sheet_name in sheets:sheet = self.wb[sheet_name]max_row = self.get_max_row(sheet)max_column = self.get_max_column(sheet)first_header = []for i in range(1, max_column + 1):first_header.append(sheet.cell(1, i).value)for i in range(2, max_row + 1):sub_data = {}for k in range(1, max_column + 1):sub_data[first_header[k - 1]] = sheet.cell(i, k).valuesub_data[FieldNames.SHEET] = sheet_nameyield sub_data