python压测工具locust基础

     因近期公司需要对ocr类产品(python语言编写的接口)进行压测,而之前的压测工具就meter、loadrunner对python语言的代码都不太兼容,故而找到了locust工具。

     使用之前在网上对locust工具进行了些了解,发现它还是比较简单的,入门很快,轻便,而且还有较多人使用,相信测试结果还是比较准确的,误差不会太大。于是,开始着手准备了。

      因locust在linux环境下运行较稳定,故我选择安装 在linux环境中。

      第一步:安装locust   pip3  install locust    安装后locust --version查看当前Locust版本

      第二步,编写脚本

    

# -*- coding: UTF-8 -*-
import requests
from locust import HttpLocust,TaskSet,task
from requests.packages.urllib3.exceptions import InsecureRequestWarning
import base64

# 禁用安全请求警告
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

image_path = '/home/ocrpython/test/2.JPG'
image_backpath = '/home/ocrpython/test/4.jpg'

def img_to_b64(img_path):
    with open(img_path, 'rb') as f:
        base64_data = base64.b64encode(f.read())
    return base64_data.decode('utf-8')


class set_task (TaskSet):
    @task(1)
    def post_api (self):
        # 定义请求头
        data_dis = {"img": img_to_b64(image_path), "orderNo": "test101901080933541948707600000004"}
        req = self.client.post("/id_ocr_batch", data=data_dis)
        http_code = req.status_code
        if http_code != 200:
            print("response null !!!")
        else:
            print(req.text)

    @task(1)
    def post_backapi(self):
        # 定义请求头
        data_dis1 = {"img": img_to_b64(image_backpath), "orderNo": "back102002200933541948707600000004"}
        req = self.client.post("/id_ocr_batch", data=data_dis1)
        http_code = req.status_code
        if http_code != 200:
            print("response null !!!")
        else:
            print(req.text)

class apiUser(HttpLocust):
    task_set = set_task
    min_wait = 5
    max_wait = 50
    host = 'http://XX.XX.XX.XX:5002'


if __name__ == '__main__':
    import os
    os.system("locust -f /home/ocrpython/locust_idocr.py")

    set_taskL类继承TaskSet类,用于描述用户行为:

1、使用@task装饰的方法为一个事务,方法的参数用于指定该行为的执行权重,如上图,正反面的比例为1:1.

2、apiUser类用于设定性能测试属性

3、此脚本只能在装有locust的环境下运行,不能在pycharm内调试,故只能运行后看代码是否报错咯,若无报错,则性能测试脚本就ok了,则进行下一步

第三步:启动  

进入代码目录,输入 locust -f  bokeyuan.py  --host=‘’http://XX.XX.XX:5002‘’

第四步:打开locust的web页面

使用上面的命令行启动Locust之后,应该打开浏览器并将其指向http://XX.XX.XX.XX:8089(XX.XX.XX.XX为装有locust的机器地址)

  • Number of users to simulate :设置模拟用户数;
  • Hatch rate(users spawned/second) :每秒产生(启动)的虚拟用户数;

  设置好模拟用户后,点击Start swarming开始测试

 

 

 

你可能感兴趣的:(locust,python基础)