职场经验：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 func
    return wrapper


def 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 func
    return 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 = None
    action = Action(initialize_data, databases)

    @classmethod
    def 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)

    @classmethod
    def 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).value
                    sub_data[FieldNames.SHEET] = sheet_name
                yield sub_data