You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
818 B
25 lines
818 B
import random
|
|
from locust import HttpUser, task, between
|
|
|
|
class APIUser(HttpUser):
|
|
# 设置模拟用户之间的等待时间（例如：每个用户之间随机等待 1-3 秒）
|
|
wait_time = between(1, 3)
|
|
|
|
# 定义一个包含多个章节 ID 的列表
|
|
chapter_ids = ["1", "3", "6", "8", "15"]
|
|
|
|
@task
|
|
def get_chapter_relations(self):
|
|
# 从章节 ID 列表中随机选择一个章节 ID
|
|
chapter_id = random.choice(self.chapter_ids)
|
|
self.client.get(f"/chapters/{chapter_id}/relations")
|
|
|
|
@task
|
|
def get_all_relations(self):
|
|
self.client.get("/chapters/relations/")
|
|
|
|
@task
|
|
def get_relations_by_level(self):
|
|
level = random.randint(1, 5) # 随机选择一个层级，范围为 1 到 5
|
|
self.client.get(f"/chapters/relations/level/{level}")
|