Flask单元测试应优先使用test_client而非requests,确保走WSGI链路;需设TESTING=True,正确mock使用位置,用function级fixture隔离app实例,断言前先验状态码再安全解析JSON。

用 test_client 发起模拟 HTTP 请求最直接
Flask 自带的 test_client 是单元测试的起点,它不启动真实服务器,直接走 WSGI 调用链路,速度快、隔离性好。别去折腾 requests 模拟外部调用——那会绕过 Flask 的上下文和中间件,测不到真实行为。
实操建议:
立即学习“Python免费学习笔记(深入)”;
- 在
pytest的fixture中初始化app.test_client(),并确保app.config['TESTING'] = True - GET 请求用
client.get('/api/user'),POST 带 JSON 用client.post('/login', json={'user': 'a', 'pwd': 'b'}) - 需要携带 Cookie 或 Header?直接传参:
client.get('/admin', headers={'Authorization': 'Bearer xyz'}) - 注意:默认不保留 session,如需跨请求保持登录态,得手动设置
client.set_cookie()或用app.test_client(use_cookies=True)
断言响应状态码和 JSON 数据要分两步写
常见错误是拿到 response 后只检查 response.status_code,却忽略内容解析失败导致后续断言静默跳过。Flask 的 response.get_json() 在非 JSON 响应(比如 404 页面返回 HTML)时返回 None,直接 assert resp.json['msg'] == 'ok' 会抛 TypeError,掩盖真正问题。
实操建议:
立即学习“Python免费学习笔记(深入)”;
- 先断言状态码:
assert response.status_code == 200 - 再安全取 JSON:
data = response.get_json(),然后assert data is not None,再断言字段 - 如果接口可能返回非 JSON(如重定向、错误页面),用
response.data.decode()+ 字符串断言更稳妥 - 对 4xx/5xx 响应,别假设一定有
json;有些错误路径直接abort(400)返回纯文本,get_json()就是None
mock 数据库或外部依赖必须 patch 对象所在位置
用 unittest.mock.patch 或 pytest-mock 时,90% 的失败是因为 patch 错了导入位置。不是“哪里定义就去哪里 patch”,而是“哪里使用就去哪里 patch”。比如你在 views.py 里写了 from models import User; User.query.filter(...),那就要 @patch('views.User'),而不是 @patch('models.User')。
实操建议:
立即学习“Python免费学习笔记(深入)”;
- 在测试函数参数中接收 mock 对象,例如
def test_get_user(mocker): user_mock = mocker.patch('views.User') - 设返回值:
user_mock.query.filter.return_value.first.return_value = mock_user - 避免 patch 整个模块;优先 patch 具体类或函数,减小副作用
- 数据库操作测试完记得清理:用
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'配内存 DB,或每次测试后db.drop_all(); db.create_all()
pytest 运行时要注意 app 生命周期和 fixture 作用域
一个被忽略的坑:把 app 实例写死在模块顶层,或在 session 级 fixture 中反复修改配置(比如切换 SQLALCHEMY_DATABASE_URI),会导致后续测试用错配置甚至连接泄漏。Flask 的 app 不是线程安全的,pytest 并行运行时更危险。
实操建议:
立即学习“Python免费学习笔记(深入)”;
- 用
function级 fixture 创建干净的app实例,每次测试独享 - 需要共享 DB 连接?改用
session级 fixture 初始化内存 DB,但所有测试必须显式使用同一套模型表结构 - 避免在
conftest.py中全局 importapp;用 fixture 注入,让依赖显性化 - 如果用了工厂函数(
create_app()),测试中务必调用它,而不是复用开发环境的app
request 或 g)、mock 位置错位、以及测试间状态污染——这些不会报错,但会让断言通过得莫名其妙。


















