1、模块导入
from locust import HttpUser, task, constant, between, constant_pacing, events（监听器）LoadTestShape, tag（标签）import logging（模块）
2、用requests库封装一个登陆的方法
def login():import requestsres = requests.post(url="http://139.***.***.129:8899/adminapi/login",json={"account":"admin","pwd":"123456","imgcode":"A"})return res.json()["data"]["token"]
3、退出码设置（所有项目都是这么用不会有任何区别）完全拷贝就可以
"""
满足以下任何条件，则将退出代码设置为非零：
超过0.01的请求失败
平均响应时间大于2000ms
响应时间的第95个百分位数大于5000毫秒
"""
@events.quitting.add_listenerdef _(environment, **kw):if environment.stats.total.fail_ratio > 0.01:logging.error("Test failed due to failure ratio > 1%")environment.process_exit_code = 1elif environment.stats.total.avg_response_time > 2000:logging.error("Test failed due to average response time ratio > 2000 ms")environment.process_exit_code = 1elif environment.stats.total.get_response_time_percentile(0.95) > 5000:logging.error("Test failed due to 95th percentile response time > 800 ms")environment.process_exit_code = 1else:environment.process_exit_code = 0
4、setup，在整个负载测试开始前，会执行此方法
@events.test_start.add_listener # todo:在整个负载测试开始前的setup方法def on_test_start(environment, **kwargs):print('test start event called.')print('test start event finished.')
5、teardown，在整个负载测试开始后，会执行此方法（看情况）
@events.test_stop.add_listener # todo：在整个负载测试结束后的 teardown方法def on_test_stop(environment, **kwargs):print('test stop event called.')
6、创建一个类
class DoStressTestOnAPI(HttpUser): # todo： 继承HttpUser类，表明新创建的类是支持HTTP协议的API测试wait_time = constant_pacing(1) # 系统自动调配等待时间保证每秒中每个user instance最多只能有一个task在执行global project_host # 声明一个全局变量host ="http://139.***.***.129:8899/" # todo： 定义项目地址headers = { # todo: 放置请求头"Accept": "application/json, text/plain, */*","Authori-zation": "Bearer " + login(), #（这把token信息放下来了）}def on_start(self): # todo:每个用户启动都会调用此方法 on_start则是每个用户开始时运行（开始前seatup方法）print('test started.')def on_stop(self): # todo:每个用户启动都会调用此方法 on_stop则是每个用户停止时运行（结束后的thendang方法）logging.info('test stopped')
用例逻辑层：
"""
tag：表示这个用例的名字，在运行时，可以通过相同的tag名称来执行性能测试场景
task：权重的设置。locust的权重表示着被用户选择并请求的概率，随着权重越高，则被用户请求的几率就会越高。
locust的权重设置比较灵活，可以设置每个方法接口访问的比例
catch_response：
布尔参数
<1>如果设置为True，则表示，如果响应代码是正常（2XX），则标记通过。
<2>如果设置为False，则表示，如果响应代码是正常（2XX），则标记不通过。
name：
name分组，如果API传参是在url后传递，也就是pathvariable。可以加上name做一个分组统计，最后报告会计算共请求了多少次
"""
7、标签
@tag("商品管理-商品搜索","1") # tag标签>（运行的标签设置）@task(1) # todo：运行DoStressTestOnAPI时，会从多个task（表示权重）任务中随机选择一个，权重增加了他的选择几率（task值越高被用户选择的概率越大）def Commodity_Search(self): # Commodity_Search接口名称命名with self.client.get(url = f"adminapi/product/product",headers=DoStressTestOnAPI.headers, # with语句 get请求 url=接口 headers=请求头data={"page":1,"limit":15,"cate_id":"","type":1,"store_name":"","excel":0})\ # data请求传参as response:logging.info('+++++++++++++++++' + str(response.status_code))logging.info(response.request.headers) #日志print(response.json())if response.status_code == 200 and response.json()['status'] == 200:# response.success()logging.info(response.json()) # 成功，写在logging日志里else:response.failure('test failed and response code is: ' + str(response.json()['code']))logging.info('error occured') # 失败，写在失败里
8、用例
@tag("商品分类-商品搜索","2",'1','3')@task(2)def Commodity_Classification(self):with self.client.get(url=f"adminapi/product/category", headers=DoStressTestOnAPI.headers,data={"pid": "", "is_show": "", "page": "1", "cate_name": "", "limit": "15"}) \as response:logging.info('+++++++++++++++++' + str(response.status_code))logging.info(response.request.headers)print(response.json())if response.status_code == 200 and response.json()['status'] == 200:# response.success()logging.info(response.json())else:response.failure('test failed and response code is: ' + str(response.json()['code']))logging.info('error occured')
如果觉得《性能测试脚本编写》对你有帮助，请点赞、收藏，并留下你的观点哦！
