requests手动添加cookies_玩转测试开发_request 添加cookie

大大的周 02-07 5125

简介：在接口测试中，大多数项目的接口是需要登录后进行操作的，经常用到requests库进行模拟登录及登录后的操作。

1、sanic：读、写、删除cookies 2、http库三剑客：requests

案例解析： 1、本案例中，假设登录接口为：http://127.0.0.1:3031/cookie

2、登录完成后得到cookies

<RequestsCookieJar[<Cookie key=024f4347-cac8-4a01-9082-3e47378c697c for 127.0.0.1/>, <Cookie test="generate cookies" for 127.0.0.1/>]>

3、登录完成后通过手动新增cookies，并在再次请求时，将my_cookie 传给cookies

my_cookie = dict() for k, v in res.cookies.items(): my_cookie[k] = v res2 = requests.get(url="http://127.0.0.1:3031/check_cookie", cookies=my_cookie)

4、服务端在检查cookies的正确性。

test_cookie = request.cookies.get('test') key_cookie = request.cookies.get('key') if test_cookie == 'generate cookies' and key_cookie == key: return response.redirect("/home") else: return response.redirect("/login")

获取cookies：

import requests response = requests.get("https://·") print(response.cookies) for key, value in response.cookies.items(): print(key + '=' + value) # 执行结果 # <RequestsCookieJar[<Cookie _xsrf=4syf8BxI1p5x3pmf8PVbSvQOui001Lng for .zhihu.com/>]> # _xsrf=4syf8BxI1p5x3pmf8PVbSvQOui001Lng

服务端源码：

import uuid from sanic import Sanic from sanic import response from sanic.response import json, text app = Sanic(__name__) key = str(uuid.uuid4()) @app.get("/check_cookie") async def check(request): test_cookie = request.cookies.get('test') key_cookie = request.cookies.get('key') if test_cookie == 'generate cookies' and key_cookie == key: return response.redirect("/home") else: return response.redirect("/login") @app.route("/cookie") async def test(request): response = text("There's a cookie up in this response") response.cookies['test'] = 'generate cookies' response.cookies['key'] = str(key) print("cookies:", response.cookies) return response if __name__ == "__main__": app.run(host="127.0.0.1", port=3031, auto_reload=True)

客户端源码：

import requests def login1(): url = 'http://127.0.0.1:3031/cookie' res = requests.get(url) print(res.status_code) print(res.cookies) res2 = requests.get(url="http://127.0.0.1:3031/check_cookie") print(res2.url) def login2(): url = 'http://127.0.0.1:3031/cookie' res = requests.get(url) print(res.status_code) print(res.cookies) my_cookie = dict() for k, v in res.cookies.items(): my_cookie[k] = v print("my_cookie:", my_cookie) res2 = requests.get(url="http://127.0.0.1:3031/check_cookie", cookies=my_cookie) print(res2.url) if __name__ == '__main__': login1() print("*" * 100) login2()

运行服务端：运行客户端：服务端生成cookies和检验cookies的正确性：

最终： 1、未登录成功的重定向到：http://127.0.0.1:3031/login 2、登录成功的重定向到：http://127.0.0.1:3031/home

微信公众号：玩转测试开发欢迎关注，共同进步，谢谢！