-
Notifications
You must be signed in to change notification settings - Fork 967
/
scenarios.py
executable file
·160 lines (134 loc) · 5.5 KB
/
scenarios.py
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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
#!/usr/bin/python
"""scenarios management tool
Usage:
scenarios.py <command> [<args>...]
scenarios.py start [<scenario>]
scenarios.py build_image [<image>]
The most commonly used scenarios.py commands are:
list List the available test scenarios
proxysql_versions List the available ProxySQL versions to be tested
mysql_versions List the available MySQL versions to be paired with ProxySQL for testing
start Start a ProxySQL testing scenario
ps List the running ProxySQL scenarios
"""
import os
import subprocess
import sys
from docopt import docopt
import nose
from test.docker_fleet import DockerFleet
PROXYSQL_SCENARIO_FILE = '/tmp/proxysql-scenario.txt'
def scenarios_list():
templates = DockerFleet().get_docker_scenario_templates()
scenario_names = sorted(templates.keys())
for scenario in scenario_names:
print('%s @ %s' % (scenario, os.path.abspath(templates[scenario]['dir'])))
def proxysql_images():
dockerfiles = DockerFleet().get_dockerfiles_for_proxysql()
images = sorted(dockerfiles.keys())
for image in images:
print('%s @ %s' % (image, os.path.abspath(dockerfiles[image]['dir'])))
def mysql_images():
dockerfiles = DockerFleet().get_dockerfiles_for_mysql()
images = sorted(dockerfiles.keys())
for image in images:
print('%s @ %s' % (image, os.path.abspath(dockerfiles[image]['dir'])))
def start(scenario, proxysql_image, mysql_image):
docker_fleet = DockerFleet()
scenario_info = docker_fleet.generate_scenarios(
scenarios=[scenario],
proxysql_filters={'names': [proxysql_image]},
mysql_filters={'names': [mysql_image]}
)[0]
dirname = docker_fleet.start_temp_scenario(scenario_info, copy_folder=True)
with open(PROXYSQL_SCENARIO_FILE, 'wt') as f:
f.write(dirname)
return dirname
def stop():
with open(PROXYSQL_SCENARIO_FILE, 'rt') as f:
dirname = ''.join(f.readlines()).strip()
DockerFleet().stop_temp_scenario(dirname, delete_folder=True)
try:
os.remove(PROXYSQL_SCENARIO_FILE)
except:
pass
def _build_image(image, dir):
subprocess.call(["docker", "rmi", "-f", "proxysql:%s" % image])
subprocess.call(["docker", "build", "-t", "proxysql:%s" % image, "."],
cwd=dir)
def build_image(image):
if image != 'all':
# Builds a docker image (either for ProxySQL or for MySQL) and commits it
# to the renecannao dockerhub repository.
dockerfiles = DockerFleet().get_dockerfiles_for_proxysql()
if image in dockerfiles:
_build_image(image, dockerfiles[image]['dir'])
return
dockerfiles = DockerFleet().get_dockerfiles_for_mysql()
if image in dockerfiles:
_build_image(image, dockerfiles[image]['dir'])
return
print("Image %s wasn't found in either ProxySQL or MySQL image list.\n"
"Please double-check the name!" % image)
else:
dockerfiles = DockerFleet().get_dockerfiles_for_proxysql()
for image in dockerfiles.iterkeys():
_build_image(image, dockerfiles[image]['dir'])
dockerfiles = DockerFleet().get_dockerfiles_for_mysql()
for image in dockerfiles.iterkeys():
_build_image(image, dockerfiles[image]['dir'])
def test(argv):
nose.run(argv=argv)
if __name__ == '__main__':
args = docopt(__doc__,
version='scenarios.py version 0.0.1',
options_first=True)
argv = [args['<command>']] + args['<args>']
if args['<command>'] == 'list':
scenarios_list()
elif args['<command>'] == 'proxysql_images':
proxysql_images()
elif args['<command>'] == 'mysql_images':
mysql_images()
elif args['<command>'] == 'start':
if len(args['<args>']) >= 1:
scenario = args['<args>'][0]
else:
# Default value for the scenario parameter: '1backend'
scenario = '1backend'
if len(args['<args>']) >= 2:
proxysql_image = args['<args>'][1].split('=')[1]
else:
# Default value for the proxysql_image parameter: 'proxysql'
proxysql_image = 'proxysql'
if len(args['<args>']) >= 3:
mysql_image = args['<args>'][2].split('=')[1]
else:
# Default value for the mysql_image parameter: 'mysql-simple-dump'
mysql_image = 'mysql-simple-dump'
if (os.path.exists(PROXYSQL_SCENARIO_FILE)):
print("Is there another scenario running? If not, delete %s" %
PROXYSQL_SCENARIO_FILE)
else:
dirname = start(scenario, proxysql_image, mysql_image)
print("Scenario started successfully at path %s" % dirname)
elif args['<command>'] == 'stop':
if (not os.path.exists(PROXYSQL_SCENARIO_FILE)):
print("There is no scenario running or file %s has been removed" %
PROXYSQL_SCENARIO_FILE)
else:
stop()
print("Scenario stopped successfully")
elif args['<command>'] == 'build_image':
if len(args['<args>']) > 0:
build_image(args['<args>'][0])
else:
build_image('all')
elif args['<command>'] == 'test':
if (not os.path.exists(PROXYSQL_SCENARIO_FILE)):
print("There doesn't seem to be a running scenario. Aborting.")
else:
if len(args['<args>']) > 0:
test(args['<args>'])
else:
test(['.'])