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

Posted 2021-01-02 backlight

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 response


class WebsiteUser(HttpLocust):  # 设置性能测试;
    host = "http://dev.sign.lixiaofeng.com"
    task_set = UserBehavior  # 指向一个定义了的用户行为类;
    min_wait = 3000  # 用户执行任务之间等待时间的下界，单位：毫秒;
    max_wait = 6000  # 用户执行任务之间等待时间的上界，单位：毫秒;