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

陌墨酱_4044

陌墨酱_4044

2026-04-25

840人浏览

原创

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,速度快、可控性强,适合验证逻辑和数据流。

  • 不要在测试中启动 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

示例(验证登录失败):

from fastapi.testclient import TestClient
from app.main import app
<p>client = TestClient(app)</p><div class="aritcle_card flexRow artxards">
											<div class="artcardd flexRow">
												<a class="aritcle_card_img" rel="nofollow" href="/xiazai/skill9965" title="macOS UI 自动化技能包 —— 用 Python 控制你的 Mac"><img
														src="https://img.php.cn/upload/skill/001/246/273/6adaf20dbae7c507.jpg" alt="macOS UI 自动化技能包 —— 用 Python 控制你的 Mac" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
												<div class="aritcle_card_info flexColumn">
													<a rel="nofollow" href="/xiazai/skill9965" title="macOS UI 自动化技能包 —— 用 Python 控制你的 Mac" class="overflowclass">macOS UI 自动化技能包 —— 用 Python 控制你的 Mac</a>
													<p class="overflowclass">让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript

系统要求
macOS 14+ (Sonoma)

Python 3.9+

需要开启:辅助功能 + 屏幕录制权限</p>
												</div>
												<a rel="nofollow" href="/xiazai/skill9965" title="macOS UI 自动化技能包 —— 用 Python 控制你的 Mac" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
												</a>
											</div>
										</div><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"]</p>

用 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 的请求节奏贴近真实用户、有没有把基础设施的配置瓶颈提前揪出来。

大量免费API接口:立即使用
涵盖生活服务API、金融科技API、企业工商API、等相关的API接口服务。免费API接口可安全、合规地连接上下游,为数据API应用能力赋能!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2023.08.11

2743

5

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

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

2024.03.19

6910

13

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

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

2024.03.19

6386

10

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

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

2023.07.20

1831

4

python能做什么
python能做什么

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

2023.07.25

4944

7

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

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

2023.07.31

1789

3

python教程
python教程

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

2023.08.03

29537

23

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

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

2023.08.04

3527

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3527

5

热门下载

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

精品课程

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

共0课时 | 0人学习

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

共0课时 | 0人学习

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

共0课时 | 0人学习