做接口测试时，大部分接口都需要登录鉴权。笔者在项目中进行接口测试时需要在请求头部添加 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请求都会自动带上这个请求头信息。

最后

本文介绍了requests.Session对象的基本使用方法和优势，并通过fixture实现了自动添加请求头的功能。其实比较简单，只要了解了requests.Session以及fixture的使用还是比较容易实现的。

这样在进行接口测试时，可以通过设置统一的请求头信息，避免在每个测试用例中重复设置请求头，提高了代码的可读性和可维护性。同时，它还能够确保在整个测试会话期间，所有的请求都使用相同的请求头，保持了一致性。

最后作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些软件测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

​

测试文档模板

​

测试简历模板

​

视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。