Reduce the number of iterations to ten in rally scenarios
[functest.git] / testcases / config_functest.py
1 #!/usr/bin/env python
2 #
3 # Copyright (c) 2015 Ericsson
4 # jose.lausuch@ericsson.com
5 # All rights reserved. This program and the accompanying materials
6 # are made available under the terms of the Apache License, Version 2.0
7 # which accompanies this distribution, and is available at
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10
11 import re, json, os, urllib2, argparse, logging, shutil, subprocess, yaml, sys, getpass
12 import functest_utils
13 from git import Repo
14 from os import stat
15 from pwd import getpwuid
16 from neutronclient.v2_0 import client as neutronclient
17
18 actions = ['start', 'check', 'clean']
19 parser = argparse.ArgumentParser()
20 parser.add_argument("action", help="Possible actions are: '{d[0]}|{d[1]}|{d[2]}' ".format(d=actions))
21 parser.add_argument("-d", "--debug", help="Debug mode",  action="store_true")
22 parser.add_argument("-f", "--force", help="Force",  action="store_true")
23 args = parser.parse_args()
24
25
26 """ logging configuration """
27 logger = logging.getLogger('config_functest')
28 logger.setLevel(logging.DEBUG)
29
30 ch = logging.StreamHandler()
31 if args.debug:
32     ch.setLevel(logging.DEBUG)
33 else:
34     ch.setLevel(logging.INFO)
35
36 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
37 ch.setFormatter(formatter)
38 logger.addHandler(ch)
39
40 REPOS_DIR=os.environ['repos_dir']
41 FUNCTEST_REPO=REPOS_DIR+'/functest/'
42 if not os.path.exists(FUNCTEST_REPO):
43     logger.error("Functest repository directory not found '%s'" % FUNCTEST_REPO)
44     exit(-1)
45 sys.path.append(FUNCTEST_REPO + "testcases/")
46
47 with open("/home/opnfv/functest/conf/config_functest.yaml") as f:
48     functest_yaml = yaml.safe_load(f)
49 f.close()
50
51
52 """ global variables """
53 # Directories
54 RALLY_DIR = FUNCTEST_REPO + functest_yaml.get("general").get("directories").get("dir_rally")
55 RALLY_REPO_DIR = functest_yaml.get("general").get("directories").get("dir_repo_rally")
56 RALLY_INSTALLATION_DIR = functest_yaml.get("general").get("directories").get("dir_rally_inst")
57 RALLY_RESULT_DIR = functest_yaml.get("general").get("directories").get("dir_rally_res")
58 VPING_DIR = FUNCTEST_REPO + functest_yaml.get("general").get("directories").get("dir_vping")
59 VIMS_TEST_DIR = functest_yaml.get("general").get("directories").get("dir_repo_vims_test")
60 ODL_DIR = FUNCTEST_REPO + functest_yaml.get("general").get("directories").get("dir_odl")
61 DATA_DIR = functest_yaml.get("general").get("directories").get("dir_functest_data")
62
63 # Tempest/Rally configuration details
64 DEPLOYMENT_MAME = functest_yaml.get("rally").get("deployment_name")
65 RALLY_COMMIT = functest_yaml.get("general").get("repositories").get("rally_commit")
66
67 #Image (cirros)
68 IMAGE_FILE_NAME = functest_yaml.get("general").get("openstack").get("image_file_name")
69 IMAGE_PATH = DATA_DIR + "/" + IMAGE_FILE_NAME
70
71 # NEUTRON Private Network parameters
72 NEUTRON_PRIVATE_NET_NAME = functest_yaml.get("general"). \
73     get("openstack").get("neutron_private_net_name")
74 NEUTRON_PRIVATE_SUBNET_NAME = functest_yaml.get("general"). \
75     get("openstack").get("neutron_private_subnet_name")
76 NEUTRON_PRIVATE_SUBNET_CIDR = functest_yaml.get("general"). \
77     get("openstack").get("neutron_private_subnet_cidr")
78 NEUTRON_ROUTER_NAME = functest_yaml.get("general"). \
79     get("openstack").get("neutron_router_name")
80
81 creds_neutron = functest_utils.get_credentials("neutron")
82 neutron_client = neutronclient.Client(**creds_neutron)
83
84 def action_start():
85     """
86     Start the functest environment installation
87     """
88     if not functest_utils.check_internet_connectivity():
89         logger.error("There is no Internet connectivity. Please check the network configuration.")
90         exit(-1)
91
92     if action_check():
93         logger.info("Functest environment already installed. Nothing to do.")
94         exit(0)
95
96     else:
97         # Clean in case there are left overs
98         logger.debug("Cleaning possible functest environment leftovers.")
99         action_clean()
100         logger.info("Starting installation of functest environment")
101
102         private_net = functest_utils.get_private_net(neutron_client)
103         if private_net is None:
104             # If there is no private network in the deployment we create one
105             if not create_private_neutron_net(neutron_client):
106                 logger.error("There has been a problem while creating the functest network.")
107                 action_clean()
108                 exit(-1)
109         else:
110             logger.info("Private network '%s' already existing in the deployment."
111                  % private_net['name'])
112
113         logger.info("Installing Rally...")
114         if not install_rally():
115             logger.error("There has been a problem while installing Rally.")
116             action_clean()
117             exit(-1)
118
119         logger.info("Installing Ruby libraries for vIMS testcase...")
120         # Install ruby libraries for vims test-case
121         script = 'source /etc/profile.d/rvm.sh; '
122         script += 'cd ' + VIMS_TEST_DIR + '; '
123         script += 'rvm autolibs enable ;'
124         script += 'rvm install 1.9.3; '
125         script += 'rvm use 1.9.3;'
126         script += 'bundle install'
127
128         logger_debug = None
129         CI_DEBUG = os.environ.get("CI_DEBUG")
130         if CI_DEBUG == "true" or CI_DEBUG == "True":
131             logger_debug = logger
132
133         cmd = "/bin/bash -c '" + script + "'"
134         functest_utils.execute_command(cmd, logger = logger_debug, exit_on_error=False)
135
136         install_promise(logger_debug)
137
138         # Create result folder under functest if necessary
139         if not os.path.exists(RALLY_RESULT_DIR):
140             os.makedirs(RALLY_RESULT_DIR)
141
142         try:
143             logger.info("CI: Generate the list of executable tests.")
144             runnable_test = functest_utils.generateTestcaseList(functest_yaml)
145             logger.info("List of runnable tests generated: %s" % runnable_test)
146         except:
147             logger.error("Impossible to generate the list of runnable tests")
148
149         exit(0)
150
151
152 def action_check():
153     """
154     Check if the functest environment is properly installed
155     """
156     errors_all = False
157     errors = False
158     logger.info("Checking current functest configuration...")
159
160     logger.debug("Checking script directories...")
161
162     dirs = [RALLY_DIR, RALLY_INSTALLATION_DIR, VPING_DIR, ODL_DIR]
163     for dir in dirs:
164         if not os.path.exists(dir):
165             logger.debug("   %s NOT found" % dir)
166             errors = True
167             errors_all = True
168         else:
169             logger.debug("   %s found" % dir)
170
171     logger.debug("Checking Rally deployment...")
172     if not check_rally():
173         logger.debug("   Rally deployment NOT installed.")
174         errors_all = True
175
176     logger.debug("Checking Image...")
177     errors = False
178     if not os.path.isfile(IMAGE_PATH):
179         logger.debug("   Image file '%s' NOT found." % IMAGE_PATH)
180         errors = True
181         errors_all = True
182     else:
183         logger.debug("   Image file found in %s" % IMAGE_PATH)
184
185
186     #TODO: check OLD environment setup
187     return not errors_all
188
189
190
191 def action_clean():
192     """
193     Clean the existing functest environment
194     """
195     logger.info("Removing current functest environment...")
196     if os.path.exists(RALLY_INSTALLATION_DIR):
197         logger.debug("Removing Rally installation directory %s" % RALLY_INSTALLATION_DIR)
198         shutil.rmtree(RALLY_INSTALLATION_DIR,ignore_errors=True)
199
200     if os.path.exists(RALLY_RESULT_DIR):
201         logger.debug("Removing Result directory")
202         shutil.rmtree(RALLY_RESULT_DIR,ignore_errors=True)
203
204     logger.info("Functest environment clean!")
205
206
207
208 def install_rally():
209     if check_rally():
210         logger.info("Rally is already installed.")
211     else:
212         logger.debug("Executing %s/install_rally.sh..." %RALLY_REPO_DIR)
213         install_script = RALLY_REPO_DIR + "/install_rally.sh --yes"
214         cmd = 'sudo ' + install_script
215         if os.environ.get("CI_DEBUG") == "false":
216             functest_utils.execute_command(cmd)
217         else:
218             functest_utils.execute_command(cmd,logger)
219
220         logger.debug("Creating Rally environment...")
221         cmd = "rally deployment create --fromenv --name="+DEPLOYMENT_MAME
222         functest_utils.execute_command(cmd,logger)
223
224         logger.debug("Installing tempest...")
225         cmd = "rally verify install"
226         functest_utils.execute_command(cmd,logger)
227
228         cmd = "rally deployment check"
229         functest_utils.execute_command(cmd,logger)
230         #TODO: check that everything is 'Available' and warn if not
231
232         cmd = "rally show images"
233         functest_utils.execute_command(cmd,logger)
234
235         cmd = "rally show flavors"
236         functest_utils.execute_command(cmd,logger)
237
238     return True
239
240 def install_promise(logger_debug):
241     logger.info("Installing dependencies for Promise testcase...")
242     current_dir = os.getcwd()
243     os.chdir(REPOS_DIR+'/promise/')
244
245     cmd = 'curl -sL https://deb.nodesource.com/setup_4.x | sudo -E bash -'
246     functest_utils.execute_command(cmd,logger = logger_debug, exit_on_error=False)
247
248     cmd = 'sudo apt-get install -y nodejs'
249     functest_utils.execute_command(cmd,logger = logger_debug, exit_on_error=False)
250
251     cmd = 'sudo npm -g install npm@latest'
252     functest_utils.execute_command(cmd,logger = logger_debug, exit_on_error=False)
253
254     cmd = 'npm install'
255     functest_utils.execute_command(cmd,logger = logger_debug, exit_on_error=False)
256     os.chdir(current_dir)
257
258
259 def check_rally():
260     """
261     Check if Rally is installed and properly configured
262     """
263     if os.path.exists(RALLY_INSTALLATION_DIR):
264         logger.debug("   Rally installation directory found in %s" % RALLY_INSTALLATION_DIR)
265         FNULL = open(os.devnull, 'w');
266         cmd="rally deployment list | grep "+DEPLOYMENT_MAME
267         logger.debug('   Executing command : {}'.format(cmd))
268         p=subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=FNULL);
269         #if the command does not exist or there is no deployment
270         line = p.stdout.readline()
271         if line == "":
272             logger.debug("   Rally deployment NOT found")
273             return False
274         logger.debug("   Rally deployment found")
275         return True
276     else:
277         return False
278
279
280 def create_private_neutron_net(neutron):
281     neutron.format = 'json'
282     logger.info("Creating network '%s'..." % NEUTRON_PRIVATE_NET_NAME)
283     network_id = functest_utils. \
284         create_neutron_net(neutron, NEUTRON_PRIVATE_NET_NAME)
285
286     if not network_id:
287         return False
288     logger.debug("Network '%s' created successfully." % network_id)
289
290     logger.info("Updating network '%s' with shared=True..." % NEUTRON_PRIVATE_NET_NAME)
291     if functest_utils.update_neutron_net(neutron, network_id, shared=True):
292         logger.debug("Network '%s' updated successfully." % network_id)
293     else:
294         logger.info("Updating neutron network '%s' failed" % network_id)
295
296     logger.info("Creating Subnet....")
297     subnet_id = functest_utils. \
298         create_neutron_subnet(neutron,
299                               NEUTRON_PRIVATE_SUBNET_NAME,
300                               NEUTRON_PRIVATE_SUBNET_CIDR,
301                               network_id)
302     if not subnet_id:
303         return False
304     logger.debug("Subnet '%s' created successfully." % subnet_id)
305     logger.info("Creating Router...")
306     router_id = functest_utils. \
307         create_neutron_router(neutron, NEUTRON_ROUTER_NAME)
308
309     if not router_id:
310         return False
311
312     logger.debug("Router '%s' created successfully." % router_id)
313     logger.info("Adding router to subnet...")
314
315     result = functest_utils.add_interface_router(neutron, router_id, subnet_id)
316
317     if not result:
318         return False
319
320     logger.debug("Interface added successfully.")
321     network_dic = {'net_id': network_id,
322                    'subnet_id': subnet_id,
323                    'router_id': router_id}
324     return True
325
326
327 def main():
328     if not (args.action in actions):
329         logger.error('argument not valid')
330         exit(-1)
331
332
333     if not functest_utils.check_credentials():
334         logger.error("Please source the openrc credentials and run the script again.")
335         #TODO: source the credentials in this script
336         exit(-1)
337
338
339     if args.action == "start":
340         action_start()
341
342     if args.action == "check":
343         if action_check():
344             logger.info("Functest environment correctly installed")
345         else:
346             logger.info("Functest environment not found or faulty")
347
348     if args.action == "clean":
349         if args.force :
350             action_clean()
351         else :
352             while True:
353                 print("Are you sure? [y|n]")
354                 answer = raw_input("")
355                 if answer == "y":
356                     action_clean()
357                     break
358                 elif answer == "n":
359                     break
360                 else:
361                     print("Invalid option.")
362     exit(0)
363
364
365 if __name__ == '__main__':
366     main()
367