时间戳

首页 > 解决方案 > How would I pass variables to a `@events.test_start.add_listener` function

python - How would I pass variables to a `@events.test_start.add_listener` function

问题描述

I'm trying to run load tests with locust but I need to run some API calls before I start spawning workers. I already pass my user credentials as environment variables but I don't want to have to hard-code the host within the locustfile. I'm ideally looking to have something that looks like

@events.test_start.add_listener
def on_test_start(**kwargs):
    init(host, username, password)

Where init will take care of all of the initial API calls.

标签： pythonlocust

解决方案

The first parameter, environment, to the init function contains locust settings in the parsed_options dict. Try getting host from there.

Something like (untested)

@events.init.add_listener
def on_locust_init(environment, **kwargs):
    init(environment.parsed_options[”host”], ...)

推荐阅读