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

2401_84562425

于 2024-05-10 08:56:01 发布

阅读量449

点赞数 5

分类专栏：程序员文章标签：单元测试功能测试学习

本文链接：https://blog.csdn.net/2401_84562425/article/details/138646648

版权

程序员专栏收录该内容

189 篇文章 0 订阅

订阅专栏

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

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

需要这份系统化的资料的朋友，可以戳这里获取

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):
        # 计数
        self.counter = 0

    @task(1)
    def sub_task1(self):
        # 执行子任务1
        # 一些任务逻辑
        pass

    @task(2)
    def sub_task2(self):
        # 执行子任务2，达到计数后中断退出子任务
        if self.counter > 100:
            self.interrupt()
        self.counter += 1
        # 一些任务逻辑
        pass

class MyTaskSet(InterruptTaskSet):

    @task(1)
    def task1(self):
        # 执行任务1
        # 一些任务逻辑
        pass

    @task(2)
    def task2(self):
        "这里被中断后，后续不会再次触发执行"
        try:
            # 执行任务2，中断控制
            self.interrupt()
        except InterruptTaskSet as e:
            # 中断任务后的处理
            pass

    @task(3)
    def task3(self):
        # 执行任务3，并嵌套执行子任务集
        self.sub_task_set()

    @task(4)
    def task4(self):
        # 执行任务4
        # 一些任务逻辑
        pass

    def sub_task_set(self):
        # 执行子任务集合，并在子任务集合中使用中断控制
        self.SubTaskSet(self.locust).run()


子任务在每次执行任务时检查计数器的值，当计数器达到100时调用`interrupt`方法中断任务执行，返回顶层。




![img](https://i-blog.csdnimg.cn/blog_migrate/cbfe7e6ba231042cb3b0832a867fb1ab.png)
![img](https://img-blog.csdnimg.cn/img_convert/1c3526d2e2f8505b57d1a7d9b11a579c.png)
![img](https://img-blog.csdnimg.cn/img_convert/0952bfc27e9ece77dcccaea7db5d915b.png)

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

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/forums/4f45ff00ff254613a03fab5e56a57acb)**

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

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/forums/4f45ff00ff254613a03fab5e56a57acb)**