Locust框架从0到1入门

Tester_孙大壮

已于 2023-04-09 17:22:01 修改

阅读量483

点赞数

分类专栏：性能测试 Python后端文章标签： python 测试工具开发语言

于 2023-03-09 21:46:46 首次发布

本文链接：https://blog.csdn.net/m0_48468018/article/details/129397103

版权

Python后端同时被 2 个专栏收录

15 篇文章 1 订阅

订阅专栏

性能测试

6 篇文章 0 订阅

订阅专栏

Locust是一个用Python编写的开源性能测试工具，擅长测试Web应用、API和数据库。它基于gevent实现高并发，并提供简单易用的接口、分布式测试能力、实时监控和可扩展性。用户可以自定义Python脚本定义用户行为，并通过Web界面管理测试。

摘要由CSDN通过智能技术生成

Locust介绍

Locust是使用Python语言编写实现的开源性能测试工具，可以用来测试Web应用程序、API、数据库等各种应用程序的性能，使用起来简洁、轻量、高效，并发机制基于gevent协程，可以实现单机模拟生成较高的并发压力。中文意为：蝗虫，蝗虫过境，寸草不生。
它的主要思想：模拟一群用户将访问你的网站。每个用户的行为由你编写的python代码定义，同时可以从
Web界面中实时观察到用户的行为。
Lucust的主要特点包括：

简单易用：Lucust采用Python编写，开发者可以通过Python代码进行测试，并且可以使用Web界面和命令行界面轻松地管理测试和监控结果。
分布式架构：Lucust支持分布式测试，可以同时运行多个进程和多个机器的测试。分布式测试还可以提高测试效率和准确性。
实时监控：Lucust提供实时监控和实时统计功能，可以实时查看测试结果和对测试数据进行分析。
可扩展性：Lucust可以通过编写自定义测试脚本、自定义模拟用户行为，以及支持多种格式的测试数据等方式进行扩展。
总的来说，Lucust是一个功能强大且易于使用的性能测试框架，适用于各种规模的应用程序测试，并且可以通过扩展和定制满足个性化的测试需求。

几款主流性能测试框架对比

性能测试通常需要有压力发生器，系统资源监控器，结果采集器，结果分析器等组件。
在这里插入图片描述
与jmeter类似，locust没有系统资源监控器

下面是locust与jemter,loadrunner的功能对比

locust框架搭建

locust的官网：https://www.locust.io/，项目地址：https://github.com/locustio/locust

在本地有python开发环境的前提下，使用 pip install locust 命令安装。安装时注意本地的C++环境，如提示需要C++2014以上，如遇到错误，根据提示安装C++环境即可。
编写性能测试脚本,并启动服务

2.1 导入所需的库和函数，如：

from locust import HttpUser, task, between

其中，HttpUser为基础用户类，task用于定义任务，between用于指定用户执行任务的时间间隔。

2.2 定义用户类：

class MyUser(HttpUser):
    wait_time = between(1, 3)
    
    @task
    def my_task(self):
        self.client.get("/path")

其中，wait_time指定用户执行任务之间的等待间隔，@task为任务定义装饰器，my_task为任务函数名称，self.client.get(“/path”)为具体的任务执行代码。

2.3. 运行Locust：

locust -f my_script.py

其中，-f参数指定要执行的脚本文件。

2.4. 在浏览器中访问Locust Web界面，如：

http://localhost:8089/

在界面中可以指定用户数、用户增长速率、任务权重等参数，并实时查看测试结果。
在这里插入图片描述

locust实例脚本

如下为一个简单实例，其中on_start为对locust内置方法的重写，另外还有on_stop方法。

from locust import HttpUser, TaskSet, task

class WebsiteTasks(TaskSet):
    def on_start(self):  # 进行初始化的工作，每个Locust用户开始做的第一件事
        payload = {
            "username": "XXXX",
            "password": "123456",
        }
        header = {
            "User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36",
        }
        self.client.post("/login", data=payload,headers=header)  # self.client属性使用Python request库的所有方法，调用和使用方法和requests完全一致；

    @task(2)  # 通过@task()装饰的方法为一个事务，方法的参数用于指定该行为的执行权重，参数越大每次被虚拟用户执行的概率越高，默认为1
    def index(self):
        self.client.get("/")

    @task(1)
    def about(self):
        self.client.get("/about/")

class WebsiteUser(HttpUser):
    host = "https://github.com/"  # 被测系统的host，在终端中启动locust时没有指定--host参数时才会用到
    tasks = [WebsiteTasks]  # TaskSet类，该类定义用户任务信息，必填。这里就是WebsiteTasks类名,因为该类继承TaskSet；
    min_wait = 3000  # 每个用户执行两个任务间隔时间的上下限（毫秒）,具体数值在上下限中随机取值，若不指定默认间隔时间固定为1秒
    max_wait = 10000

locust常用命令

1、脚本调试命令行：
locust -f locustfile.py --headless -u 1 -r 1 –t 30s
2、在Terminal以如下命令启动locust，通常情况下不需额外指定其他参数：
locust -f locustfile.py
3、Locust默认采用8089端口启动web，如果要使用其它端口，就可以使用如下参数进行指定：
locust -f locustfile.py –web-port 8089
4、–host用来指定测试主机地址
locust -f locustfile.py --host=http://example.com