前言

工作中我们需要压测的接口大部分都是需要先登陆后，带着token的接口（或者带着cookies），我们可以先登陆获取token再关联到下个接口。
比如我现在要压测一个修改用户的个人信息接口，每个用户只能修改自己的个人信息。

场景案例

我现在有一个登陆接口A,登陆成功后返回一个token值。A接口登录接口文档基本信息

访问地址：

http://127.0.0.1:9000/api/v1/login/
请求类型：

POST
请求头部：

application/json
请求参数：

{“username”:”test”, “password”:”123456”}

接口返回

{

“code”: 0,

“msg”: “login success!”,

“token”: “234af73571da46ade79ea6a74961b1d23d609b79”,

“username”: “test”

}

B接口需先登陆，修改用户的个人信息，并且用户只能修改自己的个人信息

访问地址：

http://127.0.0.1:8000/api/v1/userinfo
请求类型：

POST
请求头部：

Content-Type: application/json

Authorization: Token 234af73571da46ade79ea6a74961b1d23d609b79
请求参数：

{“name”:”test”,”sex”:”M”,”age”:23,”mail”:”123@qq.com”}

接口返回

{

“message”: “update some data!”,

“code”: 0,

“data”: {

"name": "test", "sex": "M", "age": 23, "mail": "123@qq.com"

}

}

如果先登陆再关联到修改个人信息，这样实际上是发了2个请求，真实场景是用户早就登陆了，所以并不需要去压测登陆的场景。
我们只需要拿到token直接去压测B接口就行了。

测试token准备

B接口有两个参数是一一对应的，一个是token，一个是对应的name，比如压测的时候准备100个用户，我这里以10个用户为例
先注册批量的用户用于压测，我这里注册的用户是test1,test2……test10,一起10个用户

''' 批量生成token和需要用的测试参数，如username 大前提，你需要先注册这些用户test1,test2......test10 ''' import requests # 上海-悠悠，QQ交流群：750815713


url = "http://127.0.0.1:9000/api/v1/login"

for i in range(1, 11): # 拼接用户名用户名如 test1,test2.....test10 username = "test"+str(i) body = {"username": username, "password": "123456"} r = requests.post(url, json=body) print(r.json()) # 获取token token = r.json()["token"] # a追加模式写入txt with open("token_user.txt", "a") as fp: fp.write(token+","+username+"\n")

运行后生成的token_user.txt文件内容如下

f26be862c407a6fabf8fe3290ac21392b6eaf44f,test1 bed455954cb70f2060e17cf85339758cdcf94629,test2 033a47cb03e6d99384121d8a015fbe2542a198b5,test3 a09b5af04a06b274fece0a6766ed056be773294d,test4 49f3fee051c0615128577cfcfe324f226366b824,test5 c2ebaabe55bb651682b9c1c3b3d3779543f7088b,test6 f74dcea201337665f7f6a25905b9845826eb34c8,test7 f3d7bc13d0608196d557f0197a7f2c2c407a7d0d,test8 9c6d7b893ea2e8b226daa0c315299d5968e8a10c,test9 4cfd0827e803a7415e987996c2148312843ed037,test10