一、@task 装饰器
- 为用户添加任务的最简单方法是使用task装饰器
from locust import User, task, constant
class OneTask(User):
wait_time = constant(1)
@task
def task1(self):
print("first task:" % self)
-
@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/")