Taking Screenshots & Logging

import pytest
import logging
import os
from datetime import datetime
from selenium import webdriver

# ── LOGGING CONFIGURATION ─────────────────────────────────────
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(message)s",
    handlers=[
        logging.FileHandler("test_run.log"),
        logging.StreamHandler()                  # Also log to console
    ]
)
logger = logging.getLogger(__name__)

# ── SCREENSHOT ON FAILURE (pytest conftest.py) ────────────────
@pytest.fixture(scope="function")
def driver(request):
    driver = webdriver.Chrome()
    driver.implicitly_wait(10)
    yield driver
    
    # After test completes: check if test FAILED
    if request.node.rep_call.failed:
        # Create screenshots directory
        os.makedirs("screenshots", exist_ok=True)
        
        # Timestamp + test name for unique filename
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        test_name = request.node.name.replace("/", "_")
        filename = f"screenshots/{test_name}_{timestamp}.png"
        
        # Take and save screenshot
        driver.save_screenshot(filename)
        logger.error(f"❌ Test FAILED - screenshot saved: {filename}")
    
    driver.quit()

# required pytest hook to capture failure info
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
    outcome = yield
    item.rep_call = outcome.get_result() if call.when == "call" else None

# ── TEST WITH BUILT-IN LOGGING ────────────────────────────────
def test_login_flow(driver):
    logger.info("Starting login test")
    
    driver.get("https://staging.app.com/login")
    logger.info(f"Navigated to: {driver.current_url}")
    
    email_field = driver.find_element("id", "email")
    email_field.send_keys("alice@test.com")
    logger.info("Entered email address")
    
    password_field = driver.find_element("id", "password")
    password_field.send_keys("Test@1234")
    
    driver.find_element("id", "submit").click()
    logger.info("Clicked login button")
    
    assert "/dashboard" in driver.current_url,         f"Expected redirect to dashboard, got: {driver.current_url}"
    logger.info("✅ Login successful - on dashboard")

# ── ELEMENT-LEVEL SCREENSHOT ──────────────────────────────────
def screenshot_element(driver, element, filename):
    """Take screenshot of a specific element only"""
    element.screenshot(filename)  # Selenium 4+ feature