Selenium Architecture & Setup

# ══════════════════════════════════════════════════════════════
# SETUP: Install Selenium + Webdriver Manager
# ══════════════════════════════════════════════════════════════
# pip install selenium webdriver-manager pytest

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from webdriver_manager.chrome import ChromeDriverManager

# ── Basic Setup ──────────────────────────────────────────────
def get_driver(headless=False):
    options = Options()
    if headless:
        options.add_argument("--headless")       # Run without GUI (CI environments)
    options.add_argument("--no-sandbox")          # Required in Linux CI
    options.add_argument("--disable-dev-shm-usage")  # Required in Docker
    options.add_argument("--window-size=1920,1080")
    
    # WebDriverManager auto-downloads matching ChromeDriver version
    service = Service(ChromeDriverManager().install())
    driver = webdriver.Chrome(service=service, options=options)
    driver.implicitly_wait(10)  # Global wait for element visibility
    return driver

# ── First Test - navigates, finds element, verifies title ─────
def test_google_homepage():
    driver = get_driver()
    try:
        driver.get("https://www.google.com")
        assert "Google" in driver.title, f"Expected 'Google' in title, got: {driver.title}"
        print(f"✅ Page title: {driver.title}")
    finally:
        driver.quit()  # Always quit to release browser resources

# ── pytest conftest.py - shared driver fixture ────────────────
# conftest.py:
import pytest

@pytest.fixture(scope="function")  # New driver per test function
def driver():
    driver = get_driver(headless=True)
    yield driver        # Test runs here
    driver.quit()       # Cleanup after test (even if test fails)

# ── Using the fixture in tests ─────────────────────────────────
def test_page_loads(driver):
    driver.get("https://example.com")
    assert driver.title == "Example Domain"

def test_page_has_heading(driver):
    driver.get("https://example.com")
    heading = driver.find_element("tag name", "h1")
    assert heading.text == "Example Domain"