# 端到端测试：完整的购物流程
def test_complete_shopping_flow():
    # 步骤 1：用户登录
    driver.get("https://example.com/login")
    driver.find_element(By.ID, "username").send_keys("test@example.com")
    driver.find_element(By.ID, "password").send_keys("password123")
    driver.find_element(By.ID, "login-btn").click()
    
    # 验证登录成功
    assert "欢迎" in driver.page_source
    
    # 步骤 2：搜索商品
    driver.find_element(By.ID, "search-box").send_keys("iPhone 15")
    driver.find_element(By.ID, "search-btn").click()
    
    # 验证搜索结果
    assert "iPhone 15" in driver.page_source
    
    # 步骤 3：加入购物车
    driver.find_element(By.CLASS_NAME, "add-to-cart").click()
    
    # 验证购物车数量
    cart_count = driver.find_element(By.ID, "cart-count").text
    assert cart_count == "1"
    
    # 步骤 4：结算
    driver.find_element(By.ID, "checkout-btn").click()
    
    # 步骤 5：填写地址
    driver.find_element(By.ID, "address").send_keys("北京市朝阳区xxx")
    driver.find_element(By.ID, "phone").send_keys("13800138000")
    
    # 步骤 6：选择支付方式
    driver.find_element(By.ID, "payment-wechat").click()
    
    # 步骤 7：提交订单
    driver.find_element(By.ID, "submit-order").click()
    
    # 验证订单创建成功
    assert "订单提交成功" in driver.page_source
    
    # 步骤 8：验证数据库
    order = db.query("SELECT * FROM orders WHERE user_id = 123 ORDER BY created_at DESC LIMIT 1")
    assert order.status == "pending_payment"
    assert order.total > 0

# 测试用户体验
def test_user_experience():
    # 测试页面加载速度
    start_time = time.time()
    driver.get("https://example.com")
    load_time = time.time() - start_time
    
    # 验证加载时间 < 3 秒
    assert load_time < 3, f"页面加载时间过长：{load_time}秒"
    
    # 测试按钮可点击
    add_to_cart_btn = driver.find_element(By.ID, "add-to-cart")
    assert add_to_cart_btn.is_displayed()
    assert add_to_cart_btn.is_enabled()
    
    # 测试表单验证
    driver.find_element(By.ID, "email").send_keys("invalid-email")
    driver.find_element(By.ID, "submit").click()
    
    # 验证错误提示
    error_msg = driver.find_element(By.CLASS_NAME, "error-message").text
    assert "请输入有效的邮箱地址" in error_msg

# 1. 并行执行测试
pytest -n 4  # 使用 4 个进程并行执行

# 2. 只测试关键流程
@pytest.mark.critical
def test_critical_flow():
    # 只测试核心业务流程
    pass

# 3. 使用无头浏览器
options = webdriver.ChromeOptions()
options.add_argument('--headless')  # 无头模式，更快
driver = webdriver.Chrome(options=options)

# 4. 复用浏览器会话
@pytest.fixture(scope="session")
def driver():
    driver = webdriver.Chrome()
    yield driver
    driver.quit()

# 1. 使用 Page Object Model（POM）
class LoginPage:
    def __init__(self, driver):
        self.driver = driver
        self.username_input = (By.ID, "username")
        self.password_input = (By.ID, "password")
        self.login_btn = (By.ID, "login-btn")

    def login(self, username, password):
        self.driver.find_element(*self.username_input).send_keys(username)
        self.driver.find_element(*self.password_input).send_keys(password)
        self.driver.find_element(*self.login_btn).click()

# 使用
login_page = LoginPage(driver)
login_page.login("test@example.com", "password123")

# 2. 使用数据驱动测试
@pytest.mark.parametrize("username,password", [
    ("test1@example.com", "password1"),
    ("test2@example.com", "password2"),
    ("test3@example.com", "password3"),
])
def test_login(username, password):
    login_page.login(username, password)
    assert "欢迎" in driver.page_source

# 3. 使用测试数据工厂
class UserFactory:
    @staticmethod
    def create_test_user():
        return {
            "username": f"test_{uuid.uuid4()}@example.com",
            "password": "password123",
            "name": "测试用户"
        }

user = UserFactory.create_test_user()

# 测试电商 Web 应用
def test_horizontal_e2e():
    # 1. 登录
    login_page.login("test@example.com", "password123")

    # 2. 搜索商品
    search_page.search("iPhone 15")

    # 3. 加入购物车
    product_page.add_to_cart()

    # 4. 结算
    cart_page.checkout()

    # 5. 支付
    payment_page.pay_with_wechat()

    # 6. 验证订单
    assert order_page.get_order_status() == "pending_payment"

# 测试电商系统（Web + 移动 App + 后台管理）
def test_vertical_e2e():
    # 1. 用户在 Web 下单
    web_driver.get("https://example.com")
    web_driver.find_element(By.ID, "add-to-cart").click()
    web_driver.find_element(By.ID, "checkout").click()

    # 2. 管理员在后台处理订单
    admin_driver.get("https://admin.example.com")
    admin_driver.find_element(By.ID, "orders").click()
    admin_driver.find_element(By.ID, "approve").click()

    # 3. 用户在移动 App 查看订单状态
    mobile_driver.find_element(By.ID, "my-orders").click()
    status = mobile_driver.find_element(By.ID, "order-status").text
    assert status == "已发货"

def test_user_registration_flow():
    """
    测试用例 ID: E2E-001
    测试目标: 验证用户注册流程
    前置条件: 无
    测试步骤:
      1. 打开注册页面
      2. 填写用户名、邮箱、密码
      3. 点击注册按钮
      4. 验证邮箱
      5. 登录
    预期结果: 用户注册成功并自动登录
    """
    # 步骤 1
    driver.get("https://example.com/register")

    # 步骤 2
    driver.find_element(By.ID, "username").send_keys("testuser")
    driver.find_element(By.ID, "email").send_keys("test@example.com")
    driver.find_element(By.ID, "password").send_keys("password123")

    # 步骤 3
    driver.find_element(By.ID, "register-btn").click()

    # 步骤 4
    # 模拟邮箱验证
    verification_code = get_verification_code_from_email("test@example.com")
    driver.find_element(By.ID, "code").send_keys(verification_code)
    driver.find_element(By.ID, "verify-btn").click()

    # 步骤 5
    assert "欢迎" in driver.page_source
    assert driver.current_url == "https://example.com/home"

import pytest
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class TestE2E:
    @pytest.fixture(scope="class")
    def driver(self):
        options = webdriver.ChromeOptions()
        options.add_argument('--headless')
        driver = webdriver.Chrome(options=options)
        driver.implicitly_wait(10)
        yield driver
        driver.quit()

    def test_complete_shopping_flow(self, driver):
        # 1. 登录
        driver.get("https://example.com/login")
        driver.find_element(By.ID, "username").send_keys("test@example.com")
        driver.find_element(By.ID, "password").send_keys("password123")
        driver.find_element(By.ID, "login-btn").click()

        # 等待登录成功
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.ID, "user-menu"))
        )

        # 2. 搜索商品
        driver.find_element(By.ID, "search-box").send_keys("iPhone 15")
        driver.find_element(By.ID, "search-btn").click()

        # 等待搜索结果
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.CLASS_NAME, "product-item"))
        )

        # 3. 点击第一个商品
        driver.find_element(By.CLASS_NAME, "product-item").click()

        # 4. 加入购物车
        driver.find_element(By.ID, "add-to-cart").click()

        # 等待加入成功
        WebDriverWait(driver, 10).until(
            EC.text_to_be_present_in_element((By.ID, "cart-count"), "1")
        )

        # 5. 进入购物车
        driver.find_element(By.ID, "cart-icon").click()

        # 6. 结算
        driver.find_element(By.ID, "checkout-btn").click()

        # 7. 填写地址
        driver.find_element(By.ID, "address").send_keys("北京市朝阳区xxx")
        driver.find_element(By.ID, "phone").send_keys("13800138000")

        # 8. 选择支付方式
        driver.find_element(By.ID, "payment-wechat").click()

        # 9. 提交订单
        driver.find_element(By.ID, "submit-order").click()

        # 10. 验证订单创建成功
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.CLASS_NAME, "success-message"))
        )

        success_msg = driver.find_element(By.CLASS_NAME, "success-message").text
        assert "订单提交成功" in success_msg

from playwright.sync_api import sync_playwright

def test_shopping_flow():
    with sync_playwright() as p:
        # 启动浏览器
        browser = p.chromium.launch(headless=True)
        page = browser.new_page()

        # 1. 登录
        page.goto("https://example.com/login")
        page.fill("#username", "test@example.com")
        page.fill("#password", "password123")
        page.click("#login-btn")

        # 2. 搜索商品
        page.fill("#search-box", "iPhone 15")
        page.click("#search-btn")

        # 3. 加入购物车
        page.click(".product-item:first-child")
        page.click("#add-to-cart")

        # 4. 验证购物车数量
        cart_count = page.text_content("#cart-count")
        assert cart_count == "1"

        browser.close()

# pages/login_page.py
class LoginPage:
    def __init__(self, page):
        self.page = page
        self.username_input = "#username"
        self.password_input = "#password"
        self.login_btn = "#login-btn"

    def login(self, username, password):
        self.page.fill(self.username_input, username)
        self.page.fill(self.password_input, password)
        self.page.click(self.login_btn)

# tests/test_login.py
def test_login():
    login_page = LoginPage(page)
    login_page.login("test@example.com", "password123")
    assert "欢迎" in page.content()

# 1. 详细的日志
import logging

def test_shopping_flow():
    logging.info("步骤 1：用户登录")
    driver.get("https://example.com/login")
    driver.find_element(By.ID, "username").send_keys("test@example.com")
    driver.find_element(By.ID, "password").send_keys("password123")
    driver.find_element(By.ID, "login-btn").click()
    logging.info("登录成功")
    
    logging.info("步骤 2：搜索商品")
    driver.find_element(By.ID, "search-box").send_keys("iPhone 15")
    driver.find_element(By.ID, "search-btn").click()
    logging.info("搜索成功")
    
    # ...

# 2. 截图
def test_shopping_flow():
    try:
        # 测试步骤
        pass
    except Exception as e:
        # 失败时截图
        driver.save_screenshot(f"error_{time.time()}.png")
        raise

# 3. 录屏
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

caps = DesiredCapabilities.CHROME.copy()
caps['goog:loggingPrefs'] = {'browser': 'ALL'}
driver = webdriver.Chrome(desired_capabilities=caps)

# 4. 网络请求日志
logs = driver.get_log('performance')
for log in logs:
    print(log)