以下是一个使用 Locust 封装的简单性能测试脚本示例:
from locust import HttpUser, between, task
class MyUser(HttpUser):
wait_time = between(1, 3)
@task
def my_task(self):
self.client.get("/")
def on_start(self):
print("Starting...")
def on_stop(self):
print("Stopping...")
这个脚本定义了一个名为 "MyUser" 的用户类,继承自 Locust 提供的 HttpUser 类。其中,wait_time 定义了用户任务之间的等待时间范围;@task 装饰器定义了一个用户任务,即发送 GET 请求访问根路径 "/";on_start 和 on_stop 方法分别在用户启动和停止时执行。
要运行这个脚本,可以在命令行中进入脚本所在目录,然后输入以下命令:
locust -f script.py
其中,-f 参数指定要运行的脚本文件名。然后在浏览器中打开 Locust 界面(默认地址为 http://localhost:8089),设置虚拟用户数量、每秒请求数等参数并启动测试即可。