如何使用python添加截图到诱惑报告?

时间:2015-04-28 20:25:44

标签: python webdriver pytest allure

我有这段代码:

# coding: utf-8
from selenium import webdriver
import pytest
import allure


@pytest.yield_fixture(scope='session')
def driver():
    _driver = webdriver.PhantomJS()
    yield _driver
    _driver.quit()


def test_ya(driver):
    with allure.step('open ya.ru and take screenshot'):
        driver.get('http://ya.ru/')            
        allure.attach('screenshot', driver.get_screenshot_as_png(), type='png')

我尝试截取屏幕并将其保存到诱惑报告中, 执行后我有:

>       with self._attachfile("%s-attachment.%s" % (uuid.uuid4(), attach_type.extension)) as f:
            if isinstance(body, text_type):
E           AttributeError: 'str' object has no attribute 'extension'

我该如何解决这个问题?

3 个答案:

答案 0 :(得分:7)

您需要使用allure module attachment type constant,而不是将type设置为字符串png,而Enum定义了extension属性:

from allure.constants import AttachmentType

allure.attach('screenshot', driver.get_screenshot_as_png(), type=AttachmentType.PNG)

答案 1 :(得分:7)

对于诱惑2

from allure_commons.types import AttachmentType

allure.attach(driver.get_screenshot_as_png(), name="Screenshot", attachment_type=AttachmentType.PNG)

答案 2 :(得分:0)

在conftest.py中添加以下内容,确保您已经具有驱动程序固定装置:

codeblock