最近一直在使用Locust进行压力测试,因此想总结一下学习和实践的一些成果。Locust官方文档
版本:
python: 3.*
locust: 0.9.0
1.TaskSet的属性
按照官方的说法,在TaskSet类中也可以设置min_wait、max_wait等属性,但是我实验的时候发现未起效,所以不建议这种做法。
2.setup和teardown函数
Locust类和TaskSet类都可以定义setup函数和teardown函数,用setup和teardown可以做一些准备(如创建一个数据库)和清理环境(如删除一个数据库)等等。不同与每次开始任务集都会执行的on_start,和每次运行结束时会被被调用的on_stop,setup函数与teardown函数都仅执行一次,执行顺序如下:
Locust setup # 第一次启动Locust的时候执行
TaskSet setup # 第一次实例化任务集时执行
TaskSet on_start # 每一次开始一个任务时执行
TaskSet tasks…
TaskSet on_stop # 点击页面stop时,当前所有在执行中的TaskSet执行
TaskSet teardown # 停止locust运行时执行
Locust teardown # 停止locust运行时执行
示例如下:
class MyTaskSet(TaskSet):
def setup(self):
print("task set_up")
def on_start(self):
print("start")
@task
def one(self):
print("one")
@task(3)
def two(self):
print("two")
def on_stop(self):
print("stop")
def teardown(self):
print("task tear_down")
class WebSiteUser(HttpLocust):
task_set = MyTaskSet
host = "http://www.baidu.com"
min_wait = 10000
max_wait = 10000
def setup(self):
print("locust set_up")
def teardown(self):
print("locust tear_down")
3.TaskSequence:顺序执行的TaskSet
如果我们需要执行的任务是按照一定顺序执行的,那么就可以继承TaskSequence类来实现。示例如下:
class PlayGameTaskSet(TaskSequence):
tasks = [play_one, play_two]
@seq_task(1)
def login(self):
print("login")
@seq_task(2)
@task(5)
def play_game(self):
print("the player is playing game")
@seq_task(3)
@task
def end_game(self):
print("game end")
tips:
- 使用@seq_task指定执行顺序,如果没有该注解,默认顺序是1;相同的order,会根据字母顺序排序
- 使用@task指定执行次数(!!!),由于不再是随机选择任务执行,框架会直接使用task的权重作为执行次数,如果没有该注解,默认执行1次
- 当顺序执行完一轮任务后,会重头开始新一轮的执行任务
- 如果方法即没有@task也没有@seq_task,那么这个方法就不会被认为是一个task,不会出现在执行序列当中
上述示例执行情况如下: