使用Locust进行压力测试（二）复杂场景

最近一直在使用Locust进行压力测试，因此想总结一下学习和实践的一些成果。Locust官方文档

版本：
python: 3.*
locust: 0.9.0

1.不同类型的模拟用户
考虑一种测试场景：如果我们想测试一款游戏，需要并发100个人，并且使其中10个人在大厅闲逛，另外90个人玩比赛，用locust框架应该怎么实现呢？框架的命令行有提示locust [options] [LocustClass [LocustClass2 ...]]
当我们使用框架运行一个测试脚本时，如果脚本中只有一个Locust类，则框架会以这个类作为模拟用户的模板；如果该脚本中不止一个Locust类，则需要指定类名；如果需要模拟多种类型的用户，则可以指定多个类名，框架会根据每个Locust类的权重分配模拟用户的比例（默认权重为10）。
一个简单示例如下：

from locust import TaskSet, HttpLocust, task, TaskSequence, seq_task, InterruptTaskSet


class HangOutTaskSet(TaskSet):

    @task
    def hang_out(self):
        print("the player is hanging out")


class PlayGameTaskSet(TaskSet):

    @task
    def play(self):
        print("the player is playing game")


class PlayerOwn(HttpLocust):
    weight = 10
    task_set = HangOutTaskSet
    host = "http://www.baidu.com"
    min_wait = 10000
    max_wait = 10000


class PlayerTwo(HttpLocust):
    weight = 90
    task_set = PlayGameTaskSet
    host = "http://www.baidu.com"
    min_wait = 10000
    max_wait = 10000

命令行运行：locust -f ***.py PlayerOwn PlayerTwo
如果并发10人，就会有1人在闲逛，9人在玩游戏了

多种模拟用户运行示例

2.TaskSet的tasks属性
从（一）我们知道了一种定义任务的方式，就是在TaskSet类中定义一个方法，并标注其为@task；另一个定义任务的方式就是使用TaskSet的tasks属性，这个属性值可以是数组或字典，其中表示的任务可以是一个函数，也可以是另一个TaskSet。
如果使用数组，其元素可以是元组(callable, int)，int值即为该任务的权重。

1. 函数
   示例中的tasks = [(play_one, 5), play_two]与tasks = {play_one: 5, play_two: 1}是完全等价的
   tasks中的函数和有注解@task的函数有同等的地位，都是这个TaskSet的一个基本单位。

# 函数需有一个参数表示self
def play_one(self):
    print("play_one")


def play_two(self):
    print("play_two")


class PlayGameTaskSet(TaskSet):
    tasks = [(play_one, 5), play_two]

    @task
    def play(self):
        print("the player is playing game")


class PlayerTwo(HttpLocust):
    weight = 90
    task_set = PlayGameTaskSet
    host = "http://www.baidu.com"
    min_wait = 100
    max_wait = 100

2. 嵌套TaskSet
   tasks属性的元素也可以是另一个TaskSet，这种做法就会形成嵌套的子任务。
   例如：我们测试一款游戏，需要每一个用户在大厅闲逛的时候，有1/10的概率进入游戏比赛进行另外的操作，比赛结束退出后继续闲逛，示例如下：

class PlayGameTaskSet(TaskSet):
    tasks = [play_one, play_two]

    @task(5)
    def play_game(self):
        print("the player is playing game")

    @task
    def end_game(self):
        print("game end")
        self.interrupt()


class HangOutTaskSet(TaskSet):
    tasks = {PlayGameTaskSet: 1}

    @task(9)
    def hang_out(self):
        print("the player is hanging out")

当进入嵌套的任务集执行任务后，会根据嵌套任务集的任务权重执行任务，除非调用self.interrupt()或抛出中断异常raise InterruptTaskSet否则将一直在子任务集中执行下去

3.内部类
还有一种实现嵌套子任务的方式是使用内部类，如上诉示例，等同于：

class HangOutTaskSet(TaskSet):
    
    @task(9)
    def hang_out(self):
        print("the player is hanging out")

    @task
    class PlayGameTaskSet(TaskSet):
        tasks = [play_one, play_two]

        @task(5)
        def play_game(self):
            print("the player is playing game")

        @task
        def end_game(self):
            print("game end")
            self.interrupt()

4.类的继承
如果一个TaskSet的父类是另一个TaskSet，那么父类的所有任务会同样的成为子类的任务。

5.总结
在（一）中介绍了框架的几个角色，Locust类是创建模拟用户的模板，所以如果我们要使用不同的角色进行测试时，就需要给框架提供不同的Locust类。TaskSet类是模拟需要执行的任务集，我觉得可以这么理解，这是一个任务的池子，里面的每个单位都是一个任务，这个任务可以是函数也可以是另一个TaskSet，这个任务的定义方式有几个（当前TaskSet通过@task定义，继承父类的任务，tasks属性中一个元素），但它们都是同等地位的，执行的概率与它们的权重相同，每一次模拟用户都会从这个池子里挑选任务进行执行（直到stop_timeout或页面点击Stop）。当挑选到一个类型为TaskSet的任务执行时，就进入了嵌套的子任务当中，相当于到了子任务的池子里开始挑选任务执行（直到调用self.interrupt()或抛出中断异常raise InterruptTaskSet）。