Как добавить скриншот в отчет allure с помощью python?

у меня есть этот код:

# 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')

и я пытаюсь сделать снимок экрана и сохранить его в отчете allure, после казни у меня:

>       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'

Как я могу это исправить?

6
allure pytest python webdriver

2 ответов

вместо type в строке png, вы должны использовать allure тип крепления модуля константа, который является Enum С

6

для allure 2

from allure_commons.types import AttachmentType

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