Locust 处理登录问题

做接口的性能测试时，常需要先登录认证。
一种方式是：先调用登录接口，从响应头中获取token信息；再将token写入调用接口的请求头中。这样操作的话比较麻烦。

所以我们换种方式来。

首先，Locust使用的是requests库来发送http请求的。同时我在Locust的clients模块中找到一个HttpSession类，它继承自requests.Session类。
所以，聪明的我们就应该想到了：直接初始化一个成功登陆的session,再使用这个session发起之后所有需要登录的请求即可。

下面我们就来演示一下：
新建一个模块login_demo.py
添加如下代码

# 引入需要的包
from locust import TaskSet, HttpLocust, task
from locust.clients import HttpSession

uri = '/api/to/be/tested'

user = 'username'
pwd = 'password'


class PendingTask(TaskSet):
    host = 'http://host.url.com'
    session = None
    # on_start() 在测试前的初始化，我们在这里生成包含认证信息的 session
    def on_start(self):
        # 报文请求数据，根据自己的登录接口修改
        payload = "account={account}&password={pwd}&simplePwd=false&rememberMe=0".format(account=user, pwd=pwd)
        # 登录认证的uri
        login_uri = "/login/valid"
        # 设置请求头信息，登录认证的报文数据类型指定为 x-www-form-urlencoded（根据自己的接口定义来）
        headers = {
            'Content-Type': "application/x-www-form-urlencoded",
            'Referer': "{}/login".format(self.host),
        }
        # 设置sesssion
        self.session = HttpSession(self.host)
        # 使用 session 来登录
        # 登录成功之后，就可以直接使用了
        response = self.session.post(url=self.host + login_uri, data=payload, headers=headers)

        print("LOGIN RESULT:", response.status_code, response.json())
     
    # 写一个简单的查询数据列表的接口来测试一下
    @task
    def test_get_pending_list(self):
        headers = {
            'Content-Type': "application/x-www-form-urlencoded; charset=UTF-8",
        }

        payload = "page=1&rows=10&sort=CREATE_TIME_&order=desc"
        # 直接使用session来发起请求，这里的session已经是保持登录状态了
        response = self.session.post(uri, headers=headers, data=payload.encode('utf-8'))

        print('查询结果 :', response.status_code, response.text)
        return response


class BasicHttp(HttpLocust):
    task_set = PendingTask
    host = 'http://host.url.tobetested'

跑起来：locust -f login_demo.py --web-host 0.0.0.0 -P 9000

用浏览器打开 http://localhost:9000

看看效果：

1.png 2.png

参考：locust httpsession-class