locust - 如何将自定义参数传递给蝗虫测试类？

Question

我目前正在使用环境变量将自定义参数传递给我的负载测试。例如，我的测试类如下所示:

from locust import HttpLocust, TaskSet, task
import os

class UserBehavior(TaskSet):

    @task(1)
    def login(self):
        test_dir = os.environ['BASE_DIR']
        auth=tuple(open(test_dir + '/PASSWORD).read().rstrip().split(':')) 
        self.client.request(
           'GET',
           '/myendpoint',
           auth=auth
        )   

class WebsiteUser(HttpLocust):
    task_set = UserBehavior

然后我用以下方法运行测试:

locust -H https://myserver --no-web --clients=500 --hatch-rate=500 --num-request=15000 --print-stats --only-summary

我可以通过自定义参数将更多参数传递给 locust命令行应用程序吗？

Answer 1

您可以使用类似env <parameter>=<value> locust <options>的方法，并在蝗虫脚本中使用<parameter>的值

例如。，env IP_ADDRESS=100.0.1.1 locust -f locust-file.py --no-web --clients=5 --hatch-rate=1 --num-request=500并在蝗虫脚本中使用IP_ADDRESS来访问其值(在这种情况下为100.0.1.1)。