20、使用 pytest 和 HTTPX 异步测试 API

flink9streamer

于 2025-09-19 12:17:36 发布

阅读量52

点赞数

CC 4.0 BY-SA版权

分类专栏： FastAPI构建高效数据科学应用文章标签： pytest HTTPX FastAPI

本文链接：https://blog.youkuaiyun.com/flink9streamer/article/details/152360932

FastAPI构建高效数据科学应用专栏收录该内容

32 篇文章 ¥499.90

订阅专栏¥69.90

会员秒杀 ¥9.9 重磅福利

超级会员免费看

使用 pytest 和 HTTPX 异步测试 API

在开发 API 时，测试是确保其稳定性和正确性的关键步骤。本文将介绍如何使用 pytest 和 HTTPX 异步测试 FastAPI 应用程序的不同类型端点，包括 REST API 端点、涉及数据库的端点以及 WebSocket 端点。

1. 编写 REST API 端点测试

测试 FastAPI 应用程序所需的工具已准备就绪，所有测试都可归结为执行 HTTP 请求并检查响应是否符合预期。

1.1 测试 `hello_world` 路径操作函数

以下是一个简单的测试示例：

# chapter09_app_test.py
import pytest
import httpx
from fastapi import status

@pytest.mark.asyncio
async def test_hello_world(test_client: httpx.AsyncClient):
    response = await test_client.get("/")
    assert response.status_code == status.HTTP_200_OK
    json = response.json()
    assert json == {"hello": "world"}

操作步骤如下：
1. 定义异步测试函数，并使用 pytest.mark.asyncio 装饰器。
2. 请求