最近一直在使用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值即为该任务的权重。
- 函数
示例中的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
- 嵌套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）。
网友评论