做接口的性能测试时，常需要先登录认证。
一种方式是：先调用登录接口，从响应头中获取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
