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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
import argparse
import json
import os
import shutil
import site
import sys
import tempfile
from subprocess import check_call
from molotov import __version__
from molotov.run import _parser
from molotov.run import main as run
def clone_repo(github):
# XXX security
check_call("git clone %s ." % github, shell=True)
def create_virtualenv(virtualenv, python):
# XXX security
cmd = "%s -m venv venv" % python
check_call(cmd, shell=True)
def install_reqs(reqfile):
check_call("./venv/bin/pip install --upgrade pip", shell=True)
check_call("./venv/bin/pip install -r %s" % reqfile, shell=True)
def run_test(**options):
"""Runs a molotov test."""
parser = _parser()
fields = {}
cli = []
for action in parser._actions:
if action.dest in ("help", "scenario"):
continue
op_str = action.option_strings[0]
fields[action.dest] = op_str, action.const, type(action)
for key, value in options.items():
if key in fields:
opt, const, type_ = fields[key]
is_count = type_ is argparse._CountAction
if const or is_count:
if is_count:
cli += [opt] * value
else:
cli.append(opt)
else:
cli.append(opt)
cli.append(str(value))
cli.append(options.pop("scenario", "loadtest.py"))
args = parser.parse_args(args=cli)
print("Running: molotov %s" % " ".join(cli))
return run(args)
def main():
"""Moloslave clones a git repo and runs a molotov test"""
parser = argparse.ArgumentParser(description="Github-based load test")
parser.add_argument(
"--version",
action="store_true",
default=False,
help="Displays version and exits.",
)
parser.add_argument(
"--virtualenv", type=str, default="virtualenv", help="Virtualenv executable."
)
parser.add_argument(
"--python", type=str, default=sys.executable, help="Python executable."
)
parser.add_argument(
"--directory", type=str, default=None, help="Directory to run into."
)
parser.add_argument(
"--config",
type=str,
default="molotov.json",
help="Path of the configuration file.",
)
parser.add_argument("repo", help="Github repo", type=str, nargs="?")
parser.add_argument("run", help="Test to run", nargs="?")
args = parser.parse_args()
if args.version:
print(__version__)
sys.exit(0)
if args.directory is None:
args.directory = tempfile.mkdtemp()
remove_dir = True
else:
remove_dir = False
curdir = os.getcwd()
os.chdir(args.directory)
print("Working directory is %s" % args.directory)
try:
clone_repo(args.repo)
config_file = os.path.join(args.directory, args.config)
with open(config_file) as f:
config = json.loads(f.read())
# creating the virtualenv
create_virtualenv(args.virtualenv, args.python)
# install deps
if "requirements" in config["molotov"]:
install_reqs(config["molotov"]["requirements"])
# load deps into sys.path
pyver = "%d.%d" % (sys.version_info.major, sys.version_info.minor)
site_pkg = os.path.join(
args.directory, "venv", "lib", "python" + pyver, "site-packages"
)
site.addsitedir(site_pkg)
# environment
if "env" in config["molotov"]:
for key, value in config["molotov"]["env"].items():
os.environ[key] = value
run_test(**config["molotov"]["tests"][args.run])
except Exception:
os.chdir(curdir)
if remove_dir:
shutil.rmtree(args.directory, ignore_errors=True)
raise
|
