首页 > 解决方案 > 蝗虫中的动态GET参数

问题描述

from locust import HttpLocust, TaskSet, task
import random

ids = [101, 102, 103, 104, 105, 106, 107, 108, 109, 110]


class TestAPITaskSet(TaskSet):

    @task(1)
    def test_get_id(self):
        id = random.choice(ids)
        self.client.get("/test-api/id/" + str(id), name="/id/[id]")


class TestAPILocust(HttpLocust):
    task_set = TestAPITaskSet
    min_wait = 0
    max_wait = 1000

如果我想从列表中随机选择 id,这是好还是有其他更好的蝗虫原生方法?

标签: pythonlocust

解决方案


没有更好的 locust-native 方法,你的代码看起来很合理!

如果您的列表开始变得很长并且您可以从列表中按顺序选择,您可以从 locust-plugins 中查看 CSVReader:https ://github.com/SvenskaSpel/locust-plugins/blob/master/示例/csvreader.py


推荐阅读