Python (pytest and selenium) - fixture for taking screenshots on test failures

test_screenshot_fixture.py

# example fixture for taking screenshots on test failures
#
# requires:
#  - pytest
#  - selenium

import time

import pytest
from selenium import webdriver


@pytest.fixture
def driver(request):
    driver = webdriver.Chrome()
    failed_before = request.session.testsfailed
    yield driver
    if request.session.testsfailed > failed_before:
        filename = f"./screenshot_{int(time.time() * 1000)}.png"
        driver.save_screenshot(filename)
    driver.quit()


def test_fail(driver):
    driver.get("https://example.com/")
    assert "wrong text" in driver.title  # fails and a screenshot is saved


def test_pass(driver):
    driver.get("https://example.com/")
    assert True  # passes and no screenshot is saved