原博客地址：https://testerhome.com/topics/4839

背景

最近自己開發了一個小的接口，功能測完了，突然想測下性能，原來做性能測試，我一直用的是HP的LoadRunner，前一段時間正好看過locust，想想就用這個來測測性能吧。
由於對LR比較熟，正好做個對比，這樣更利於對新東西的理解。

基礎

locust 的官網：http://locust.io/

也可以參考論壇里其他同學的介紹：https://testerhome.com/topics/2888

目前locust還只支持Python 2版本。

測試需求

驗證在相同的服務器端的情況下，使用LR和locust分別進行性能測試，在相同並發用戶的情況下，驗證平均響應時間，TPS值等性能測試指標的差異。
為了方便，使用http協議，一個get請求，一個post請求，交易比例為1：1。

服務器端

為了簡單易理解，用Python的bottle框架寫了一個服務器端，2個交易，一個get，一個post請求，交易中加了2個不同的sleep。
代碼如下：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 'among,lifeng29@163.com'

from bottle import *
from time import sleep

app = Bottle()


@app.route('/transaction_1', method='GET')
def tr1():
    sleep(0.2)
    resp = dict()
    resp['status'] = 0
    resp['value'] = 'xxx'
    return resp


@app.route('/transaction_2', method='POST')
def tr2():
    parm1 = request.forms.get('parm1')
    parm2 = request.forms.get('parm2')
    sleep(0.5)
    resp = dict()
    resp['status'] = 0
    resp['value'] = 'yyy'
    return resp


run(app=app, server='cherrypy', host='0.0.0.0', port=7070, reloader=False, debug=False)

服務器端部署在一個單獨的Windows的機器中，基於Python 3，啟動后，監聽7070端口。

LR中的測試腳本

在另外的一個Windows機器中，使用LR 11，用的是http/html協議的腳本，主要代碼如下：
用了2個action，用於划分交易比例。
action1:

Action1()
{
    lr_start_transaction("get");
    web_reg_find("Text=xxx",
        LAST);
    web_custom_request("Head",
        "URL=http://10.0.244.108:7070/transaction_1", 
        "Method=GET",
        "Resource=0",
        "Referer=",
        LAST);
    lr_end_transaction("get", LR_AUTO);
    return 0;
}

action2:

Action2()
{
    lr_start_transaction("post");
    web_reg_find("Text=yyy",
        LAST);  
    web_custom_request("Head",
        "URL=http://10.0.244.108:7070/transaction_2", 
        "Method=POST",
        "Resource=0",
        "Referer=",
        "Body=parm1=123&parm2=abc",
        LAST);
    lr_end_transaction("post", LR_AUTO);
    return 0;
}

使用1:1的比例設置2個transaction的執行比例：

LR中的執行方法，直接放到場景中，執行即可。

locust中的測試腳本

在另外的mac中，使用locust執行測試，全部通過代碼實現。代碼如下：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 'among,lifeng29@163.com'

from locust import *

class mytest(TaskSet):
    @task(weight=1)
    def transaction_1(self):
        with self.client.get(name='get', url='/transaction_1', catch_response=True) as response:
            if 'xxx' in response.content:
                response.success()
            else:
                response.failure('error')

    @task(weight=1)
    def transaction_2(self):
        dt = {
            'parm1': '123',
            'parm2': 'abc'
        }

        with self.client.post(name='post', url='/transaction_2', data=dt, catch_response=True) as response:
            if 'yyy' in response.content:
                response.success()
            else:
                response.failure('error')


class myrun(HttpLocust):
    task_set = mytest
    host = 'http://10.0.244.108:7070'
    min_wait = 0
    max_wait = 0