python locust提取参数_python locust 性能测试：locust 关联---提取返回数据并使用

from locust import HttpLocust, TaskSet, task

import json

from common import readConfig

class UserBehavior(TaskSet): # 定义用户行为

def on_start(self): # 当模拟用户开始执行TaskSet类的时候，on_start方法会被调用

pass def get_headers(self):　　 """会员登录""" headers = { "Content-Type": "application/json", "channel": "SHOP" } body = { "unionid": readConfig.unionid # 这样使用会报错，没有 common ... } res = self.client.post(‘/customers/login‘, headers=headers, data=json.dumps(body)).text assert ‘成功‘ in res # 断言，判断接口返回是否成功 res = json.loads(res) uid = res[‘data‘][‘uid‘] ukey = res[‘data‘][‘ukey‘] return [uid, ukey] # 会员登录返回的uid和ukey @task(1) # @task() 装饰该方法为一个任务;1表示一个Locust实例被挑选执行的权重，数值越大，执行频率越高; def members_can_withdraw_account_information(self): # 一个行为 """使会员获取会员状态""" res = self.get_headers() headers = { "channel": "SHOP", "uid": str(res[0]), "ukey": res[1] } response = self.client.put("/customers/customer_state", headers=headers).text # client.get()用于指请求的路径，可加headers,params,body参数; assert ‘成功‘ in response @task(1) def get_member_account_withdrawal_details(self): """获取会员账户提现明细""" res = self.get_headers() headers = { "channel": "SHOP", "uid": str(res[0]), "ukey": res[1] } params = { "page": "10", "size": "10" } response = self.client.get("/customers/account/withdraw/log", headers=headers, params=params).text assert ‘成功‘ in responseclass WebsiteUser(HttpLocust): # 设置性能测试; host = "http://dev.sign.lixiaofeng.com" task_set = UserBehavior # 指向一个定义了的用户行为类; min_wait = 3000 # 用户执行任务之间等待时间的下界，单位：毫秒; max_wait = 6000 # 用户执行任务之间等待时间的上界，单位：毫秒;