Losust性能脚本demo

　　我们实际工作中接口测试经常会遇到这样的问题，一个接口的响应值作为另一个接口的请求值，而且处于动态变化中，这样就需要我们编写方法将变量提取出来引用了，否则复制、粘贴将是一个很无聊的过程。性能测试则更是无法运行，所以，提取、引用参数是我们必须要会的点。

　　最近在学习locust性能压测，结合实际工作编写了本demo，实现获取登录tocken值，拼接后作为请求头参数参与其他接口请求具体代码如下：

from locust import HttpUser, task
import json


class SearchTest(HttpUser):

    def on_start(self):
        pass

    def get_authorization(self):
        """
        执行login
        :return:access_token
        """
        login_interface = "/auth/login"
        post_data = {"username": "trader1", "password": "admin123"}
        request_headers = {"Content-Type": "application/json;charset=UTF-8"}
        with self.client.post(url=login_interface, headers=request_headers, json=post_data,
                              catch_response=True, name="登录") as response:
            if json.loads(response.text).get("code") == 200:
                print("---------------登录成功--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
                access_token = json.loads(response.text).get("data")["access_token"]
                authorization = "Bearer " + access_token
                return authorization
            else:
                print("---------------登录失败--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
                print("response_massage:" + str(json.loads(response.text).get("msg")))

    def on_stop(self):
        """
        执行logout
        :return: response
        """
        logout_interface = "/auth/logout"
        with self.client.delete(url=logout_interface, name="登出") as response:
            if json.loads(response.text).get("code") == 200:
                print("---------------登出成功--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
                print(response.text)
            else:
                print("---------------登出失败--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
                print("response_massage:" + str(json.loads(response.text).get("msg")))

    @task(1)
    def search_interface(self):
        """
        任务一：查询接口,GET方法
        :return: 查询接口性能压测数据
        """
        get_url = "/system/entry/37"
        authorization = self.get_authorization()
        print("认证标识 Authorization：" + authorization)
        request_headers = {"Authorization": authorization,
                           "Content-Type": "application/json;charset=UTF-8"}
        with self.client.get(url=get_url, headers=request_headers, catch_response=True, name="入库单查询") as response:
            if json.loads(response.text).get("code") == 200:
                print("---------------查询成功--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
            else:
                print("---------------查询失败--------------")
                print("response code:" + str(json.loads(response.text).get("code")))
                print("response_massage:" + str(json.loads(response.text).get("msg")))

    @task(1)
    def create_form(self):
        """
        任务二
        :return:
        """
        pass

运行服务，打开服务地址，录入参数，点击“start swarming”开始测试（具体参数含义及配置请参考我之前的locust基础篇）：

以下是执行结果：

以上，希望对大家能有所帮助。