很多网站需要先登录才能访问其他的接口，下面以先登录然后访问->群管理页面->问题件管理页->索赔件管理为例
# 保存为locustfile.py
# coding=utf-8
from locust import HttpLocust, TaskSet, task
# 实现场景：先登录（只登录一次），然后访问->群管理页面->问题件管理页->索赔件管理
class UserBehavior(TaskSet):
def _login(self):
'''登录方法'''
# host = 'https://gd.baoshi56.com' # 服务器地址
loginUrl = "/admin/common/login.html"
h = {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.80 Safari/537.36",
"Content-Type": "application/x-www-form-urlencoded",
}
body = {"account": "********", # 账号
"password": "******", # 密码
}
# self.client调用get和post方法，跟requests是一样的
# requests.get 对应client.get
# requests.post 对应client.post
r = self.client.post(loginUrl, data=body, headers=h)
if r.status_code == 200:
print("登录成功")
else:
print(r.status_code + "登录失败")
# 定义on_start()相当于用例的准备操作
def on_start(self):
'''任务开始准备工作：只登录一次'''
self._login() #用户登录
# 任务1-访问群管理页面
# @task装饰该方法表示为用户行为。
# 括号里面参数表示该行为挑选执行的权重，数值越大，执行频率越高，不设置默认是1
@task(2)
def admin_group(self):
r = self.client.get("/admin/group/grouplist.html/")
if r.status_code == 200:
print("访问群管理页面功")
else:
print(r.status_code + "访问群管理页面失败")
# 任务2-问题件管理页
@task(1)
def admin_question(self):
r = self.client.get("/admin/question_record/questionrecordlist.html/")
if r.status_code == 200:
print("访问问题件管理页面功")
else:
print(r.status_code + "访问问题件管理页面失败")
# 任务3-项目
@task(1)
def admin_claim(self):
r = self.client.get("/admin/claim_record/claimrecordlist.html/")
if r.status_code == 200:
print("访问索赔件管理页面功")
else:
print(r.status_code + "访问索赔件管理页面失败")
# 性能测试配置
"""
WebsiteUser()类用于设置性能测试。
task_set ：指向一个定义了的用户行为类。
min_wait ：用户执行任务之间等待时间的下界，单位：毫秒。
max_wait ：用户执行任务之间等待时间的上界，单位：毫秒。
"""
class WebsiteUser(HttpLocust):
task_set = UserBehavior
min_wait = 1000
max_wait = 1000
if __name__ == "__main__":
import os
os.system("locust -f locustfile.py --host=https://gd.baoshi56.com")
# -f 参数是指定运行的脚本
# --host是指定运行项目的host地址，
# 这里用的https://gd.baoshi56.com，
# 和代码里面访问的地址拼接起来就是完整地址了
设置1个虚拟用户，每秒启动1个服务，点start后运行结果：
从结果可以看到登录的请求只访问了一次，然后是任务1的次数差不多是任务2、任务3请求次数的2倍
image.png
image.png
