pytest html-将图像从测试文件传递到conftest.py中的钩子

时间:2020-05-26 05:20:53

标签: python pytest pytest-html

我有一个通过pytest-html生成HTML输出的测试。

我得到了报告,但是我想添加对失败和期望图像的引用;我将它们保存在主test.py文件中,并将该钩子添加到conftest.py

现在,我不知道如何将这些图像传递给函数;执行测试后,将调用该挂钩;目前,我正在对输出文件进行硬编码,并将其附加;但是我想改为通过测试传递图像的路径,特别是因为我需要编写更多测试,这些测试可能会从我的常规文件夹中保存在其他地方,并且可能具有不同的名称。

这是我在conftest.py中拥有的钩子

@pytest.mark.hookwrapper
def pytest_runtest_makereport(item, call):

    timestamp = datetime.now().strftime('%H-%M-%S')

    pytest_html = item.config.pluginmanager.getplugin('html')
    outcome = yield
    report = outcome.get_result()
    extra = getattr(report, 'extra', [])
    if report.when == 'call':
        # Attach failure image, hardcoded...how do I pass this from the test?
        extra.append(pytest_html.extras.image('/tmp/image1.png'))

        # test report html
        extra.append(pytest_html.extras.url('http://www.theoutput.com/'))
        xfail = hasattr(report, 'wasxfail')
        if (report.skipped and xfail) or (report.failed and not xfail):
            # only add additional data on failure
            # Same as above, hardcoded but I want to pass the reference image from the test
            extra.append(pytest_html.extras.image('/tmp/image2.png'))
            extra.append(pytest_html.extras.html('<div>Additional HTML</div>'))
        report.extra = extra

如何从pytest测试文件传递到挂钩,该变量包含要附加的图像的路径?

1 个答案:

答案 0 :(得分:0)

我发现了一种解决方法,尽管它并不漂亮。

在测试文件的模块级别添加一个变量,允许我使用item.module.varname,因此,如果我在模块测试中设置varname,然后在测试中分配它;我可以在pytest_runtest_makereport

中访问它

在testfile.py

import pytest

myvar1 = None
myvar2 = None

class VariousTests(unittest.TestCase):

    def test_attachimages():

        global myvar1
        global myvar2

        myvar1 = "/tmp/img1.png"
        myvar2 = "/tmp/img2.png"

在conftest.py

@pytest.mark.hookwrapper
def pytest_runtest_makereport(item, call):

    timestamp = datetime.now().strftime('%H-%M-%S')

    pytest_html = item.config.pluginmanager.getplugin('html')
    outcome = yield
    report = outcome.get_result()
    extra = getattr(report, 'extra', [])
    if report.when == 'call':
        # Attach failure image
        img1 = item.module.myvar1
        img2 = item.module.myvar2
        extra.append(pytest_html.extras.png(img1))
        # test report html
        extra.append(pytest_html.extras.url('http://www.theoutput.com/'))
        xfail = hasattr(report, 'wasxfail')
        if (report.skipped and xfail) or (report.failed and not xfail):
            # only add additional data on failure
            # Same as above, hardcoded but I want to pass the reference image from the test
            extra.append(pytest_html.extras.png(img2))
            extra.append(pytest_html.extras.html('<div>Additional HTML</div>'))
        report.extra = extra