Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Allow selecting user classes using LOCUST_USER_CLASSES env var #2355

Merged
merged 2 commits into from
Jun 7, 2023
Merged

Allow selecting user classes using LOCUST_USER_CLASSES env var #2355

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
81f1973
Parse user_classes from LOCUST_USER_CLASSES env
ibarbech Jun 2, 2023
aadd93f
Create new test to test parser from env: test_parse_options_from_env
Jun 5, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion locust/argument_parser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -583,7 +583,8 @@ def setup_parser_arguments(parser):
"user_classes",
nargs="*",
metavar="UserClass",
help="Optionally specify which User classes that should be used (available User classes can be listed with -l or --list)",
help="Optionally specify which User classes that should be used (available User classes can be listed with -l or --list). LOCUST_USER_CLASSES environment variable can also be used to specify User classes",
default=os.environ.get("LOCUST_USER_CLASSES", "").split(),
)


Expand Down
28 changes: 28 additions & 0 deletions locust/test/test_parser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,34 @@ def test_parse_options(self):
# check default arg
self.assertEqual(8089, options.web_port)

def test_parse_options_from_env(self):
os.environ["LOCUST_LOCUSTFILE"] = "locustfile.py"
os.environ["LOCUST_USERS"] = "100"
os.environ["LOCUST_SPAWN_RATE"] = "10"
os.environ["LOCUST_RUN_TIME"] = "5m"
os.environ["LOCUST_RESET_STATS"] = "true"
os.environ["LOCUST_STOP_TIMEOUT"] = "5"
os.environ["LOCUST_USER_CLASSES"] = "MyUserClass"
options = parse_options(args=[])

self.assertEqual("locustfile.py", options.locustfile)
self.assertEqual(100, options.num_users)
self.assertEqual(10, options.spawn_rate)
self.assertEqual("5m", options.run_time)
self.assertTrue(options.reset_stats)
self.assertEqual("5", options.stop_timeout)
self.assertEqual(["MyUserClass"], options.user_classes)
# check default arg
self.assertEqual(8089, options.web_port)

del os.environ["LOCUST_LOCUSTFILE"]
del os.environ["LOCUST_USERS"]
del os.environ["LOCUST_SPAWN_RATE"]
del os.environ["LOCUST_RUN_TIME"]
del os.environ["LOCUST_RESET_STATS"]
del os.environ["LOCUST_STOP_TIMEOUT"]
del os.environ["LOCUST_USER_CLASSES"]

def test_parse_locustfile(self):
with mock_locustfile() as mocked:
locustfiles = parse_locustfile_option(
Expand Down