pytest + yaml 框架 - 3.全局仅登录一次,添加Authentication token认证
2023-02-19 12:20:25 时间
前言
我们在使用自动化测试框架的时候,经常会遇到一个需求,希望在全局用例中,仅登录一次,后续所有的用例自动带上请求头部token 或者cookies。
环境准备
Python 3.8版本 Pytest 7.2.0 最新版
pip 安装插件
pip install pytest-yaml-yoyo
需使用v1.0.2 版本
登录fixture 功能
我在pytest + yaml 框架框架中封装了一个内置fixture叫requests_session
, 它的作用范围是scope="session"
,也就是全部session用例会话中仅实例化一次。
现在我只需在conftest 中写一个登录的fixture功能,获取token后添加到requests_session
头部
import pytest
import uuid
"""
全局仅登录一次,获取token,
在请求头部添加Authentication Bearer 认证
内置fixture requests_session
"""
def login():
"""登录方法"""
# 调用登录方法,返回token
return str(uuid.uuid4()) # noqa
@pytest.fixture(scope="session", autouse=True)
def login_first(requests_session):
"""全局仅一次登录, 更新session请求头部"""
# 调用登录方法,获得token
token = login()
headers = {
"Authentication": f"Bearer {token}"
}
requests_session.headers.update(headers)
上面代码中,我用login() 函数实现登录功能,这里返回一个随机值,主要是为了验证我只调用了一次登录方法 接着我写2个yaml文件(**注意,yaml文件中也不需要重复去添加请求头部了**)
test_get_demo.yml
config:
name: get
teststeps:
-
name: get
request:
method: GET
url: http://httpbin.org/get
validate:
- eq: [status_code, 200]
test_post_demo.yml
config:
name: post示例
variables:
username: test
password: "123456"
teststeps:
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: ${username}
password: ${password}
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$..username, test]
- eq: [body.json.username, test]
运行用例
在命令行中输入pytest
运行
抓包看发过去的请求
于是可以看到,在2个用例中都自动带上了请求头部参数。 (登录cookies的使用原理也是一样的,登录后cookies一般会自动保持)
其它需求
那有些接口不需要登录怎么办呢?比如登录和注册的接口,是不需要带上登录token的,那不能一刀切。
我除了默认用到一个requests_session
全局的内置fixture,还预留了2个
- requests_module: 每个yaml文件中用一次
- requests_function: 每个用例中用一次
在yaml文件中切换到指定fixture功能,requests_module
和 requests_function
后续会实现,先实现大需求,解决大部分人遇到的痛点问题!
相关文章
- 笔记本怎样连接投影仪-linux(ubuntu)下外接投影仪或着显示器的方法
- 【毕业设计】单片机红外热成像仪 - 嵌入式 stm32 树莓派
- vc60修改快捷键-如何解决Visual c++不能打开文件以及工程不能添加文件的问题
- c 线程安全的单例模式-设计模式之单例模式(C++版)
- c 线程安全的单例模式-C++单例模式(线程安全、内存释放)
- 嵌入式:ARM相关开发工具概述
- 嵌入式:ARM符号定义伪操作详解
- c 线程安全的单例模式-详解C++实现线程安全的单例模式
- C++ OpenCV手动截取图像做透视变换
- 超简单的pyTorch训练->onnx模型->C++ OpenCV DNN推理(附源码地址)
- pyTorch入门(三)——GoogleNet和ResNet训练
- pyTorch入门(四)——导出Minist模型,C++ OpenCV DNN进行识别
- Github Actions Hexo 自动部署
- c 线程安全的单例模式-std string与线程安全_这才是现代C++单例模式简单又安全的实现
- 我是这么用 Google 的....
- 《HelloGitHub》第 13 期
- 《HelloGitHub》第 12 期
- [linux]为阿里云ECS(CentOS7)配置IPv6地址
- 《HelloGitHub月刊》第 11 期
- [linux]孤儿进程与僵尸进程