测试中的终极神器：使用 pytest fixture 和 session 自动添加请求头，让接口测试更高效

原创于 2024-12-31 14:14:43 发布 · 1k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#pytest #程序员 #自动化测试 #软件测试 #功能测试 #学习 #职场经验

软件测试同时被 3 个专栏收录

2697 篇文章

订阅专栏

职场经验

1479 篇文章

订阅专栏

自动化测试

462 篇文章

订阅专栏

📝 面试求职： 「面试试题小程序」，内容涵盖测试基础、Linux操作系统、MySQL数据库、Web功能测试、接口测试、APPium移动端测试、Python知识、Selenium自动化测试相关、性能测试、性能测试、计算机网络知识、Jmeter、HR面试，命中率杠杠的。（大家刷起来…）

📝 职场经验干货：

软件测试工程师简历上如何编写个人信息（一周8个面试）

软件测试工程师简历上如何编写专业技能（一周8个面试）

软件测试工程师简历上如何编写项目经验（一周8个面试）

软件测试工程师简历上如何编写个人荣誉（一周8个面试）

软件测试行情分享（这些都不了解就别贸然冲了.）

软件测试面试重点，搞清楚这些轻松拿到年薪30W+

软件测试面试刷题小程序免费使用（永久使用）

做接口测试时，大部分接口都需要登录鉴权。笔者在项目中进行接口测试时需要在请求头部添加 Authentication token ，使用 requests.get()、 requests.post() 等函数发送请求时，通常需要传入参数 headers，像这样 requests.post(url='',json=payload,headers={'Authorization':''})，但每个请求都需要手动添加 headers ，还是比较麻烦的，如何自动添加 Authentication token 认证呢？这篇文章将揭晓。

知识回顾

在进入正文前，需要先知道什么是 requests.Session 。

requests.Session 是 requests 库提供的一个会话对象，用于管理和处理HTTP请求。相比于单独使用 requests.get() 、 requests.post() 等函数发送请求，使用 Session 对象可以更加灵活和高效地处理请求和响应，同时还可以共享一些相关数据和状态信息。

什么是Session对象

Session对象是在HTTP请求过程中通信的一部分，是与服务器交互的一种方式。

HTTP协议是无状态的，也就是说每个请求之间是没有联系的。而通过使用Session对象，我们可以在多个请求之间保持某些状态信息的一致性，比如cookies、身份验证信息等。

使用 Session 对象时，可以先创建一个实例，然后使用该实例发送多个请求，这些请求会自动共享相同的cookies、身份验证信息等状态。例如：

import requests

s = requests.Session()
s.get('http://httpbin.org/cookies/set/sessioncookie/123456789')
r = s.get('http://httpbin.org/cookies')

print(r.text)
# 输出结果：
# {
#   "cookies": {
#     "sessioncookie": "123456789"
#   }
# }

在上述代码中，首先使用 requests.Session() 创建了一个Session对象实例 s ，然后使用 s.get() 发送了两个请求，其中第一个请求设置了名为 sessioncookie 的cookie值为 123456789 ，第二个请求获取了当前会话中的所有cookies并打印出来。由于这两个请求共用了同一个 Session 对象，所以第二个请求可以正确获取到第一个请求设置的cookie值

Session对象的优点

相比于单独使用 requests.get() 、 requests.post() 等函数发送请求，使用 Session 对象有以下优点：

1、自动管理cookies：Session对象会自动保存和发送cookies，使得多次请求时不需要手动添加或传递cookies。

2、保持状态信息的一致性：Session对象可以共享所有相关的状态信息，如cookies、headers、身份验证信息等，避免了多个请求之间信息不一致的问题。

3、多次请求共享底层TCP连接：使用 Session 对象可以通过复用底层的TCP连接，减少HTTP连接的建立和关闭的开销，提高请求效率。

4、方便错误处理：Session对象提供了一个类似于try-except的机制，对网络错误、服务器返回的错误响应等进行统一处理，进一步简化了代码。

使用示例

下面是一个简单的使用 Session 对象发送HTTPGET请求的示例：

import requests

with requests.Session() as s:
    response = s.get('https://www.baidu.com/')
    print(response.content)

在上述代码中，首先使用 requests.Session() 创建了一个 Session 对象实例s，然后使用 s.get() 发送GET请求获取百度首页的内容，并将相应结果打印出来。

另外，Session 对象还支持一些其他的方法和属性：

s.headers：用于设置请求头
s.auth：用于设置身份验证信息
s.cert：用于设置SSL证书路径
s.proxies：用于设置代理地址
s.stream：用于设置响应是否分块下载
s.cookies：用于设置和获取cookies
s.purge(url,**kwargs)：用于清除指定URL的缓存
s.resolve(url,**kwargs)：用于解析指定URL

下面是一个使用 Session 对象发送POST请求的示例：

import requests

payload = {'username': 'user', 'password': 'password'}
with requests.Session() as s:
    s.post('http://httpbin.org/post', data=payload, headers={'Content-Type': 'application/json'})
    r = s.get('http://httpbin.org/headers')
    print(r.json())

在上述代码中，首先创建了一个字典对象 payload 来设置POST请求时的传输数据，然后使用 s.post() 方法发送POST请求，同时传入参数 data 和 headers ，分别表示POST请求的传输数据和请求头信息。接下来，使用 s.get() 方法发送GET请求获取当前会话中的所有headers并打印出来。

实现自动添加 Authentication

当然，这里我们还是会借助 fixture 来实现。

实现过程

定义会话级别的 Session fixture

from requests import Session

@pytest.fixture(scope="session")
def requests_session():
    s = Session()
    s.mount('http://', HTTPAdapter(max_retries=2))
    s.mount('https://', HTTPAdapter(max_retries=2))
    yield s
    s.close()

这段代码，通过 s=Session() 创建一个 requests.Session 对象，并为此会话对象设置了最大重试次数为2次。

接下来，将该会话对象作为fixture的返回值，使用 yield 关键字进行暂停，以便在测试环境中使用这个会话对象进行请求。当测试完成后，使用 s.close() 关闭会话，释放相关资源。

自动添加 headers

@pytest.fixture(scope="session", autouse=True)
def headers(requests_session):
    authorization = "Bearer ******"
    headers = {'Authorization': authorization}
    requests_session.headers.update(headers)

这段代码，在每个测试会话开始之前，它会自动调用headers函数，使用了requests_session.headers.update(headers)将设置好的请求头信息更新到requests_session对象中。通过这样的操作，后续的所有使用requests_session对象发送的HTTP请求都会自动带上这个请求头信息。