最近做webUI自动化时，希望用例执行失败的时候，可以自动截图，以便于判断失败原因

pytest 有个很好的钩子函数 pytest_runtest_makereport 可以获取到用例执行的结果，所以我们在这个钩子函数里面判断用例失败后截图就可以了。
allure报告添加截图可以使用 allure.attach 方法

在 conftest.py 文件写用例执行的钩子函数

@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
    """
    获取每个用例状态的钩子函数
    :param item: 测试用例
    :param call: 测试步骤
    :return:
    """
    # 获取钩子方法的调用结果
    out_come = yield
    rep = out_come.get_result()  # 从钩子方法的调用结果中获取测试报告
    # rep.when表示测试步骤，仅仅获取用例call 执行结果是失败的情况, 不包含 setup/teardown
    if rep.when == "call" and rep.failed:
        mode = "a" if os.path.exists("failures") else "w"
        with open("failures", mode) as f:
            if "tmpdir" in item.fixturenames:
                extra = " (%s)" % item.funcargs["tmpdir"]
            else:
                extra = ""
            f.write((rep.nodeid + extra + "\n"))
        # 添加allure报告截图
        dirver = Driver.get_driver()
        if hasattr(dirver, "get_screenshot_as_png"):
            with allure.step('用例执行失败时，添加失败截图...'):
                logger.error("用例执行失败，捕获当前页面......")
                allure.attach(dirver.get_screenshot_as_png(), "失败截图", allure.attachment_type.PNG)

item是测试用例，call是测试步骤，具体执行过程如下：

• 先执行when='setup' 返回setup 的执行结果
• 然后执行when='call' 返回call 的执行结果
• 最后执行when='teardown'返回teardown 的执行结果

test_login.py 

import pytest
from selenium import webdriver
import allure


def test_login():
    browser = webdriber.Chrome()

    with allure.step("step1：打开登录首页"):
        browser.get("http://ip:6009/admin/login/?next=/admin/")
    with allure.step("step2：输入账号：admin"):
        browser.find_element_by_name("username").send_keys("admin")
    with allure.step("step2：输入密码：123456"):
        browser.find_element_by_name("password").send_keys("123456")

    # 故意断言失败，看是否会截图
    assert 1 == 2

 运行用例后，截图会存到./report 报告目录，allure报告展示如下：

参考文章：pytest---allure报告添加用例失败截图 - wakey - 博客园