# 系统测试：完整的电商系统测试
class TestEcommerceSystem:
    
    def test_complete_system(self):
        """测试完整的电商系统"""
        
        # 1. 测试用户注册
        response = requests.post("https://api.example.com/register", json={
            "username": "testuser",
            "email": "test@example.com",
            "password": "password123"
        })
        assert response.status_code == 201
        user_id = response.json()["user_id"]
        
        # 2. 测试用户登录
        response = requests.post("https://api.example.com/login", json={
            "email": "test@example.com",
            "password": "password123"
        })
        assert response.status_code == 200
        token = response.json()["token"]
        
        # 3. 测试商品搜索
        response = requests.get(
            "https://api.example.com/products/search",
            params={"q": "iPhone 15"},
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 200
        products = response.json()["products"]
        assert len(products) > 0
        
        # 4. 测试加入购物车
        product_id = products[0]["id"]
        response = requests.post(
            "https://api.example.com/cart/add",
            json={"product_id": product_id, "quantity": 1},
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 200
        
        # 5. 测试下单
        response = requests.post(
            "https://api.example.com/orders",
            json={
                "address": "北京市朝阳区xxx",
                "phone": "13800138000",
                "payment_method": "wechat"
            },
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 201
        order_id = response.json()["order_id"]
        
        # 6. 测试支付
        response = requests.post(
            f"https://api.example.com/orders/{order_id}/pay",
            json={"payment_method": "wechat"},
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 200
        
        # 7. 验证订单状态
        response = requests.get(
            f"https://api.example.com/orders/{order_id}",
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 200
        assert response.json()["status"] == "paid"
        
        # 8. 验证数据库
        db = connect_to_database()
        order = db.query(f"SELECT * FROM orders WHERE id = {order_id}")
        assert order.status == "paid"
        assert order.user_id == user_id

# 测试订单和库存的集成
def test_order_inventory_integration():
    """测试下单时库存是否正确扣减"""
    
    # 1. 获取商品库存
    response = requests.get("https://api.example.com/products/123")
    initial_stock = response.json()["stock"]
    
    # 2. 下单
    response = requests.post(
        "https://api.example.com/orders",
        json={
            "product_id": 123,
            "quantity": 2
        },
        headers={"Authorization": f"Bearer {token}"}
    )
    assert response.status_code == 201
    
    # 3. 验证库存扣减
    response = requests.get("https://api.example.com/products/123")
    new_stock = response.json()["stock"]
    
    # ✅ 系统测试能发现库存扣减是否正确
    assert new_stock == initial_stock - 2

# 性能测试
def test_performance():
    """测试系统性能"""
    
    start_time = time.time()
    response = requests.get("https://api.example.com/products")
    end_time = time.time()
    
    # 验证响应时间 < 1 秒
    assert end_time - start_time < 1
    
    # 验证吞吐量
    assert response.json()["total"] > 1000

# 安全测试
def test_security():
    """测试系统安全性"""
    
    # 测试未授权访问
    response = requests.get("https://api.example.com/orders")
    assert response.status_code == 401
    
    # 测试 SQL 注入
    response = requests.get(
        "https://api.example.com/products/search",
        params={"q": "'; DROP TABLE products; --"}
    )
    assert response.status_code == 200
    assert "error" not in response.json()

# 兼容性测试
def test_compatibility():
    """测试浏览器兼容性"""
    
    browsers = ["chrome", "firefox", "safari", "edge"]
    
    for browser in browsers:
        driver = webdriver.Chrome() if browser == "chrome" else webdriver.Firefox()
        driver.get("https://example.com")
        
        # 验证页面正常加载
        assert "电商平台" in driver.title
        
        driver.quit()

# 测试数据工厂
class TestDataFactory:
    
    @staticmethod
    def create_user(username=None, email=None):
        """创建测试用户"""
        username = username or f"user_{uuid.uuid4()}"
        email = email or f"{username}@example.com"
        
        response = requests.post("https://api.example.com/register", json={
            "username": username,
            "email": email,
            "password": "password123"
        })
        
        return response.json()
    
    @staticmethod
    def create_product(name=None, price=None):
        """创建测试商品"""
        name = name or f"Product {uuid.uuid4()}"
        price = price or random.randint(100, 10000)
        
        response = requests.post("https://api.example.com/products", json={
            "name": name,
            "price": price,
            "stock": 100
        }, headers={"Authorization": f"Bearer {admin_token}"})
        
        return response.json()
    
    @staticmethod
    def create_order(user_id, product_id, quantity=1):
        """创建测试订单"""
        response = requests.post("https://api.example.com/orders", json={
            "user_id": user_id,
            "product_id": product_id,
            "quantity": quantity
        }, headers={"Authorization": f"Bearer {token}"})
        
        return response.json()

# 使用
user = TestDataFactory.create_user()
product = TestDataFactory.create_product()
order = TestDataFactory.create_order(user["id"], product["id"])

# 1. 并行执行测试
pytest -n 8 tests/system/  # 使用 8 个进程

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

# 3. 使用测试分层
# - 冒烟测试（5分钟）：测试核心功能
# - 完整测试（1小时）：测试所有功能

# 4. 使用缓存
@pytest.fixture(scope="session")
def test_user():
    """创建一次测试用户，所有测试共享"""
    user = TestDataFactory.create_user()
    yield user
    # 测试结束后清理
    requests.delete(f"https://api.example.com/users/{user['id']}")

def test_user_registration():
    """测试用户注册功能"""
    response = requests.post("https://api.example.com/register", json={
        "username": "testuser",
        "email": "test@example.com",
        "password": "password123"
    })
    assert response.status_code == 201
    assert "user_id" in response.json()

from locust import HttpUser, task, between

class EcommerceUser(HttpUser):
    wait_time = between(1, 3)

    @task
    def browse_products(self):
        self.client.get("/products")

    @task
    def search_products(self):
        self.client.get("/products/search?q=iPhone")

# 运行：locust -f test_performance.py --users 1000 --spawn-rate 10

def test_unauthorized_access():
    """测试未授权访问"""
    response = requests.get("https://api.example.com/orders")
    assert response.status_code == 401

def test_sql_injection():
    """测试 SQL 注入"""
    response = requests.get(
        "https://api.example.com/products/search",
        params={"q": "'; DROP TABLE products; --"}
    )
    assert response.status_code == 200
    assert "error" not in response.json()

def test_browser_compatibility():
    """测试浏览器兼容性"""
    browsers = [
        webdriver.Chrome(),
        webdriver.Firefox(),
        webdriver.Safari(),
        webdriver.Edge()
    ]

    for browser in browsers:
        browser.get("https://example.com")
        assert "电商平台" in browser.title
        browser.quit()

def test_usability():
    """测试可用性"""
    driver = webdriver.Chrome()
    driver.get("https://example.com")

    # 测试搜索框是否易于找到
    search_box = driver.find_element(By.ID, "search-box")
    assert search_box.is_displayed()

    # 测试按钮是否易于点击
    add_to_cart_btn = driver.find_element(By.ID, "add-to-cart")
    assert add_to_cart_btn.size["width"] >= 100
    assert add_to_cart_btn.size["height"] >= 40

    driver.quit()

def test_reliability():
    """测试可靠性"""

    # 测试系统在数据库故障时的表现
    stop_database()

    response = requests.get("https://api.example.com/products")

    # 系统应该返回友好的错误信息，而不是崩溃
    assert response.status_code == 503
    assert "服务暂时不可用" in response.json()["message"]

    start_database()

# 1. 正常数据
normal_user = {
    "username": "testuser",
    "email": "test@example.com",
    "password": "password123"
}

# 2. 边界数据
boundary_user = {
    "username": "a" * 50,  # 最大长度
    "email": "test@example.com",
    "password": "p" * 20  # 最大长度
}

# 3. 异常数据
invalid_user = {
    "username": "",  # 空用户名
    "email": "invalid-email",  # 无效邮箱
    "password": "123"  # 密码太短
}

def test_user_registration():
    """
    测试用例 ID: SYS-001
    测试目标: 验证用户注册功能
    前置条件: 系统正常运行
    测试步骤:
      1. 发送注册请求
      2. 验证响应状态码
      3. 验证用户创建成功
      4. 验证数据库记录
    预期结果: 用户注册成功
    """
    # 步骤 1
    response = requests.post("https://api.example.com/register", json={
        "username": "testuser",
        "email": "test@example.com",
        "password": "password123"
    })

    # 步骤 2
    assert response.status_code == 201

    # 步骤 3
    assert "user_id" in response.json()
    user_id = response.json()["user_id"]

    # 步骤 4
    db = connect_to_database()
    user = db.query(f"SELECT * FROM users WHERE id = {user_id}")
    assert user is not None
    assert user.username == "testuser"

# conftest.py
import pytest

@pytest.fixture(scope="session")
def test_database():
    """创建测试数据库"""
    db = create_test_database()

    # 初始化数据
    db.execute("INSERT INTO users ...")
    db.execute("INSERT INTO products ...")

    yield db

    # 清理数据
    db.execute("DELETE FROM users")
    db.execute("DELETE FROM products")
    db.close()

@pytest.fixture(scope="function")
def test_user(test_database):
    """创建测试用户"""
    user = TestDataFactory.create_user()
    yield user
    # 测试结束后删除
    test_database.execute(f"DELETE FROM users WHERE id = {user['id']}")

import logging

logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('system_test.log'),
        logging.StreamHandler()
    ]
)

def test_shopping_flow():
    logging.info("开始测试购物流程")

    logging.info("步骤 1：用户登录")
    response = requests.post("https://api.example.com/login", ...)
    logging.info(f"登录响应：{response.json()}")

    logging.info("步骤 2：搜索商品")
    response = requests.get("https://api.example.com/products/search", ...)
    logging.info(f"搜索结果：{len(response.json()['products'])} 个商品")

    logging.info("测试完成")