Locust文件就是一般的Python文件。唯一的需求就是它至少需要一个继承于Locust的类.

Locust类

Locust类代表一个用户(如果愿意，也可以是一个准备出动的蝗虫)。Locust会为每一个模拟用户生成一个locust类实例。同时会有一些locust类属性被定义。

task_set属性

task_set属性是指向一个定义用户行为的TaskSet类，下面会有详细的介绍。

min_wait和max_wait属性

除了task_set属性，另外一个经常被使用的就是min_wait和max_wait属性。是用于各自以毫秒为单位的最小值和最大值，一个模拟用户将会在每个任务执行时的等待执行的时间间隔。min_wait和max_wait默认设置为1000，如果不声明的话，Locust会默认在每个任务间等待1秒。

参考下面的代码，每个用户将会在每个任务间等待5至15秒:

from locust import Locust, TaskSet, task_set

class MyTaskSet(TaskSet):
    @task
    def my_task(self):
        print "executing my_task"

class MyLocust(Locust):
    task_set = MyTaskSet
    min_wait = 5000
    max_wait = 15000

min_wait和max_wait属性可以用于重写TaskSet类。

weight属性

你可以通过同一个文件来运行两个locust，就像这样:

locust -f locust_file.py WebUserLocust MobileUserLocust

如果你更倾向于用这种方法来运行，便可以在这些类中尝试weight属性。比如，就像这样来定义web用户比Mobile用户多3倍：

class WebUserLocust(Locust):
    weight = 3
    ...

class MobileUserLocust(Locust):
    weight = 1
    ...

host属性

host属性是到要加载目标URL的前缀(如:"http://google.com")。通常情况下，当Locust被启动时，在命令行中是需要通过--host来指定的。如果host属性在locustfile文件中被声明，则在命令行中则不需要使用--host属性来再次声明。

TaskSet类

如果Locust类代表一只准备出动的蝗虫，那么你可以说TaskSet类代表蝗虫的大脑。每一个Locust类中必须要包含一个指向TaskSet的task_set属性设置。

TaskSet就像它的名字一样，是一个任务集合。这些任务是常规的Python调用,如果我们压力测试一个拍卖网站，便可以做这些操作加载启动页面、搜索一些产品、竞标。

当一个压力测试被启动时，每一个准备的Locust类实例将会开始执行它们的TaskSet。接下来是每一个TaskSet找到它的task并调用它。它将在min_wait和max_wait属性值之间随机等待几毫秒(除非min_wait和max_wait被定义在TaskSet中，在这种情况下将会使用TaskSet设置的值)。然后，它将会找到一个新task并调用，再次等待，一直这样持续下去。

声明task

对于TaskSet来说，典型的声明task的方法是直接使用task。

参考这个例子:

from locust import Locust, TaskSet, task
class MyTaskSet(TaskSet):
    @task
    def my_task(self):
        print "Locust instance (%r) executing my_task" % (self.locust)

class MyLocust(Locust):
    task_set = MyTaskSet

@task 将会获取一个可选的权重参数，用于说明任务执行的比率。在下面的例子中 task2 将会比 task1 执行的次数多两倍:

from locust import Locust, TaskSet, task
class MyTaskSet(TaskSet):
    min_wait = 5000
    max_wait = 15000

    @task(3)
    def task1(self):
        pass

    @task(6)
    def task2(self):
        password

class MyLocust(Locust):
    task_set = MyTaskSet

task属性

使用@task操作符来声明task是一种便捷的方法，并且经常是最好的方式。然而，也可以定义TaskSet中的task通过设置tasks属性(使用操作符@task比tasks属性更流行)。

tasks 属性不是python列表的调用就是一个<callbale:int>字典。tasks是python调用接收执行task的TaskSet类实例参数。下面是一个极其简单的示例(不会影响任何测试):

from locust import Locust, TaskSet

def my_task(l):
    pass

class MyTaskSet(TaskSet):
    tasks = [my_task]

class MyLocust(Locust):
    task_set = MyTaskSet

如果task属性被定义在列表中，每次任务被执行时，将会随机从 tasks 属性中选择。如果 tasks 是一个带有关健字和数值调用的字典，被执行的任务将会被随机选择以数字的比率来执行。就像下面的这样:

{my_task: 3, another_task:1}

my_task 将会比 another_task 多执行三倍。

TaskSet可以嵌套

TaskSet有一个重要的属性就是可以被嵌套，由于真实的网站是有一定的业务层级结构的，并带有一些子模块。嵌套的TaskSet将会帮助我们来定义更加真实的用户行为。比如，我们可以定义TaskSet像下面的结构

• Main user behaviour
  • Index page
  • Forum page
    • Read thread
      • Reply
    • New thread
    • View next page
  • Browser categories
    • Watch movies
    • Filter movies
  • About page

嵌套TaskSet的方法就像使用task属性来说明task一样，但代替参考Python函数，你可以参考下面的TaskSet:

class ForumPage(TaskSet):
    @task(20)
    def read_thread(self):
        pass

    @task(1)
    def new_thread(self):
        pass

    @task(5)
    def stop(self):
        self.interrupt()

class UserBehaviour(TaskSet):
    tasks = {ForumPage:10}

    @task
    def index(self):
        pass

在上面的示例中，当UserBehaviour的TaskSet执行时，ForumPage会被选中来执行，接下来ForumPage的TaskSet将会开始执行。ForumPage的TaskSet会找到它的tasks并执行它，再等待，一直这样持续下去。

针对上面的例子中有一个重要的事情要注意，就是在ForumPage页面中的Stop方法中调用self.interrupt()。这个做的事情是停止执行ForumPage任务并在UserBehaviour实例中继续执行。如果在ForumPage中，我们没有调用interrupt()方法，除非被调用否则Locust不会调用ForumPage任务。但通过interrupt函数 ，我们可以结合weight任务来定义模拟用户离开Forum.

也可以在类内部声明嵌套TaskSet,通过使用@task操作符，像声明正常的task一样：

class MyTaskSet(TaskSet):
    @task
    class SubTaskSet(TaskSet):
        @task
        def my_task(self):
            pass

on_start函数

TaskSet可以选择声明on_start函数。如果这样的话，当模拟用户开始执行TaskSet类时，函数被调用。

关联Locust实例，或父TaskSet实例

TaskSet实例有locust属性来指向它的Locust实例，属性parent用来指向它的父TaskSet（它会指向Locsut实例，在基类TaskSet中）。

HTTP请求

到现在为止，我们仅覆盖了一个Locsut用户的部分任务计划。为了真实的压力测试一个系统时，我们需要生成HTTP请求。为了帮助我们实现这个功能，可以使用HttpLocust类。当使用这个类时，每一个实例将会获得一个用于生成Http请求的HttpSession实例的client属性。

class HttpLocust

表示一个用于压力测试的孵化和攻击系统的HTTP 用户。

这个用户的行为通过task_set属性来定义，直接指向TaskSet类。

这个类创建一个client属性，在初始化时，HTTP客户端支持为每一个用户在请求间保存session。

   client=None

HttpSession实例在Locust初始化时被创建。client支持cookies，同时在请求间会保存session。

当从HttpLocust类继承时，我们可以使用client属性来对服务器生成HTTP请求。下面是一个locust文件示例用于在一个网站的两个URL / 和 /about/ 。

from locust import HttpLocust, TaskSet, task

class MyTaskSet(TaskSet):
    @task(2)
    def index(self):
        self.client.get('/')

    @task(1)
    def about(self):
        self.client.get('/about/')
class MyLocust(HttpLocust):
    task_set = MyTaskSet
    min_wait = 5000
    max_wait = 15000

使用上面的Locust类，每一个模拟用户将间隔5-15秒内请求，并且/将会比/about/请求数量多2倍。

细心的读者会发现有一些奇怪，我们使用self.client关联HttpSession实例，而不是TaskSet,也不是self.locust.client。我们可以这样做，是因为TaskSet类有一个属性调用client简单的返回self.locust.client。

使用HTTP client

每一个HttpLocust实例在client属性中有一个HttpSession实例。HttpSession类实际上是requests.Session的子类，可使用get post put delete head patch 和 options方法来生成HTTP请求，用于Locust的数据统计。HttpSession实例在请求间维护cookies，因此可用于登录网站并保存session在请求之间。client可以通过Locust实例的TaskSet实例来关联，因此很容易获取client并在任务中生成HTTP请求。

下面是一个生成GET请求到 /about 路径的示例(在这里，我们可以假设 self 是一个TaskSet 或 HttpLocust 类的实例):

response = self.client.get("/about")
print "Response staus code:", response.status_code
print "Response content:", response.content

下面是一个生成POST请求的示例:

response = self.client.post("/login", {"username": "testuser", "password": "password"})

安全模式

HTTP client被配制运行在safe_mode。这样做是任何请求在连接超时、错误、相似失败时将不会抛出异常，而是返回一个空的假Response对象。请求将会在Locust统计中算做一次失败。返回假Response内容属性将会被设置为None，并且它的status_code将会是0.

手动设置请求是成功或失败

默认情况下，请求被标记为失败除非在返回状态码是OK(2XX)。大部分时间内，这个默认就是你所需要的。然而，比如在测试一个URL节点，你期待返回状态码为404，或者测试一个即使错误发生也会返回200的系统，因此，需要手工控制locust来判断是成功还是失败。

一个可以生成失败请求，即使当响应代码是OK，通过使用catch_response参数和with语法:

with client.get("/", catch_response = True) as response:
    if response.content != "Success":
        response.failure("Got wrong response")

就像一个可以使用响应为OK的请求当做失败来处理，一个方法就是可以使用catch_response参数和with语法来让请求HTTP错误时，仍然统计数据为成功:

with client.get("/does_not_exist/", catch_response = True) as response:
    if response.status_code = 404:
        response.success()

使用动态参数来分组URL请求

针对网站，有一个常用的功能是获取URL中包含一些动态参数的页面数据。通常情况下，在Locust统计中，使用动态分组在URL中是很有意义的。通过name参数来给HttpSession传递不同的请求方法。

比如:

# Statistics for these requests will be grouped under: /blog/?id=[id]
for i in range(10):
  client.get("/blog?id=%i" % i, name = "/blog?id=[id]")

常用库

通常，大家想分享多个locust文件用于分享常用的库。在这种情况下，定义项目根目录用于调用Locsut是很重要的，建议将所有的locust文件有些话在项目的根目录中。

一个平铺的结构像下面这样:

• 项目根目录
  • commonlib_conf.py
  • commonlib_auth.py
  • locustfile_web_app.py
  • locsutfile_api.py
  • locustfile_ecommerce.py

locust文件可以调用常用的库通过使用import commonlib_auth.然而，这种方法不会从locust文件中，清晰分辨出常用库。

子文件夹可以有一个清晰的方法(查看下面的示例)，但是locust仅会有运行locsut文件的位置引用相关的模块。如果你想从你的根目录导入(如，你运行locust命令的位置)，确保在任何locust文件中添加常用库前有代码sys.path.append(os.getcwd())，会生成导入根目录(如，当前工作目录)。

• project root
  • __init__.py
  • common/
    • __init__.py
    • config.py
    • auth.py
  • locustfiles/
    • __init__.py
    • web_app.py
    • api.py
    • ecommerce.py

使用上面的项目结构，你的locust文件可以通过下面代码导入常用的库:

sys.path.append(os.getcwd())
import common.auth

• 本文Locust版本0.7.5
• 原文地址：http://docs.locust.io/en/latest/writing-a-locustfile.html