使用Locust进行压力测试（三）其他

最近一直在使用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，不会出现在执行序列当中

上述示例执行情况如下：

TaskSequence执行结果