Locust简单例子

Locust实例
以下是一个简单的locustfile.py的一个简单例子

from locust import HttpLocust, TaskSet

def login(l):
    l.client.post("/login", {"username":"ellen_key", "password":"education"})

def index(l):
    l.client.get("/")

def profile(l):
    l.client.get("/profile")

class UserBehavior(TaskSet):
    tasks = {index: 2, profile: 1}

    def on_start(self):
        login(self)

class WebsiteUser(HttpLocust):
    task_set = UserBehavior
    min_wait = 5000
    max_wait = 9000

这里定义了一个Locust任务,里面定义了两个类,一个UserBehavior类,用于定义测试业务,一个WebsiteUser类,用于定义模拟用户,我们定义一个模拟用户在执行任务之间应该等待多长时间

另外一种简单的方法,用@task 构造器。上面是随机执行,下面这个是顺序执行

from locust import HttpLocust, TaskSet, task

class UserBehavior(TaskSet):
    def on_start(self):
        """ on_start is called when a Locust start before any task is scheduled """
        self.login()

    def login(self):
        self.client.post("/login", {"username":"ellen_key", "password":"education"})

    @task(2)
    def index(self):
        self.client.get("/")

    @task(1)
    def profile(self):
        self.client.get("/profile")

class WebsiteUser(HttpLocust):
    task_set = UserBehavior
    min_wait = 5000
    max_wait = 9000

启动Locust

如果locustfile.py位于当前工作目录中,我们可以运行:

locust --host=http://example.com

如果Locust文件位于与locustfile.py不同的子目录下,请使用-f以下命令指定:

locust -f locust_files/my_locust_file.py --host=http://example.com

分布式启动Locust

主进程:

locust -f locust_files/my_locust_file.py --master --host=http://example.com

子进程

locust -f locust_files/my_locust_file.py --slave --host=http://example.com

如果要在多台机器上运行Locust,我们还需要在启动从机时指定主机(在主机默认为127.0.0.1时,在单台机器上运行Locust时不需要):

locust -f locust_files/my_locust_file.py --slave --master-host=192.168.0.100 --host=http://example.com

打开Locust的Web界面
您使用上述命令行启动了蝗虫,您可以打开浏览器并输入网址http://127.0.0.1:8089(如果您在本地运行Locust),如下界面:
这里写图片描述

第一行Number of users to simulate是模拟用户的数量(虚拟用户数),第二行Hatch rate (users spawned/second)表示产生模拟用户的速度,填写完成后点击“Start swarming”即可开始测试

这里写图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章