Locust是纯运行在python语言下的测试工具,采用了gevent开发框架,底层利用greenlet微线程机制,单机并发可以支持非常高,直接HTTP等协议接口,特点就是协程,web可视化管理工具。
What is Locust? — Locust 2.10.3.dev11 documentation
from locust import task, TaskSet, HttpUser
class UserBehavior(TaskSet):
@task
def get_slots(self):
header = {
"Content-Type": "application/json"
}
data = {
"query": "测试"
}
// /test根据自己情况而定
response = self.client.post("/test", headers=header, json=data)
print(response.json())
class WebsiteUser(HttpUser):
tasks = [UserBehavior]
min_wait = 1000 # 最小等待时间
max_wait = 3000 # 最大等待时间
host = "http://ip:port" # 服务器地址,也可以在写命令时 指定host地址
if __name__ == '__main__':
import os
os.system('locust -f test/test_pressure.py') # 此处导入os,可以在pycharm中直接运行此py文件