FastAPI 测试

构建基于 FastAPI 的测试平台或实现接口测试，可以充分利用其异步特性、高性能、自动生成文档以及强大的依赖注入系统。以下是一个完整的实现方案，涵盖项目初始化、核心模块设计、测试执行集成以及部署策略。 ### 3.2 模块化项目的路径管理 在构建测试平台时，良好的项目结构是关键。建议采用模块化设计，将 API 路由、测试用例、数据库模型、配置等模块分离。例如： ```python # backend/main.py from fastapi import FastAPI from .routers import test_cases, reports app = FastAPI() app.include_router(test_cases.router) app.include_router(reports.router) ``` ```python # backend/routers/test_cases.py from fastapi import APIRouter, HTTPException from pydantic import BaseModel from typing import List router = APIRouter() class TestCase(BaseModel): name: str description: str test_cases_db = [] @router.post("/test_cases/", response_model=TestCase) async def create_test_case(test_case: TestCase): test_cases_db.append(test_case) return test_case @router.get("/test_cases/", response_model=List[TestCase]) async def read_test_cases(): return test_cases_db ``` ### 3.3 生产级数据库连接管理 使用 SQLAlchemy 2.0 和异步数据库驱动（如 `asyncpg` 或 `aiosqlite`）可以实现高效的数据库连接管理。通过 `lifespan` 机制确保在应用启动时初始化数据库连接池，并在关闭时正确释放资源： ```python # backend/database.py from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession from sqlalchemy.orm import declarative_base, sessionmaker DATABASE_URL = "postgresql+asyncpg://user:password@localhost/dbname" engine = create_async_engine(DATABASE_URL, echo=True) AsyncSessionLocal = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False) Base = declarative_base() async def get_db(): async with AsyncSessionLocal() as session: yield session ``` ```python # backend/main.py from fastapi import Depends from sqlalchemy.orm import Session from .database import get_db @app.get("/items/{item_id}") async def read_item(item_id: int, db: Session = Depends(get_db)): # 使用 db 查询数据库 return {"item_id": item_id} ``` ### 3.4 测试执行集成 可以使用 `pytest` 和 `subprocess` 模块来执行测试脚本，并将结果返回给前端或存储在数据库中。例如，定义一个 API 接口触发测试执行： ```python # backend/routers/test_execution.py import subprocess from fastapi import APIRouter router = APIRouter() @router.post("/run-tests/") async def run_tests(): result = subprocess.run(["pytest", "test_cases/"], capture_output=True, text=True) return {"stdout": result.stdout, "stderr": result.stderr} ``` ### 3.5 前后端分离的静态资源部署 FastAPI 支持静态文件服务，适合部署前端资源如 HTML、CSS 和 JavaScript： ```python # backend/main.py from fastapi.staticfiles import StaticFiles app.mount("/static", StaticFiles(directory="frontend"), name="static") ``` 前端页面可以使用 Jinja2 模板引擎进行动态渲染： ```python # backend/main.py from fastapi.templating import Jinja2Templates from fastapi import Request templates = Jinja2Templates(directory="frontend") @app.get("/") async def read_index(request: Request): return templates.TemplateResponse("index.html", {"request": request}) ``` ### 3.6 异步支持与性能优化 FastAPI 天然支持异步函数，可以用于处理数据库查询、网络请求等耗时操作。例如： ```python @app.get("/async-example") async def async_example(): data = await fetch_data_from_api() return data ``` ### 3.7 错误处理与中间件 FastAPI 提供了便捷的错误处理机制和中间件支持，确保平台的健壮性： ```python @app.get("/items/{item_id}") async def read_item(item_id: str): if item_id not in items: raise HTTPException(status_code=404, detail="Item not found") return {"item": items[item_id]} ``` ```python from fastapi.middleware.trustedhost import TrustedHostMiddleware app.add_middleware( TrustedHostMiddleware, allowed_hosts=["example.com", "*.example.com"] ) ``` ### 3.8 自动化文档与接口测试 FastAPI 自动生成交互式 API 文档（Swagger UI 和 ReDoc），便于接口测试和调试： ```bash # 访问 http://localhost:8000/docs 查看交互式文档 ``` ---

在 FastAPI 中，测试接口小锁是用来限制只有测试目的的访问的。这个功能可以防止一些敏感数据或者操作被非测试人员访问和执行。当你在测试接口中使用 `@pytest.mark.usefixtures("client")` 装饰器时，FastAPI 会自动识别这个标记，并在测试期间添加一个小锁来表示这个接口只能用于测试。这个小锁不会影响接口的正常使用，只是提醒其他人不要随意访问和执行这个接口。