pytest生成简单自定义测试结果的html报告

def pytest_configure(config):
    # 添加配置
    config._metadata["项目名称"] = "测试报告"
    # 删除配置
    config._metadata.pop("JAVA_HOME")
    config._metadata.pop("Plugins")
    config._metadata.pop("Packages")
    config._metadata.pop("Platform")

# 定义一个全局变量，用于存储内容
global_data = {}
@pytest.fixture(scope="function")
def set_global_data():
    """
    设置全局变量，用于关联参数
    :return:
    """
    def _set_global_data(key, value):
        global_data[key] = value
    yield _set_global_data
    global_data.clear()

@user2ize("data", case_list)
def test_case(data, set_global_data):
    set_global_data("id", data.get("id"))
    set_global_data("method", data.get("method"))
    set_global_data("case_input", data.get("case_input"))
    set_global_data("case_output", data.get("case_output"))
    try:
        assert data.get("case_input") == data.get("case_output")
    except AssertionError:
        set_global_data("error_step", "断言失败")
        raise

@user3hook
def pytest_html_results_table_header(cells):
    """ 更改表头信息
    :param cells:
    :return:
    """
    cells.insert(1, html.th('用例ID', class_="sortable", col="id"))
    cells.insert(2, html.th('方法', class_="sortable", col="method"))
    cells.insert(3, html.th('输入', class_="sortable", col="case_input"))
    cells.insert(4, html.th('输出', class_="sortable", col="case_output"))
    cells.pop(-1)  # 删除link
    cells.pop(-2)  # 删除Test
@user4hook
def pytest_html_results_table_row(cells):
    """更改表中数据信息"""
    cells.insert(1, html.td(global_data.get("id")))
    cells.insert(2, html.td(global_data.get("method")))
    cells.insert(3, html.td(global_data.get("case_input")))
    cells.insert(4, html.td(global_data.get("case_output")))
    cells.pop(-1)  # 删除link
    cells.pop(-2)  # 删除Test

@user5hook
def pytest_html_results_table_html(report, data):
    if report.failed:
        del data[:]
        data.append(html.span(f"失败步骤:{global_data.get('error_step')}\n输出结果:{global_data.get('case_output')}",
                              class_='fail log'))
    elif report.passed:
        del data[:]
        data.append(html.div(f"输出结果:{global_data.get('case_output')}", class_='success log'))