使用Python的requests库进行接口测试——session对象的妙用

from:http://blog.csdn.net/liuchunming033/article/details/48131051

在进行接口测试的时候，我们会调用多个接口发出多个请求，在这些请求中有时候需要保持一些共用的数据，例如cookies信息。

妙用1

requests库的session对象能够帮我们跨请求保持某些参数，也会在同一个session实例发出的所有请求之间保持cookies。

举个栗子，跨请求保持cookies，在命令行上输入下面命令：

1. # 创建一个session对象
2. s = requests.Session()
3. # 用session对象发出get请求，设置cookies
4. s.get('http://httpbin.org/cookies/set/sessioncookie/123456789')
5. # 用session对象发出另外一个get请求，获取cookies
6. r = s.get("http://httpbin.org/cookies")
7. # 显示结果
8. r.text
9. '{"cookies": {"sessioncookie": "123456789"}}'

妙用2

requests库的session对象还能为我们提供请求方法的缺省数据，通过设置session对象的属性来实现。

举个栗子，设置请求的auth和headers默认的缺省参数：

1. # 创建一个session对象
2. s = requests.Session()
3. # 设置session对象的auth属性，用来作为请求的默认参数
4. s.auth = ('user', 'pass')
5. # 设置session的headers属性，通过update方法，将其余请求方法中的headers属性合并起来作为最终的请求方法的headers
6. s.headers.update({'x-test': 'true'})
7. # 发送请求，这里没有设置auth会默认使用session对象的auth属性，这里的headers属性会与session对象的headers属性合并
8. r = s.get('http://httpbin.org/headers', headers={'x-test2': 'true'})
9. # 查看发送请求的请求头
10. r.request.headers

得到的请求头部是这样的：

{'Authorization': 'Basic dXNlcjpwYXNz', 'x-test': 'false'}

注意1：方法层的参数覆盖会话的参数，举个栗子：

将上面的请求中加上auth参数：

1. r = s.get('http://httpbin.org/headers', auth=('user','hah'), headers={'x-test2': 'true'})

获取该请求的请求头

{'Authorization': 'Basic dXNlcjpoYWg=', 'x-test': 'false'}

我们发现，请求方法中的Authorization变化了，也就是说方法层的参数覆盖了session的属性。

注意2：在request请求中，省略session对象中设置的属性，只需简单地在方法层参数中将那个键的值设置为 None ，那个键就会被自动省略掉。

再举个栗子：

TBD