41-pytest-报告中添加失败用例截图

---

前言

• 在UI自动化测试中，经常需要在用例失败时保存截图，那么本篇来学习下在allure和html报告中添加失败用例的截图。

allure报告

• conftset.py

# -*-coding:utf-8一*-
# @Time：2022/10/13
# @Author: 大海
import os
import allure
import pytest
from selenium import webdriver

driver = None

@pytest.mark.hookwrapper
def pytest_runtest_makereport(item, call):
    outcome = yield
    res = outcome.get_result()
    if res.when == "call" and res.failed:
        mode = "a" if os.path.exists("failures") else "w"
        with open("failures", mode) as f:
            # let's also access a fixture for the fun of it
            if "tmpdir" in item.fixturenames:
                extra = " (%s)" % item.funcargs["tmpdir"]
            else:
                extra = ""
            f.write(res.nodeid + extra + "\n")
        # 添加allure报告截图
        if hasattr(driver, "get_screenshot_as_png"):
            with allure.step('添加失败截图...'):
                allure.attach(driver.get_screenshot_as_png(), "失败截图", allure.attachment_type.PNG)


@pytest.fixture(scope='session')
def browser():
    global driver
    if driver is None:
        # chromedriver 路径已添加到系统环境变量中，所以此处可省略chromedriver路径参数
        driver = webdriver.Chrome()
    yield driver
    driver.quit()

• 编写case

# -*-coding:utf-8一*-
# @Time：2022/10/13
# @Author: 大海

import allure
import os
from selenium.webdriver.common.by import By


@allure.title("测试搜索页title")
def test_search(browser):
    """测试搜索页title"""
    with allure.step("step1：打开百度首页"):
        browser.get("https://www.baidu.com/")
    with allure.step("step2：搜索自动化测试"):
        browser.find_element(By.ID, "kw").send_keys("pytest")
    # 断言失败,验证是否截图
    print(browser.title)
    assert browser.title == "百度"


if __name__ == '__main__':
    os.system('pytest -s test_72.py --alluredir=./allure-data --clean-alluredir')
    # 打开allure报告 （目录与上面生成结果目录需一致）
    os.system('allure generate -c -o ./allure-report ./allure-data')
    os.system('allure open ./allure-report')

• 查看报告

html报告

• conftest.py

# -*-coding:utf-8一*-
# @Time：2022/10/13
# @Author: 大海

import pytest
from selenium import webdriver

@pytest.hookimpl(hookwrapper=True, tryfirst=True)
def pytest_runtest_makereport(item, call):
    pytest_html = item.config.pluginmanager.getplugin('html')
    outcome = yield
    report = outcome.get_result()
    extra = getattr(report, 'extra', [])

    if report.when == 'call' or report.when == "setup":
        xfail = hasattr(report,  'wasxfail')
        if (report.skipped and xfail) or (report.failed and not xfail):
            file_name = report.nodeid.replace("::", "_") + ".png"
            screen_img = _capture_screenshot()
            if file_name:
                html = '<div><img src="data:image/png;base64,%s" alt="screenshot" style="width:600px;height:300px;" ' \
                       'οnclick="window.open(this.src)" align="right"/></div>' % screen_img
                extra.append(pytest_html.extras.html(html))
        report.extra = extra
      


@pytest.fixture(scope='session')
def browser():
    global driver
    if driver is None:
        driver = webdriver.Chrome()
    yield driver
    driver.quit()


def _capture_screenshot():
    """截图"""
    return driver.get_screenshot_as_base64()

• 编写case

# -*-coding:utf-8一*-
# @Time：2022/10/13
# @Author: 大海

import allure
import os
from selenium.webdriver.common.by import By


@allure.title("测试搜索页title")
def test_search(browser):
    """测试搜索页title"""
    with allure.step("step1：打开百度首页"):
        browser.get("https://www.baidu.com/")
    with allure.step("step2：搜索自动化测试"):
        browser.find_element(By.ID, "kw").send_keys("pytest")
    # 断言失败,验证是否截图
    print(browser.title)
    assert browser.title == "百度"


if __name__ == '__main__':
    os.system('pytest -s -v test_72.py --html=report.html --self-contained-html')

• 查看报告