locust官方文档-debugging_advanced.py

最新推荐文章于 2024-01-19 12:51:14 发布

原创最新推荐文章于 2024-01-19 12:51:14 发布 · 219 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

locust 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了如何使用Locust库创建两个用户类(User1和User2)，分别执行GET请求并触发StopUser异常来停止任务执行。通过实例展示了如何在主函数中运行这两个用户类。这是关于 Locust调试实践的基础示例。

from locust import HttpUser, task, run_single_user
from locust.exception import StopUser


class User1(HttpUser):
    host = "http://localhost"

    @task
    def hello_world(self):
        with self.client.get("/hello1", catch_response=True) as resp:
            pass
        raise StopUser()


class User2(HttpUser):
    host = "http://localhost"

    @task
    def hello_world(self):
        with self.client.get("/hello2", catch_response=True) as resp:
            pass
        raise StopUser()


if __name__ == "__main__":
    print("running User1")
    run_single_user(User1)
    print("running User2")
    run_single_user(User2)
    print("done!")