讲师中心 微信公众号
AI工具推荐 视频效率加速

Python后端FastAPI怎么进行单元测试与接口压测_结合TestClient与Locust编写自动化测试

陌墨酱_4044

陌墨酱_4044

发布时间:2026-04-25 08:40:19

|

840人浏览过

|

来源于php中文网

原创

FastAPI单元测试必须用TestClient,压测必须用Locust——二者定位不同,Postman或手点无法覆盖并发、校验尖刺、异常路径等真实问题;TestClient绕过网络直接调ASGI,Locust模拟用户行为链路并需关注host、token、wait_time、阶梯加压及环境隔离。

python后端fastapi怎么进行单元测试与接口压测_结合testclient与locust编写自动化测试

FastAPI 的单元测试必须用 TestClient,压测必须用 Locust —— 二者定位完全不同,混用或跳过任一环节,线上崩得毫无预兆。

为什么不能只靠 Postman 或手点接口做验证

Postman 只能验证“单次请求通不通”,但无法覆盖:并发场景下数据库连接池耗尽、Pydantic 模型校验在高负载时的 CPU 尖刺、异常路径(如 404/422)是否被正确捕获并返回结构化 JSON。这些只有自动化测试能暴露。

常见错误现象包括:本地跑通所有接口,上线后 500 Internal Server Error 频发,日志里却只看到 asyncio.TimeoutError 或 ConnectionRefusedError —— 这往往不是代码 bug,而是没做过真实流量压力下的边界验证。

用 TestClient 写单元/集成测试的实操要点

TestClient 是 FastAPI 官方提供的同步 HTTP 测试客户端,它绕过网络栈,直接调用 ASGI app,速度快、可控性强,适合验证逻辑和数据流。

立即学习“Python免费学习笔记(深入)”;

  • 不要在测试中启动 uvicorn:直接传 app 实例给 TestClient(app),否则会引入进程、端口、异步事件循环等干扰因素
  • 路径前缀要对齐真实路由:如果用了 app.include_router(router, prefix="/api"),测试时 URL 必须写成 client.get("/api/items/123"),否则返回 404
  • 测试异常响应要显式检查:比如 response.status_code == 422 且 "detail" in response.json(),不能只断言 response.status_code == 200
  • 数据库操作必须隔离:用内存 SQLite 或 pytest 的 tmp_path 创建临时 DB 文件;避免测试间共享真实 MySQL 表,否则会出现 IntegrityError: UNIQUE constraint failed

示例(验证登录失败):

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
from fastapi.testclient import TestClient
from app.main import app
<p>client = TestClient(app)</p><p>def test_login_missing_password():
response = client.post("/api/login", json={"username": "test"})
assert response.status_code == 422
assert "password" in response.json()["detail"][0]["loc"]

用 Locust 做接口压测的关键配置项

Locust 不是“多开几个 requests.Session 就行”的玩具工具,它模拟的是真实用户行为链路,核心在于 TaskSet(或新版本的 SequentialTaskSet)和 wait_time 的设置是否贴合业务。

  • host 必须填完整 URL(含协议和端口),比如 https://api.example.com,不能只写 api.example.com,否则 GET 请求会拼出错误地址
  • 带 token 的请求,要在 on_start 中登录一次并存为实例变量:self.token = response.json()["token"],后续请求用 headers={"Authorization": f"Bearer {self.token}"}
  • wait_time = between(1, 3) 比 constant(1) 更真实:真实用户不会每秒精准发请求,随机等待能避免请求脉冲打垮服务
  • 阶梯加压必须用 --step-load 启动参数,而不是手动改 users 数字:命令形如 locust -f locustfile.py --step-load --step-users 50 --step-time 180

示例(关键片段):

from locust import HttpUser, task, between
<p>class ApiUser(HttpUser):
wait_time = between(1, 3)</p><pre class='brush:python;toolbar:false;'>def on_start(self):
    # 登录拿 token
    with self.client.post("/api/login", json={"u": "t", "p": "123"}, catch_response=True) as resp:
        if resp.status_code != 200:
            resp.failure("login failed")
        self.token = resp.json().get("token")

@task
def get_orders(self):
    self.client.get("/api/orders", headers={"Authorization": f"Bearer {self.token}"})

容易被忽略的压测准备细节

很多人跑完 Locust 报告就以为结束了,但以下三点不确认,压测结果基本无效:

  • 被测服务必须关闭 debug=True 和 reload=True:这两项会让 uvicorn 启动多个 worker 并监听文件变化,严重干扰性能指标
  • 数据库连接池大小要匹配压测并发数:比如 Locust 设了 400 用户,而 MySQL 连接池 maxconnections=10,那 90% 请求会卡在 POOL.connection() 等待,报出大量 TimeoutError,但这不是 API 层问题
  • 压测机和被测服务不能跑在同一台机器上:CPU、网络、磁盘 IO 会互相争抢,RPS 和 P95 延迟 数据完全失真;至少分开部署,最好用云厂商不同可用区的两台 ECS

真正决定压测价值的,从来不是脚本写了多少行,而是你有没有把 TestClient 的边界 case 覆盖全、有没有让 Locust 的请求节奏贴近真实用户、有没有把基础设施的配置瓶颈提前揪出来。

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

相关专题

更多
python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2743

2023.08.11

前端和后端的区别
前端和后端的区别

前端关注的是用户界面的设计和交互,而后端则注重数据处理和逻辑控制。想了解更多前端后端的相关内容,可以阅读本专题下面的文章。

6910

2024.03.19

后端的主要工作内容介绍
后端的主要工作内容介绍

后端是应用程序的服务端部分,负责核心任务,如数据库交互、业务逻辑处理和响应客户端请求。想了解更多后端的相关内容,可以阅读本专题下面的文章。

6386

2024.03.19

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1831

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4944

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1789

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

29537

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3527

2023.08.04

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

60

2026.10.28

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn