pytest + yaml 框架 -35.根据不同运行环境生成全局token

上海-悠悠 / 2023-06-08 / 原文

前言

我们在使用自动化测试框架的时候，经常会遇到一个需求，希望在全局用例中，仅登录一次，后续所有的用例自动带上请求头部token 或者cookies。
我们可以自定义fixture 更新内置的 requests_session, 在请求头部添加token 来实现全局登录。

全局 token 实现

在项目下 conftest.py 文件中实现先登录，更新请求头部 token

import pytest
from requests import Session
import jsonpath
"""
全局仅登录一次，获取token，
在请求头部添加 Authorization: bearer **token** 认证
  也有这种格式 Authorization: Token **token**
内置fixture requests_session
"""


@pytest.fixture(scope="session", autouse=True)
def login_first(requests_session: Session) -> None:
    """全局仅一次登录， 更新session请求头部"""
    # 调用登录方法，获得token
    url = "http://127.0.0.1:8200/api/v1/login"
    body = {
        "username": "test1",
        "password": "123456"
    }
    r = requests_session.post(url, json=body)
    token = jsonpath.jsonpath(r.json(), '$..token')[0]
    print(f'账号 test1 登录获取到token: {token}')
    headers = {
        "Authorization": f"Token {token}"
    }
    requests_session.headers.update(headers)

cases/test_info.yml 中用例内容

test_user_info:
  name: 用户信息
  request:
    url: /api/v1/userinfo
    method: GET
  validate:
    - eq: [$.code, 0]
    - eq: [$.msg, success!]

用例中只写了相对路径，还需要在pytest.ini 配置 base_url 环境地址

[pytest]

log_cli = true
log_cli_level = info
base_url = http://127.0.0.1:8200

执行pytest命令运行用例，会看到请求头部自动带上token了

2023-06-08 11:12:02 [INFO]: 运行用例-> test_user_info
2023-06-08 11:12:02 [INFO]: --------  request info ----------
2023-06-08 11:12:02 [INFO]: yml raw  -->: {'url': '/api/v1/userinfo', 'method': 'GET'}
2023-06-08 11:12:02 [INFO]: method   -->: GET
2023-06-08 11:12:02 [INFO]: url      -->: /api/v1/userinfo
2023-06-08 11:12:02 [INFO]: headers  -->: {'User-Agent': 'python-requests/2.31.0', 'Accept-Encoding': 'gzip, deflate', '
Accept': '*/*', 'Connection': 'keep-alive', 'Authorization': 'Token 8846dbad632813f097c577d0ff6aafc4547090f9'}

这样就可以实现全局仅登录一次，后续用例全部自动带上登录的请求头部'Authorization': 'Token ....'.
上面这种方式虽然实现了全局token ，但是登录的token地址是写死的，如果有多套环境需要切换，并且不同环境登录的账号不一样，就需要根据环境的切换来自动读取环境配置了。

多套环境配置

多套环境配置，参考前面这篇https://www.cnblogs.com/yoyoketang/p/16979211.html
在项目下 config.py 中配置不同环境

还需要在 pytest.ini 配置环境名称，这样 pytest.ini 中的 base_url 就不用配置了，可以全部写到 config.py 中的配置里用大写的 BASE_URL

[pytest]

log_cli = true
log_cli_level = info

env = test

在项目下 conftest.py 文件中实现先登录，修改后如下

import pytest
from requests import Session
import jsonpath
"""
全局仅登录一次，获取token，
在请求头部添加 Authorization: bearer **token** 认证
  也有这种格式 Authorization: Token **token**
内置fixture requests_session
"""


@pytest.fixture(scope="session", autouse=True)
def login_first(requests_session: Session, environ) -> None:
    """全局仅一次登录， 更新session请求头部"""
    # 调用登录方法，获得token
    url = environ.BASE_URL + "/api/v1/login"
    print(url)
    body = {
        "username": environ.USERNAME,
        "password": environ.PASSWORD
    }
    r = requests_session.post(url, json=body)
    token = jsonpath.jsonpath(r.json(), '$..token')[0]
    print(f'账号 test1 登录获取到token: {token}')
    headers = {
        "Authorization": f"Token {token}"
    }
    requests_session.headers.update(headers)

切换环境运行有2种方式
方式1：修改pytest.ini 中的 env值