性能测试---locust HttpUser 类和@task任务

一、@task 装饰器

1. 为用户添加任务的最简单方法是使用task装饰器

from locust import User, task, constant

class OneTask(User):
    wait_time = constant(1)

    @task
    def task1(self):
        print("first task:" % self)

2. @task接受一个可选的权重参数，可用于指定任务的执行率

   如下，task2被先选中执行的几率更大

from locust import User, task, constant

class OneTask(User):
    wait_time = constant(1)

    @task(2)
    def task1(self):
        print("first task:" % self)
    @task(5)
    def task2(self):
        print("two task:" % self)

二、HttpUser 类

HttpUser是最常用的User ,添加了一个client用于发出 HTTP 请求的属性

client是一个实例HttpSession ,像requests.Session，它在请求之间保留 cookie，可以保持登录状态

包含所有 HTTP 的方法：get, post, put，delete等，可直接调用即可

代码如下（示例）：

from locust import HttpUser, task, between

class OneTask(HttpUser):
    wait_time = between(5, 15)

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

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