pytest + yaml 框架 - 3.全局仅登录一次，在用例中自动在请求头部添加Authentication token认证

news/2024/5/8 11:09:22/文章来源:https://blog.csdn.net/qq_27371025/article/details/128125913

前言

我们在使用自动化测试框架的时候，经常会遇到一个需求，希望在全局用例中，仅登录一次，后续所有的用例自动带上请求头部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():"""登录方法"""# 调用登录方法，返回tokenreturn str(uuid.uuid4())  # noqa@pytest.fixture(scope="session", autouse=True)
def login_first(requests_session):"""全局仅一次登录， 更新session请求头部"""# 调用登录方法，获得tokentoken = login()headers = {"Authentication": f"Bearer {token}"}requests_session.headers.update(headers)

上面代码中，我用login() 函数实现登录功能，这里返回一个随机值，主要是为了验证我只调用了一次登录方法
接着我写2个yaml文件（注意，yaml文件中也不需要重复去添加请求头部了）
test_get_demo.yml

config:name: getteststeps:
-name: getrequest:method: GETurl: http://httpbin.org/getvalidate:- eq: [status_code, 200]

test_post_demo.yml

config:name: post示例variables:username: testpassword: "123456"teststeps:
-name: postrequest:method: POSTurl: http://httpbin.org/postjson:username: ${username}password: ${password}validate:- eq: [status_code, 200]- eq: [headers.Server, gunicorn/19.9.0]- eq: [$..username, test]- eq: [body.json.username, test]