1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#!/usr/bin/env python3
import sys, subprocess, os
from datetime import datetime
def usage():
usage = """
This script only adds defaults if there are no options
Usage: run-load-test [options]
Defaults:
-H [--host] http://localhost:8080
-u [--users] 1
-s [--spawn_rate] 1
-t [--run_time] 1m
-f [--locustfile] locust/load-test/load-test.py
-T [--tags] example
-c [--csv] <current_date_time>
--html <current_data_time>.html
"""
print(usage)
def add_default_args(program_args):
defaults = [['--headless'], ['-H', 'http://localhost:8080'],
["-u", '1'], ["-r", '1'], ["-t", '1m'], ['-f', locust_filepath],
['--html', f'{timestamp}.html'], ['--tags', 'example'], ['--csv', timestamp]]
additional_args = []
for default_arg in defaults:
found = False
for arg in program_args:
if arg == default_arg[0]:
found = True
if not found:
additional_args += default_arg
return additional_args + program_args
subprocess.run([sys.executable, "-m", "pip", "install", "locust", "pyyaml", "-U"])
missing_defaults = []
dir_path = os.path.dirname(os.path.realpath(__file__))
workdir = os.getcwd()
locust_filepath = os.path.join(dir_path, 'load-test', 'load-test.py')
timestamp = datetime.now().strftime("%d-%m-%YT%H:%M:%S")
sys_args = sys.argv[1:]
if '--help' in sys_args:
subprocess.run(["locust", "--help"])
usage()
exit()
locust_args = add_default_args(sys_args)
subprocess.run(["locust"] + locust_args)
|
