virtualenv venv
# Windows
venv\Scripts\activate
# macOS/Linux
source venv/bin/activate

# test_math_utils.py
import pytest

# 待测试的工具函数
def add(a: int, b: int) -> int:
    return a + b

def divide(a: float, b: float) -> float:
    if b == 0:
        raise ValueError("除数不能为0")
    return a / b

# 测试用例
def test_add_positive_numbers():
    """测试两个正数相加"""
    assert add(2, 3) == 5

def test_add_negative_numbers():
    """测试两个负数相加"""
    assert add(-1, -1) == -2

def test_divide_normal():
    """测试正常除法"""
    assert divide(10, 2) == 5.0

def test_divide_zero():
    """测试除以0的异常情况"""
    with pytest.raises(ValueError, match="除数不能为0"):
        divide(10, 0)

# 参数化测试：一次测试多组数据
@pytest.mark.parametrize("a, b, expected", [
    (0, 0, 0),
    (100, 200, 300),
    (-5, 5, 0),
    (3, -3, 0),
])
def test_add_parametrized(a, b, expected):
    """参数化测试加法函数"""
    assert add(a, b) == expected

# test_api.py
import requests
import pytest

BASE_URL = "https://jsonplaceholder.typicode.com"

def test_get_posts():
    """测试获取文章列表接口"""
    response = requests.get(f"{BASE_URL}/posts")
    # 断言状态码为200
    assert response.status_code == 200
    # 断言返回内容是JSON格式
    posts = response.json()
    # 断言返回至少10篇文章
    assert len(posts) >= 10
    # 断言第一篇文章有id和title字段
    assert "id" in posts[0]
    assert "title" in posts[0]

def test_create_post():
    """测试创建新文章接口"""
    payload = {
        "title": "Python自动化测试教程",
        "body": "这是一篇关于Python自动化测试的文章",
        "userId": 1
    }
    response = requests.post(f"{BASE_URL}/posts", json=payload)
    assert response.status_code == 201
    new_post = response.json()
    assert new_post["title"] == payload["title"]
    assert new_post["body"] == payload["body"]
    assert "id" in new_post

def test_get_post_by_id():
    """测试根据ID获取单篇文章"""
    post_id = 1
    response = requests.get(f"{BASE_URL}/posts/{post_id}")
    assert response.status_code == 200
    post = response.json()
    assert post["id"] == post_id
    assert "userId" in post

# test_ui_playwright.py
from playwright.sync_api import Page, expect
import pytest

def test_baidu_search(page: Page):
    """测试百度搜索功能"""
    # 打开百度首页
    page.goto("https://www.baidu.com")
    # 验证页面标题包含"百度"
    expect(page).to_have_title("百度一下，你就知道")
    # 在搜索框输入"Python自动化测试"
    page.locator("#kw").fill("Python自动化测试")
    # 点击搜索按钮
    page.locator("#su").click()
    # 等待搜索结果加载完成
    page.wait_for_selector("#content_left")
    # 验证搜索结果包含相关内容
    expect(page.locator("#content_left")).to_contain_text("Python自动化测试")
    # 截图保存结果
    page.screenshot(path="search_result.png", full_page=True)

def test_douban_movie(page: Page):
    """测试豆瓣电影排行榜"""
    page.goto("https://movie.douban.com/chart")
    # 验证页面标题
    expect(page).to_have_title("豆瓣电影排行榜")
    # 获取排名前10的电影
    movies = page.locator(".item").all()[:10]
    assert len(movies) == 10
    # 打印前10名电影名称
    for i, movie in enumerate(movies, 1):
        title = movie.locator(".pl2 a").first.text_content().strip()
        print(f"第{i}名：{title}")

# locustfile.py
from locust import HttpUser, task, between, tag

class WebsiteUser(HttpUser):
    # 每个用户之间等待1-3秒
    wait_time = between(1, 3)
    # 测试的目标域名
    host = "https://www.baidu.com"

    @tag("search")
    @task(3)  # 权重3，执行频率是其他任务的3倍
    def search_test(self):
        """搜索测试"""
        self.client.get("/s?wd=Python自动化测试")

    @tag("homepage")
    @task(1)
    def visit_homepage(self):
        """访问首页"""
        self.client.get("/")

    def on_start(self):
        """每个用户启动时执行的操作"""
        print("新用户开始访问")

    def on_stop(self):
        """每个用户结束时执行的操作"""
        print("用户结束访问")

# test_compatibility.py
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
import pytest

# 测试不同浏览器
@pytest.mark.parametrize("browser", ["chrome", "firefox", "edge"])
def test_browser_compatibility(browser):
    """测试网站在不同浏览器下的兼容性"""
    driver = None
    try:
        # 初始化不同浏览器的驱动
        if browser == "chrome":
            driver = webdriver.Chrome()
        elif browser == "firefox":
            driver = webdriver.Firefox()
        elif browser == "edge":
            driver = webdriver.Edge()
        
        driver.maximize_window()
        driver.get("https://www.baidu.com")
        
        # 等待搜索框出现
        search_box = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.ID, "kw"))
        )
        search_box.send_keys(f"兼容性测试 - {browser}")
        
        # 点击搜索按钮
        search_btn = driver.find_element(By.ID, "su")
        search_btn.click()
        
        # 等待结果加载
        WebDriverWait(driver, 10).until(
            EC.title_contains("兼容性测试")
        )
        
        # 验证页面加载成功
        assert f"兼容性测试 - {browser}" in driver.title
        print(f"{browser} 浏览器测试通过")
        
    finally:
        if driver:
            driver.quit()

# test_security.py
import requests
import re
from urllib.parse import urljoin, urlparse

def test_sql_injection(url: str):
    """测试SQL注入漏洞"""
    payloads = ["'", "' OR '1'='1", "' OR 1=1--", "\" OR \"1\"=\"1"]
    vulnerable = False
    
    for payload in payloads:
        test_url = f"{url}?id={payload}"
        try:
            response = requests.get(test_url, timeout=5)
            # 检查响应中是否包含SQL错误信息
            if any(error in response.text.lower() for error in ["sql syntax", "mysql error", "postgresql error", "ora-"]):
                print(f"[!] 可能存在SQL注入漏洞: {test_url}")
                vulnerable = True
        except Exception as e:
            print(f"[-] 请求失败: {e}")
    
    return vulnerable

def test_xss_vulnerability(url: str):
    """测试XSS跨站脚本漏洞"""
    payload = ""
    test_url = f"{url}?q={payload}"
    
    try:
        response = requests.get(test_url, timeout=5)
        if payload in response.text:
            print(f"[!] 可能存在XSS漏洞: {test_url}")
            return True
    except Exception as e:
        print(f"[-] 请求失败: {e}")
    
    return False

def test_https_security(url: str):
    """测试HTTPS配置是否安全"""
    parsed = urlparse(url)
    if parsed.scheme != "https":
        print(f"[!] 网站未使用HTTPS协议")
        return False
    
    try:
        response = requests.get(url, timeout=5, verify=True)
        if response.status_code == 200:
            print(f"[+] HTTPS配置正常")
            return True
    except requests.exceptions.SSLError:
        print(f"[!] SSL证书无效或过期")
        return False
    
    return False

if __name__ == "__main__":
    target_url = "http://example.com"  # 替换为要测试的网站
    print(f"开始安全测试: {target_url}")
    test_sql_injection(target_url)
    test_xss_vulnerability(target_url)
    test_https_security(target_url)

# .github/workflows/test.yml
name: 自动化测试

on:
  push:
    branches: [ main, develop ]  # 推送到main或develop分支时触发
  pull_request:
    branches: [ main ]  # PR到main分支时触发
  schedule:
    - cron: '0 1 * * *'  # 每天凌晨1点自动执行

jobs:
  test:
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ["3.10", "3.11", "3.12"]  # 测试多个Python版本

    steps:
    - uses: actions/checkout@v4
    - name: 设置Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.python-version }}
        cache: 'pip'
    
    - name: 安装依赖
      run: |
        python -m pip install --upgrade pip
        pip install pytest requests playwright pytest-playwright
        playwright install --with-deps chromium
    
    - name: 运行单元测试
      run: |
        pytest tests/unit -v --junitxml=unit-test-results.xml
    
    - name: 运行接口测试
      run: |
        pytest tests/api -v --junitxml=api-test-results.xml
    
    - name: 运行UI测试
      run: |
        pytest tests/ui -v --junitxml=ui-test-results.xml
    
    - name: 上传测试结果
      uses: actions/upload-artifact@v4
      if: always()  # 即使测试失败也上传
      with:
        name: test-results
        path: '*-test-results.xml'

# test_allure_report.py
import allure
import pytest
from datetime import datetime

@allure.feature("用户管理模块")
@allure.story("用户登录功能")
class TestUserLogin:
    
    @allure.title("正常登录测试")
    @allure.severity(allure.severity_level.CRITICAL)
    def test_login_success(self):
        """测试使用正确的用户名和密码登录"""
        with allure.step("步骤1：输入正确的用户名"):
            print("输入用户名：admin")
        with allure.step("步骤2：输入正确的密码"):
            print("输入密码：123456")
        with allure.step("步骤3：点击登录按钮"):
            print("点击登录")
        with allure.step("步骤4：验证登录成功"):
            assert True, "登录成功"
        # 附加测试截图
        allure.attach.file("search_result.png", name="登录成功截图", attachment_type=allure.attachment_type.PNG)

    @allure.title("密码错误登录测试")
    @allure.severity(allure.severity_level.NORMAL)
    def test_login_wrong_password(self):
        """测试使用错误的密码登录"""
        with allure.step("步骤1：输入正确的用户名"):
            print("输入用户名：admin")
        with allure.step("步骤2：输入错误的密码"):
            print("输入密码：wrongpass")
        with allure.step("步骤3：点击登录按钮"):
            print("点击登录")
        with allure.step("步骤4：验证登录失败，提示密码错误"):
            assert False, "密码错误"  # 模拟测试失败

    @allure.title("用户不存在登录测试")
    @allure.severity(allure.severity_level.NORMAL)
    def test_login_user_not_exist(self):
        """测试使用不存在的用户名登录"""
        with allure.step("步骤1：输入不存在的用户名"):
            print("输入用户名：nonexist")
        with allure.step("步骤2：输入任意密码"):
            print("输入密码：123456")
        with allure.step("步骤3：点击登录按钮"):
            print("点击登录")
        with allure.step("步骤4：验证登录失败，提示用户不存在"):
            assert False, "用户不存在"  # 模拟测试失败

@allure.feature("商品管理模块")
@allure.story("商品查询功能")
def test_product_search():
    """测试商品搜索功能"""
    with allure.step("输入商品关键词"):
        print("输入关键词：Python自动化测试")
    with allure.step("点击搜索按钮"):
        print("点击搜索")
    with allure.step("验证搜索结果"):
        assert True, "搜索结果正确"