selenium-python-framework/conftest.py at main · ayushGadag/selenium-python-framework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import pytest
from utils.driver_factory import get_driver
from pages.login_page import LoginPage
from datetime import datetime
import os
import allure


# ---------- ADD CLI OPTION (HEADLESS MODE) ----------
def pytest_addoption(parser):
    parser.addoption(
        "--headless",
        action="store_true",
        default=False,
        help="Run tests in headless mode"
    )


# ---------- DRIVER FIXTURE ----------
@pytest.fixture
def driver(request):
    headless = request.config.getoption("--headless")

    driver = get_driver(headless=headless)
    yield driver
    driver.quit()


# ---------- LOGIN FIXTURE ----------
@pytest.fixture
def logged_in_user(driver):
    driver.get("https://www.saucedemo.com")

    login = LoginPage(driver)
    login.login("standard_user", "secret_sauce")

    return driver


# ---------- SCREENSHOT + ALLURE ATTACHMENT ----------
@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item):
    outcome = yield
    rep = outcome.get_result()

    if rep.when == "call" and rep.failed:

        driver = item.funcargs.get("driver", None)

        if driver:
            os.makedirs("screenshots", exist_ok=True)

            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            screenshot_path = f"screenshots/failure_{timestamp}.png"

            # Save screenshot
            driver.save_screenshot(screenshot_path)

            # Attach to Allure
            with open(screenshot_path, "rb") as image_file:
                allure.attach(
                    image_file.read(),
                    name="Failure Screenshot",
                    attachment_type=allure.attachment_type.PNG
                )

            print(f"Screenshot saved: {screenshot_path}")