Python Easy-Locust 自动生成 locust 脚本

我叫GTD · January 22, 2020 · Last by 我叫GTD replied at February 06, 2020 · 2172 hits

前言

在做接口性能测试的时候也使用过几个工具:AB、Jmeter、Locust、Vegeta。

这几个工具中,应该当属Jmeter的功能最强大,使用的人也最多,但使用上也比较繁琐,如果测的接口比较单一,需要得到的值也就是并发量、每秒请求数、响应时间这三个的话,那就推荐使用个简单点儿的工具。在这儿主要介绍Locust,原因无非是使用简单,另外在web界面能实时显示性能测试情况,可以随时调整参数,也支持分布式。

使用Locust也需要编写脚本,对于不熟悉Python的人来说,又是一个障碍。为此笔者编写了一个工具Easy-Locust,Locust的几种使用方式都有,并且参数和Locust也保持一致,需要操作的就是在Excel中编辑好参数就行。

GitHub: Easy-Locust

使用介绍

安装

执行下面命令会自动安装easy-locust及locustio>=0.13.5的版本

pip install easy-locust

参数介绍

此处只介绍和Locust不同的地方,如果想了解详细参数,请查阅:Locustio

-f          如果为.py文件则直接执行,如果为.xls文件则根据里面的配置生成locustfile并执行
--demo 在当前文件夹下生成`PtDemo.xls`文件和`demo.json`,可以根据此文件进行配置
--xf 根据.xls文件生成locustfile但不执行(或根据json文件生成locustfile
-d master模式下此参数生效,自动向slave机器发送locustfile文件并在slave机器中开启服务,前提是slave机器中装有locustio>=0.13.5

API调用介绍

传入对应的dict格式,返回string格式的Locustfile。

import easy_locust

data = {
"config": {},
"apis": [],
"auth": {},
"user_info": [],
"master_ip": "",
"slaves": []
}

locustfile_str = easy_locust.factory(data)

XLS模板配置介绍

一共4个sheet页,逐一进行介绍

  • PT:locust主要参数
    PT
    下面为参数说明:

    Host            ——需要进行测试的接口URL前缀
    Min Wait ——最小等待时间(s),默认0.3s
    Min Wait ——最大等待时间(s),默认0.5s
    Request Mode ——请求模式,Locust中模式有两种,一种为HttpRequest基于python-requests,另一种为FastHttpRequest基于geventhttpclient,性能是前者的5-6
    Get Token ——是否获取token进行认证,如果选择YES的话需配置AuthInfo
    Run in order ——是否按顺序执行下面的接口
    # 下面为对接口的配置
    Weight ——接口权重,如果Run in orderTRUE,则为每轮次接口执行次数
    Url ——接口url
    Method ——接口请求方法
    Query ——query参数(json格式)
    Request data ——请求体(json格式)
    Expect Status Code ——期望响应码(为空默认则判断4XX/5XX为失败)
    Expect Str in Response ——在返回体中的期望字符串
  • AuthInfo:认证参数
    AuthInfo

    TokenUrl      ——获取token的接口(默认POST方法)
    Body ——请求体不局限于模板中的格式,如果填写参数中填写UserNamePassWord的话,UserInfo中的数据会生效
    X-Auth-Token ——图例中为从返回的Header中获取x-subject-token的值,并放到性能测试接口的x-auth-token参数中,表格中x-auth-token值可以更改

    除了可以从Header中获取token外还可以从json中获取,格式为Json: body.token,获取到的值会在调用接口时填充到header中,这个地方在Excel demo文件中无法配置,如果要改的话,可以生成locustfile文件,然后小改一下locustfile文件即可。

  • Slave:分布式压测,压测机信息
    Slave

    Master IP      ——当前机器的IP地址
    Slave IP ——压测机IP
    Username ——后台登陆用户名
    Password ——后台登陆密码

    注意:需要指定-d参数才可生效;各压测机中需要提前安装好相同版本的locustio>=0.13.5

  • UserInfo:用户登陆信息,配合AuthInfo使用
    UserInfo

    username    ——用户名
    password ——密码

    如果希望使用多个用户信息随机进行压测则可以使用此功能。

与Jmeter对比

这篇比较的文章比较不错,贴一下:性能测试工具Locust和JMeter比较,文章中提到并发用户数那,jmeter和locust实现的机制不一样,所以测得的值也是有差别的,一般同样并发数下jmeter的每秒请求数更高一些。
Locust开启服务只占用一个进程,如果机器是多核的,则可以开启master-slave模式,开一个master,剩下几核就开启几个slave,充分利用机器性能。下图是在本地笔记本(虚拟4核)上执行的普通模式master-slave模式的对比:
普通模式使用单核

master-slave模式充分发挥机器性能

master-slave模式下RPS最高值达到了2600,最低值1300,大概稳定在1900左右。

下面是使用Jmeter的一个测试数据:
Jmeter

Throughput峰值达到了3000,最低值1900,后续稳定在2200左右,但出现了报错,不过服务端那并没有报错。
上面三张图对比,可以看出在单核机器上如果使用Jmeter和Locust的话测得的数据应该相差不大,多核机器上Locust需要使用master-slave模式才能达到Jmeter的性能。不过说到底,两者之间对并发用户数的实现是不同的,测得的数据必定有差值。

后记

性能调优或测个粗指标的话,十八般工具都可以用,只要保证一直使用同一个工具测就行了。如果测精细指标的话,还得是工业级别的工具/测试仪来完成。

另外Locust + Boomer在相同并发数的情况下,RPS比Jmeter还要高,得益于Go语言天生的优势,可以直接使用Boomer示例中的client文件,获取编译好的文件及简单使用方法点击下面链接:
http://note.youdao.com/noteshare?id=38202fe283539461e3a49b800949bc9d&sub=683B4DBECED3468CACFC386E51F8B33E

共收到 6 条回复 时间 点赞

图片破损--

迷惘 回复

重新传了一遍,这次应该能看到了

这块资料比较少,数据准确性很重要

这篇主要写easy-locust这个工具的,顺带和jmeter比对了一下,关于locust和其他工具的详细比对,testhome上是有的,写的也挺不错的

我叫GTD 回复

我之前也想使用这个工具,你们在项目中有实际用起来么

怎么说呢,萝卜青菜各有所爱,我自己是在项目中使用过这个工具的,公司其他测试组用jmeter的居多,毕竟jmeter是一个工业级的软件,locust要想达到jmeter的性能就得分布式,locust工具达到瓶颈之后就添加slave就行了,jmeter也一样存在性能的瓶颈。另外我所在的项目组开发也在用这个工具了,有的接口较低的并发就会报错,我直接把这个工具甩给开发,调试到不报错了再找我测。

需要 Sign In 后方可回复, 如果你还没有账号请点击这里 Sign Up