jmeter压测学习8-压测带token的接口

前言

工作中我们需要压测的接口大部分都是需要先登陆后，带着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+"
")

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

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

jmeter CSV数据文件设置

导入csv 测试数据文件，设置token和user两个变量

HTTP 信息头管理器引用token

请求参数引用user

运行结果

接下来就可以设置线程组愉快的压测了

比如我设置2个线程，4次循环，这样会请求8次，每次都从测试文件里面循环取值

上海-悠悠，QQ交流群：750815713