FastAPI(61)- 异步测试
FastAPI(61)- 异步测试
- 前面讲解了通过 pytest 进行单元测试,是针对同步函数的:
- 但它无法再 pytest 中测试或运行任何异步函数
- 能够在测试中使用异步函数可能很有用
- 例如,当异步查询数据库时,假设想要测试向 FastAPI 应用程序发送请求,然后验证后端是否成功在数据库中写入了正确的数据,同时使用异步数据库
FastAPI 代码
from fastapi import FastAPI
app = FastAPI()
@app.get("/")
async def root():
return {"message": "Tomato"}
单元测试代码
- 即使 FastAPI 应用程序使用普通 def 函数而不是 async def,它仍然是一个异步应用程序
- TestClient 在内部使用标准 pytest 在正常 def 测试函数中调用异步 FastAPI 应用程序做了一些魔术
- 但是当在异步函数中使用调用异步 FastAPI 应用程序时,这种魔法就不再起作用了
- 通过异步运行测试用例,不能再在测试函数中使用 TestClient,此时有一个不错的替代方案,称为 HTTPX
- HTTPX 是 Python 3 的 HTTP 客户端,它允许像使用 TestClient 一样查询 FastAPI 应用程序
- HTTPX 的 API 和 requests 库几乎相同
- 重要的区别:用 HTTPX 不仅限于同步,还可以发出异步请求
@pytest.mark.anyio
- 通过使用 FastAPI app 创建一个 AsyncClient,并使用 await 向它发送异步请求
- 需要搭配 async/await 一起使用
- 本文作者: 小菠萝测试笔记
- 本文链接: