性能测试工具locust简单应用

简介

• Locust是一种易于使用的分布式用户负载测试工具。可用于对网站（或系统）负载测试，并依据响应数据计算出系统支持的并发用户数。

---

安装及调试（以下操作在windows环境下进行）

• Locust基于python语言发开发，只需要装有python环境即可通过以下命令安装：

  1. pip install locustio

  提示安装成功后，可运行命令查看可用选项：

  1. locust --help

• 调试代码
  在locust中，所有用户行为都是通过python代码定义。可以通过locust命令和web端的UI界面提供的几个参数（虚拟用户总数/孵化率/Host）获取的请求信息，来模拟生成大量虚拟用户。以下是结合青享云平台编写的调试代码：
  Example locustfile.py

import random
from locust import HttpUser, task, between
class QuickstartUser(HttpUser):
 wait_time = between(5, 9)
 # 要测试的场景1
 @task
 def index_page(self):
     self.client.get("/3d/consumer.html")
 # 要测试的场景2
 # task带的数值代表当前任务在压测过程中的执行比重
 # 本例中表示index_page任务与view_item任务压力比为1:3
 @task(3)
 def view_item(self):
     self.client.get("[商品详情页地址]")
 # 每个任务执行脚本执行的操作，类似于setUp()方法，同理on_stop()为任务结束后执行的操作
 def on_start(self):
     self.client.post(
                 url='地址',
                 json={"username": "账号", "password": "密码", "captcha": "", "remember": True, "requestId": ""})

当前文件名为locustfile的python文件是locust默认运行的task文件，在当前目录下命令行执行命令locust即可运行启动locust服务；如果创建的文件名和位置不在当前运行路径下，可通过命令行控制运行，如下：

locust -f practice/my_qdcloud_locustfile.py

此时，可以在浏览器中访问本地8089端口地址 http://localhost:8089 打开locust UI界面，填写参数，即可开始测试：

参数说明：

Number of total users to simulate：需要创建的虚拟用户总数（即并发数）
Spawn rate：孵化率（每秒创建的虚拟用户数），所有的虚拟用户每秒创建多少
Host：base地址，代码中的地址优先级高于此处

执行后结果信息展示如上图所示，locust自带有数据表、统计图、fails统计信息、异常请求的展示界面、测试数据下载，感兴趣的同学自行百度。

核心参数：

Type：请求类型，即接口的请求方法；

Name：请求名称；
requests：当前已完成的请求数量；
fails：当前失败的数量；
Median：响应时间的中间值，即50%的响应时间在这个数值范围内，单位为毫秒；
90%ile：根据正态分布，90%的响应时间在正态分布平均值下方，即小于这个值；
Average：平均响应时间，单位为毫秒；
Min：最小响应时间，单位为毫秒；
Max：最大响应时间，单位为毫秒；
average Size：平均每个请求的数据量，单位为字节；
current RPS（requests per second）：每秒钟处理请求的数量，即RPS。

---

实例代码：

import time
import execjs
from locust import HttpUser, task, between
import requests
class QuickstartUser(HttpUser):
    @task
    def scan_goods_page(self):
        self.client.get('路径')
        # wait_time = between(2, 5)
    """
    # 订单生成任务请求
    @task
    def create_orders(self):
  
        header = {
            "Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5......",
            "Content-Type": "application/json;charset=UTF-8"}
        self.client.request(
            method="post",
            headers=header,
            url="路径",
            data='{"totalFees":"0.01","totalCount":1,"cashFees":"0.01","shippingFees":"0.00","deliveryAddrId":483,'
                 '"receGoodsContacts":"黑色桌子","receGoodsTel":"13299990000",'
                 '"receGoodsAddr":"河北省｜秦皇岛市｜北戴河区｜北戴河北戴河北戴河北戴河北戴河北戴河北戴河北戴河","skuAndCount":[{"skuId":27,'
                 '"defaultPicUri":"地址'
                 '13:38:49_af80dc87c10f48eab55093713973db9d.jpg","goodsName":"卡萨帝BCD-551WDCTU1","sellPrice":"0.02",'
                 '"earnest":"0.01","count":1,"subtotal":"0.01","sellModeFlag":"1","brandId":8,"shelvesId":1,'
                 '"partnerId":1,"specList":[{"specName":"尺寸","specValue":"475"},{"specName":"颜色","specValue":"白色"},'
                 '{"specName":"容量","specValue":"100ml"}]}],"invoice":{"invoiceType":"1","invoiceTitle":"个人",'
                 '"invoiceTitleCode":"","invoiceTitleType":"0","invoiceContent":"商品明细","customerName":"黑色桌子",'
                 '"customerPhone":"13299990000","companyName":"","dutyCode":"","bank":"","bankAccount":"",'
                 '"consignee":"","mobile":"","email":"","registeredAddress":"","registeredMobile":"","pcd":"",'
                 '"province":"","city":"","district":"","address":"","orderCode":"","orderId":"","userId":""}}')
        time.sleep(2)
    """
    def GetCookie():
        s = requests.session()
        loginUrl = 'url'
        postData = {"username": "用户名", "password": "密码", "captcha": "", "remember": True, "requestId": ""}
        rs = s.post(loginUrl, postData)
        requests.cookies.RequestsCookieJar()
        c = requests.cookies.RequestsCookieJar()  # 利用RequestsCookieJar获取
        c.set('cookie-name', 'cookie-value')
        s.cookies.update(c)
        print(s.cookies.get_dict())
        # print(c.get("ibps-3.3.3-token")
        # print(s.cookies.get_dict())
    def on_start(self):
        with self.client.post(url='地址',
                              json={"username": "15866800632", "password": "1a2s3d4f", "captcha": "", "remember": True,
                                    "requestId": ""},
                              catch_response=True) as response:
            if response.status_code != 200:
                print("失败！请求状态码是：%s " % response)
            else:
                print("请求结果成功！")
                # return print(response)
                s = requests.session()
                loginUrl = '地址'
                postData = {"username": "15866800632", "password": "1a2s3d4f", "captcha": "", "remember": True,
                            "requestId": ""}
                rs = s.post(loginUrl, postData)
                requests.cookies.RequestsCookieJar()
                c = requests.cookies.RequestsCookieJar()  # 利用RequestsCookieJar获取
                c.set('token', 'e09c769b-8191-42de-9ed1-3660067010a3')
                s.cookies.update(c)
                # get_token = execjs.compile(open(r"auth.js").read()).call('getToken', '')
                return print(s.cookies.get_dict())
    def on_stop(self):
        status = self.client.post(url="地址",
                                  data={'access_token': 'e09c769b-8191-42de-9ed1-3660067010a3'})
        return print(status.status_code)

以上脚本可实现压测场景1、页面访问；2、订单生成场景；3、登录、退出