是否可以添加自定义蝗虫命令行参数

Question

我想知道是否有办法将自定义命令行/配置参数传递给 locust。我目前正在将我的用户凭据作为环境变量传递到我的测试中，但我想通过命令行将它们传递给如下所示的内容：

locust -f <locustfile> --host <host> --username <username> --password  <password>

理想情况下，我想避免创建自己的自定义蝗虫扩展，但我猜我必须这样做。我很乐意听到任何建议。

Answer 1

是的！ github上有一个例子：https://github.com/locustio/locust/blob/master/examples/add_command_line_argument.py

@events.init_command_line_parser.add_listener
def init_parser(parser):
    parser.add_argument(
        '--my-argument',
        help="It's working"
    )

class WebsiteUser(HttpUser):
    @task
    def my_task(self):
        print(self.environment.parsed_options.my_argument)