使用Locust进行接口性能测试：Locust and TaskSet类详细分析_locust 报告接口

from locust import Locust, TaskSet, task
class MyUser(Locust):

…

class MyTaskSet(TaskSet):
@task(1) # 设置任务执行频率，数值越大越高
def task1(self):

执行任务1

pass
@task(2) # 设置任务执行频率，数值越大越高
def task2(self):

执行任务2

pass

在这个例子中，MyTaskSet类中定义了两个任务函数task1和task2，它们都被标记为可执行任务。Locust将随机选择并执行这两个任务。

2.2 tasks属性

tasks属性也可以设置频率

from locust import Locust, TaskSet, task

class MyUser(Locust):

…

class MyTaskSet(TaskSet):
tasks = {task1: 2, task2: 1} # 设置执行频率

def task1(self):

执行任务1

pass

def task2(self):

执行任务2

pass

task1的执行概率是task2的两倍。

2.3 on_start函数

on_start用于任务开始执行任务前的预处理操作。例如，登录或初始化数据。它主要是使用了requests.Session，所以执行一次就可以让后续所有任务执行过程中都有登录态。

from locust import Locust, TaskSet, task

class MyUser(Locust):

…

class MyTaskSet(TaskSet):
@task(1)
def my_task(self):
self.client.get(“/list”)

其他逻辑

pass

def on_start(self):

在用户开始执行任务前执行的操作，例如登录

self.client.post(“/login”)
pass

on_start函数使用非常频繁。

2.4 控制任务的执行顺序

@seq_task()可以控制任务的执行顺序。
按顺序执行任务：

from locust import Locust, TaskSequence, seq_task, task, between

class MyUser(Locust):
wait_time = between(1, 3) # 设置任务执行的等待时间范围

…

class MyTaskSequence(TaskSequence):
@seq_task(1) # 顺序1
def login(self):

执行登录操作

pass

@seq_task(2) # 顺序2
def view_items(self):

执行查看商品列表的操作

pass

@seq_task(3) # 顺序3
@task(3) # 为该任务设置执行概率
def add_to_cart(self):

随机执行添加商品到购物车的操作，概率为3/4

pass

@seq_task(4) # 顺序4
def checkout(self):

执行结账操作

pass

2.5 休眠等待（wait）

在任务之间插入休眠时间wait_time，可以模拟用户在执行任务时的等待行为，且Locust内置一些函数，返回wait_time方法

• between(min, max) 生成指定范围内的随机等待时间, min_wait和max_wait分别表示等待时间的最小值和最大值（以毫秒为单位）
• constant(constant_wait) 生成固定的等待时间,constant_wait表示等待的时间（以毫秒为单位）
• constant_pacing(pacing_interval) 指定任务执行的固定间隔，比如：constant_pacing(2) 表示用户执行任务2秒，然后再继续下一个任务。

from locust import Locust, between, constant_pacing

class MyUser(Locust):

使用between函数，每个用户等待1到3秒之间的随机时间

wait_time = between(1000, 3000)

-----------------分割线------------------------------------

class MyUser(Locust):

使用constant_pacing属性，每个用户任务之间固定间隔2秒

wait_time = constant_pacing(2)

2.6 TaskSets 嵌套

可以在一个TaskSet中嵌套其他TaskSet，实现更复杂的用户行为模拟。例如：

from locust import Locust, TaskSet, task

class MyUser(Locust):

…

class SubTaskSet(TaskSet):
@task(1)
def sub_task1(self):

执行子任务1

pass

@task(2)
def sub_task2(self):

执行子任务2

pass

class MyTaskSet(TaskSet):
@task
def task1(self):

执行任务1

pass

@task
def task2(self):

执行任务2，并嵌套执行子任务集

self.sub_task_set()

tasks = {task1: 1, task2: 1}

def sub_task_set(self):
self.client.get(“/otherurl”)
self.SubTaskSet(self.locust).run()

在MyTaskSet中的task2任务中，我们嵌套执行了SubTaskSet。这样就可以实现一些非常复杂的业务流程的压测了，比如：用户打开网站首页，可能会直接离开，也可能会去流量商品，也可能加入商品到购物车下单等等场景一起模拟。

2.7 中断控制(interrupt)

做压测的时候，有些场景下是要提前结束某些任务的，那么可以设置一个计数器，在计数到一定值后中断任务的执行，比上述的嵌套场景中也就刚刚好符合这点，嵌套子任务执行达到某一个阈值的时候，就中断回到父任务，这样就阻止了子任务一直执行的情况发生。

from locust import Locust, TaskSet, InterruptTaskSet, task, between

class MyUser(Locust):
wait_time = between(1, 3) # 设置任务执行的等待时间范围

class SubTaskSet(TaskSet):
def no_start(self):

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数软件测试工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年软件测试全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上软件测试开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以添加V获取：vip1024b （备注软件测试）

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新**

如果你觉得这些内容对你有帮助，可以添加V获取：vip1024b （备注软件测试）
[外链图片转存中…(img-m7tpLlYS-1713013268981)]

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！