python模块学习之locust性能测试

背景

安装

pip install locustio

测试代码

#! /usr/bin/python
# coding:utf-8
"""
@author:Bingo.he
@file: locust_practis.py
@time: 2017/12/29
"""
from locust import HttpLocust, TaskSet, task
import hashlib

class UserBehavior(TaskSet):

@staticmethod
def param_order(dic):
"""
参数转换函数
:param dic: 字典类型参数
:return: key1=value1&key2=value2
"""
l = []
if isinstance(dic, dict):
for k, v in dic.items():
l.append(k + '=' + str(v))
l.sort()
# print ('&'.join(l))
return '&'.join(l)
else:
return 'Not a dict'

def get_signMsg(self, soure_para, bizKey):
"""
API密钥生成函数
:param soure_para : 传递的参数
:param bizKey:
:return: str
"""
param = self.param_order(soure_para)
return hashlib.md5((param + bizKey).encode('utf8')).hexdigest()

def pack_param(self):
soure_para = {
'bizNo': '0000001',
'mobiles': '18565100000,18565100001,18565100002,18565100003,18565100004,18565100005,18565100006,','content': u'{}',
}
sign_msg = self.get_signMsg(soure_para, 'GyCK9XC4qYbs')
soure_para['signMsg'] = sign_msg  # 添加签名信息
return soure_para

@task(1)  # 表示第一个执行，可以通过它组合指定场景
def sms(self):
param = self.pack_param()
self.client.post("/sms", param)

class WebsiteUser(HttpLocust):
task_set = UserBehavior
min_wait = 300  # 不同请求间间隔时间
max_wait = 600

运行命令

locust -f ../locust_files/locust_practice.py --host=http://example.com