FastAPI 的单元测试

  • 对于服务端来说,通常会对功能进行单元测试,也称白盒测试
  • FastAPI 集成了第三方库,让我们可以快捷的编写单元测试
  • FastAPI 的单元测试是基于 Pytest + Request 的

Pytest 学习

https://www.cnblogs.com/poloyy/tag/Pytest/

TestClient 简单的栗子

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog: https://www.cnblogs.com/poloyy/
# time: 2021/9/29 10:55 下午
# file: 37_pytest.py
"""
import uvicorn
from fastapi import FastAPI
from fastapi.testclient import TestClient app = FastAPI() @app.get("/")
async def read_main():
return {"msg": "Hello World"} # 声明一个 TestClient,把 FastAPI() 实例对象传进去
client = TestClient(app) # 测试用
def test_read_main():
# 请求 127.0.0.1:8080/
response = client.get("/")
assert response.status_code == 200
assert response.json() == {"msg": "Hello World"} if __name__ == '__main__':
uvicorn.run(app="37_pytest:app", reload=True, host="127.0.0.1", port=8080)

在该文件夹下的命令行敲

pytest 37_pytest.py

运行结果

TestClient 的源码解析

继承了 requests 库的 Session

所以可以像使用 requests 库一样使用 TestClient,拥有 requests 所有方法、属性

重写了 Session.requests 方法

重写了 requests 方法,不过只是加了一句 url = urljoin(self.base_url, url) url 拼接代码,还有给函数参数都加了类型指示,更加完善啦~

自定义 websocket 连接方法

后面学到 webSocket 再详细讲他

重写了 __enter__、__exit__ 方法

  • Session 的这两个方法还是比较简陋的,TestClient 做了一次重写,主要是为了添加异步的功能(异步测试后面详解,这篇举栗子的都是普通函数 def)
  • 前面讲过有 __enter__、__exit__ 方法的对象都是上下文管理器,可以用 with .. as ..语句来调用上下文管理器

.get() 方法

上面代码 client.get(),直接调用的就是 Session 提供的 get() 方法啦!

复杂的测试场景

服务端

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog: https://www.cnblogs.com/poloyy/
# time: 2021/9/29 10:55 下午
# file: s37_pytest.py
"""
import uvicorn
from fastapi import FastAPI
from fastapi.testclient import TestClient app = FastAPI() @app.get("/")
async def read_main():
return {"msg": "Hello World"} # 声明一个 TestClient,把 FastAPI() 实例对象传进去
client = TestClient(app) # 测试用
def test_read_main():
# 请求 127.0.0.1:8080/
response = client.get("/")
assert response.status_code == 200
assert response.json() == {"msg": "Hello World"} from typing import Optional from fastapi import FastAPI, Header, HTTPException
from pydantic import BaseModel # 模拟真实 token
fake_secret_token = "coneofsilence" # 模拟真实数据库
fake_db = {
"foo": {"id": "foo", "title": "Foo", "description": "There goes my hero"},
"bar": {"id": "bar", "title": "Bar", "description": "The bartenders"},
} app = FastAPI() class Item(BaseModel):
id: str
title: str
description: Optional[str] = None # 接口一:查询数据
@app.get("/items/{item_id}", response_model=Item)
async def read_main(item_id: str, x_token: str = Header(...)):
# 1、校验 token 失败
if x_token != fake_secret_token:
raise HTTPException(status_code=400, detail="x-token 错误") # 2、若数据库没有对应数据
if item_id not in fake_db:
raise HTTPException(status_code=404, detail="找不到 item_id")
# 3、找到数据则返回
return fake_db[item_id] # 接口二:创建数据
@app.post("/items/", response_model=Item)
async def create_item(item: Item, x_token: str = Header(...)):
# 1、校验 token 失败
if x_token != fake_secret_token:
raise HTTPException(status_code=400, detail="x-token 错误") # 2、若数据库已经存在相同 id 的数据
if item.id in fake_db:
raise HTTPException(status_code=400, detail="找不到 item_id") # 3、添加数据到数据库
fake_db[item.id] = item # 4、返回添加的数据
return item if __name__ == '__main__':
uvicorn.run(app="s37_test_pytest:app", reload=True, host="127.0.0.1", port=8080)

单元测试

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog: https://www.cnblogs.com/poloyy/
# time: 2021/9/29 10:55 下午
# file: s37_pytest.py
"""
from fastapi.testclient import TestClient
from .s37_test_pytest import app client = TestClient(app) def test_read_item():
expect = {"id": "foo", "title": "Foo", "description": "There goes my hero"}
headers = {"x-token": "coneofsilence"}
resp = client.get("/items/foo", headers=headers)
assert resp.status_code == 200
assert resp.json() == expect def test_read_item_error_header():
expect = {"detail": "x-token 错误"}
headers = {"x-token": "test"}
resp = client.get("/items/foo", headers=headers)
assert resp.status_code == 400
assert resp.json() == expect def test_read_item_error_id():
expect = {"detail": "找不到 item_id"}
headers = {"x-token": "coneofsilence"}
resp = client.get("/items/foos", headers=headers)
assert resp.status_code == 404
assert resp.json() == expect def test_create_item():
body = {"id": "foos", "title": "Foo", "description": "There goes my hero"}
headers = {"x-token": "coneofsilence"}
resp = client.post("/items/", json=body, headers=headers)
assert resp.status_code == 200
assert resp.json() == body def test_create_item_error_header():
body = {"id": "foo", "title": "Foo", "description": "There goes my hero"}
expect = {"detail": "x-token 错误"}
headers = {"x-token": "test"}
resp = client.post("/items/", json=body, headers=headers)
assert resp.status_code == 400
assert resp.json() == expect def test_create_item_error_id():
expect = {"detail": "找不到 item_id"}
body = {"id": "foo", "title": "Foo", "description": "There goes my hero"}
headers = {"x-token": "coneofsilence"}
resp = client.post("/items/", json=body, headers=headers)
assert resp.status_code == 400
assert resp.json() == expect

命令行运行

pytest test.py -sq

运行结果

> pytest s37_pytest.py -sq
......
6 passed in 0.40s

FastAPI(43)- 基于 pytest + requests 进行单元测试的更多相关文章

  1. pytest+requests+Python3.7+yaml+Allure+Jenkins+docker实现接口自动化测试

    接口自动化测试框架(用例自动生成) 项目说明 本框架是一套基于pytest+requests+Python3.7+yaml+Allure+Jenkins+docker而设计的数据驱动接口自动化测试框架 ...

  2. 基于Python Requests的数据驱动的HTTP接口测试

    发表于:2017-8-30 11:56  作者:顾翔   来源:51Testing软件测试网原创 http://www.51testing.com/html/69/n-3720769-2.html   ...

  3. Appium 并发多进程基于 Pytest框架

    前言: 之前通过重写unittest的初始化方法加入设备参数进行并发,实现了基于unittest的appium多设备并发,但是考虑到unittest的框架实在过于简陋,也不方便后期的Jenkins的持 ...

  4. unit vs2017基于nunit framework创建单元测试

    unit  vs2017基于nunit framework创建单元测试 一.简叙: 单元测试大型项目中是必备的,所以不可忽视,一个项目的成败就看是否有单元测试,对后期的扩展维护都带来了便利. 二.安装 ...

  5. 基于Pytest豆瓣自动化测试【1】

    -- Pytest基础使用教程[1] 引言 Pytest 是一个非常实用的自动化测试框架,目前来说资料也是非常多了.最近某友人在学习 Python的一些测试技术,帮其网上搜了下教程:发现大多数文章多是 ...

  6. 基于Python+Requests+Pytest+YAML+Allure实现接口自动化

    本项目实现接口自动化的技术选型:Python+Requests+Pytest+YAML+Allure ,主要是针对之前开发的一个接口项目来进行学习,通过 Python+Requests 来发送和处理H ...

  7. 基于spring与mockito单元测试Mock对象注入

    转载:http://www.blogjava.net/qileilove/archive/2014/03/07/410713.html 1.关键词 单元测试.spring.mockito 2.概述 单 ...

  8. 【Pytest】python单元测试框架pytest简介

    1.Pytest介绍 pytest是python的一种单元测试框架,与python自带的unittest测试框架类似,但是比unittest框架使用起来更简洁,效率更高.根据pytest的官方网站介绍 ...

  9. 如何写好、管好单元测试?基于Roslyn+CI分析单元测试,严控产品提测质量

    上一篇文章中,我们谈到了通过Roslyn进行代码分析,通过自定义代码扫描规则,将有问题的代码.不符合编码规则的代码扫描出来,禁止签入,提升团队的代码质量. .NET Core技术研究-通过Roslyn ...

随机推荐

  1. rabbitMq可靠性投递之配置(消息至交换机,至队列不通的回调)

    @Bean public RabbitTemplate rabbitTemplate(CachingConnectionFactory factory) { //若使用confirm-callback ...

  2. BST B树 B+树

    二叉排序树/二叉搜索树 (BST) 定义 左子树节点值<根节点值<右子树节点值 默认不允许两个节点的关键值相同 进行中序遍历可以得到递增的有序序列 查找效率 取决与树的高度,最好O(log ...

  3. 模拟BS服务器

    一.模拟BS服务器分析 二.BS模拟服务器代码实现 图片都是单独请求,后台单独线程,这边是通过构造方法传入的Runable接口的实现类匿名对象创建线程: 创建本地输入流读取到网络输出流传过来的信息再放 ...

  4. (5)java Spring Cloud+Spring boot+mybatis企业快速开发架构之SpringCloud-Spring Boot简介

    ​Spring Boot 是由 Pivotal 团队提供的全新框架,其设计目的是简化新 Spring 应用的初始搭建以及开发过程.该框架使用了特定的方式进行配置,从而使开发人员不再需要定义样板化的配置 ...

  5. Oracle列值拼接

    最近在学习的过程中,发现一个挺有意思的函数,它可实现对列值的拼接.下面我们来看看其具体用法. 用法: 对其作用,官方文档的解释如下: For a specified measure, LISTAGG  ...

  6. HTTP证书申请,设置应用程序服务器使用HTTPS

    HTTP证书申请,设置应用程序服务器使用HTTPS https://certs.godaddy.com/repository/ 根证书和中级证书下载地址(godaddy) ######Godaddy购 ...

  7. visual studio下载速度为0解决方法

    步骤: 一,更改网络设置 二,cmd刷新dns 一,更改网络设置 1,点开控制面板,打开网络和Internet 2,点击网络和共享中心 3,点击你连接的网络,那个是你连接的WIFI名字 4,点击属性 ...

  8. Python中with...as...的用法详解

    简介 with是从Python2.5引入的一个新的语法,它是一种上下文管理协议,目的在于从流程图中把 try,except 和finally 关键字和资源分配释放相关代码统统去掉,简化try-.exc ...

  9. php图片处理库

    <?php namespace app\common\library; /** * include 'imagick.class.php'; $image = new lib_image_ima ...

  10. 【转载】小心 int 乘法溢出!

    C/C++ 语言里, 绝大部分平台上 int 类型是 32 位的, 无论你的操作系统是否是 64 位的. 而一些常用的函数, 如 malloc(), 它接受的参数是 size_t 类型: void * ...