学习使用locust作性能测试

Sun, 19 Feb 2017 12:06:13 +0000

之前用tsung做过性能测试，但是要用xml去配置接口参数，而且对字符的转义也做得不是很友好。后来听说有个locust性能测试工具很强大，于是在家试了一下。目前Locust只支持python2

首先我用Flask写了一个简单的接口,代码如下：

# coding: utf-8
from flask import Flask, request,jsonify
app = Flask(__name__)

@app.route('/')
@app.route('/learn_locust',methods=['GET','POST'])
def locust_test():
 _id = request.args.get('username')
 rv = {"code": {"data": {"username": _id, "created": "2017-02-19 18:00:00"}}}
 error = {"code":{"data":{"error":"not good"}}}
 if _id == 'bb8':
 return jsonify(rv)
 else:
 return jsonify(error)

if __name__ == '__main__':
 app.run(host='0.0.0.0', debug=True, port=5000)

启动flask服务，然后调用一下：

curl http://localhost:5000/learn_locust\?username=bb8

发现返回如下图：

嗯。没有问题，接下来就是写locust脚本。脚本如下：

from locust import HttpLocust, TaskSet, task
'''
首先从locust类中导入HttpLocust类，这个类用来做http协议的请求
然后再导入TaskSet类，定义一组用户要执行的任务集。这个类会执行用户的task属性的任务。TaskSet可以被嵌套，一个TaskSet的task装饰器可以嵌套另一个TaskSet。
TaskSet在执行task过程中的时候会有一个最小和最大等待时间，min_wait和max_wait参数。

task类，是可以很方便的让我们用装饰器去为TaskSet声明一个事务，他可以带上一个权重参数，如果权值越高，那么被选中的概率就越大
'''


class UserBehavior(TaskSet):
 def on_start(self):
 ''' on_start is called when a Locust start before any task is scheduled '''
 self.gslb()

 @task()
 def gslb(self):
 url = self.locust.host + '/learn_locust'
 ''' 设定好url， headers和url的参数'''
 headers = {
 'User-Agent': 'Dalvik/1.6.0 (Linux; U; Android 4.4.4; HM NOTE 1S MIUI/7.2.9)',
 'Connection': 'Keep-Alive',
 'Accept-Encoding': 'gzip',
 }

 params = {
 'username': 'bb8'
 }

 self.response = self.client.request(
 method='GET',
 url=url,
 headers=headers,
 params=params,
 )

 ### Additional tasks can go here ###


class WebsiteUser(HttpLocust):
 task_set = UserBehavior
 min_wait = 1000
 max_wait = 3000

然后将上述文件命令为locustfile.py，接着执行

性能测试 on Lost Temple

学习使用locust作性能测试