Python Locust性能测试框架实践

[本文出自天外归云的博客园]

Locust的介绍

Locust是一个python的性能测试工具，你可以通过写python脚本的方式来对web接口进行负载测试。

Locust的安装

首先你要安装python2.6以上版本，而且有pip工具。之后打开命令行，分别安装locustio和pyzmq（命令如下）：

pip install locustio
pip install pyzmq

之后我们就可以写性能测试脚本了。

PS：如果是python3，不能使用pip安装目前。需要在github上下载locust项目包到本地，然后在包路径下执行命令安装：

python setup.py install

Locust脚本编写

接下来我们拿两个接口做一下测试，编写脚本如下（每一步都有注释）。我来解释一下，首先我们要import进来三个类，分别是HttpLocust（用来模拟发请求的类）、TaskSet（顾名思义，任务集）、task（任务类）。额外的，为了方便观察接口测试的执行结果，我引入了json类用来解析web接口的返回值。我还引入了subprocess类用来执行一下shell命令，自启动Locust。这里有三个类，一个是UserBehavior（名字随便起，但传入TaskSet参数，说明这是一个包含了任务集的类），里面on_start函数可有可无，他会先于所有task函数运行。剩下被@task装饰器装饰的方法都是任务方法，里面包含了待请求的接口等信息，传入的参数代表了权重，如下所示两个被@task装饰的方法分别传入1和2，这意味着每3个人里有两个人会有1个模拟用户执行list_header方法，2个模拟用户执行list_goods方法。这个参数你也可以不传入，那就意味着模拟用户会随机访问所有被@task装饰的方法。这里面我对于每个接口的返回值都做了一下判断，首先将返回的字符串转成json格式并获取返回字段result的值，如果不是100就用Locust自带的报错方法打印出错信息；另两个类是HttpLocust类（仍然是名字随便起但传入参数必须得是HttpLocust），是用来模拟用户的类，包含了一些模拟用户信息，其中task_set变量的值用来指定模拟用户所对应要完成的TaskSet类中包含的请求，min_wait和max_wait（最小等待时间和最大等待时间用来模拟用户每两步操作之间的间隔时间，这里也就是模拟用户每执行两个请求之间所间隔的时间）。对Locust类我们可以指定权重，对weight变量的值进行指定。如下所示，两个Locust类的权重分别为1和3，这意味着两个Locust类的模拟用户人数为1:3的关系。最后我加了一个main函数用来执行shell命令，这个shell命令也可以不再本文件中执行，如果写在脚本中的话，直接在命令行中调用该python文件即可，如果不写在脚本中（注释掉最后两行），则需要在命令行终端里对Locust项目进行启动。

from locust import HttpLocust,TaskSet,task
import subprocess
import json

#This is the TaskSet class.
class UserBehavior(TaskSet):
    #Execute before any task.
    def on_start(self):
        pass

    #the @task takes an optional weight argument.
    @task(1)
    def