Enable tempest multinode tests
[functest.git] / functest / opnfv_tests / openstack / tempest / conf_utils.py
1 #!/usr/bin/env python
2 #
3 # Copyright (c) 2015 All rights reserved
4 # This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 #
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10 import ConfigParser
11 import logging
12 import os
13 import pkg_resources
14 import shutil
15 import subprocess
16
17 import yaml
18
19 from functest.utils.constants import CONST
20 import functest.utils.functest_utils as ft_utils
21 import functest.utils.openstack_utils as os_utils
22
23
24 IMAGE_ID_ALT = None
25 FLAVOR_ID_ALT = None
26 GLANCE_IMAGE_PATH = os.path.join(
27     CONST.__getattribute__('dir_functest_images'),
28     CONST.__getattribute__('openstack_image_file_name'))
29 TEMPEST_RESULTS_DIR = os.path.join(CONST.__getattribute__('dir_results'),
30                                    'tempest')
31 TEMPEST_CUSTOM = pkg_resources.resource_filename(
32     'functest', 'opnfv_tests/openstack/tempest/custom_tests/test_list.txt')
33 TEMPEST_BLACKLIST = pkg_resources.resource_filename(
34     'functest', 'opnfv_tests/openstack/tempest/custom_tests/blacklist.txt')
35 TEMPEST_DEFCORE = pkg_resources.resource_filename(
36     'functest',
37     'opnfv_tests/openstack/tempest/custom_tests/defcore_req.txt')
38 TEMPEST_RAW_LIST = os.path.join(TEMPEST_RESULTS_DIR, 'test_raw_list.txt')
39 TEMPEST_LIST = os.path.join(TEMPEST_RESULTS_DIR, 'test_list.txt')
40 REFSTACK_RESULTS_DIR = os.path.join(CONST.__getattribute__('dir_results'),
41                                     'refstack')
42 TEMPEST_CONF_YAML = pkg_resources.resource_filename(
43     'functest', 'opnfv_tests/openstack/tempest/custom_tests/tempest_conf.yaml')
44 TEST_ACCOUNTS_FILE = pkg_resources.resource_filename(
45     'functest',
46     'opnfv_tests/openstack/tempest/custom_tests/test_accounts.yaml')
47
48 CI_INSTALLER_TYPE = CONST.__getattribute__('INSTALLER_TYPE')
49 CI_INSTALLER_IP = CONST.__getattribute__('INSTALLER_IP')
50
51 """ logging configuration """
52 logger = logging.getLogger(__name__)
53
54
55 def get_verifier_id():
56     """
57     Returns verifier id for current Tempest
58     """
59     cmd = ("rally verify list-verifiers | awk '/" +
60            CONST.__getattribute__('tempest_deployment_name') +
61            "/ {print $2}'")
62     p = subprocess.Popen(cmd, shell=True,
63                          stdout=subprocess.PIPE,
64                          stderr=subprocess.STDOUT)
65     deployment_uuid = p.stdout.readline().rstrip()
66     if deployment_uuid == "":
67         logger.error("Tempest verifier not found.")
68         raise Exception('Error with command:%s' % cmd)
69     return deployment_uuid
70
71
72 def get_verifier_deployment_id():
73     """
74     Returns deployment id for active Rally deployment
75     """
76     cmd = ("rally deployment list | awk '/" +
77            CONST.__getattribute__('rally_deployment_name') +
78            "/ {print $2}'")
79     p = subprocess.Popen(cmd, shell=True,
80                          stdout=subprocess.PIPE,
81                          stderr=subprocess.STDOUT)
82     deployment_uuid = p.stdout.readline().rstrip()
83     if deployment_uuid == "":
84         logger.error("Rally deployment not found.")
85         raise Exception('Error with command:%s' % cmd)
86     return deployment_uuid
87
88
89 def get_verifier_repo_dir(verifier_id):
90     """
91     Returns installed verifier repo directory for Tempest
92     """
93     if not verifier_id:
94         verifier_id = get_verifier_id()
95
96     return os.path.join(CONST.__getattribute__('dir_rally_inst'),
97                         'verification',
98                         'verifier-{}'.format(verifier_id),
99                         'repo')
100
101
102 def get_verifier_deployment_dir(verifier_id, deployment_id):
103     """
104     Returns Rally deployment directory for current verifier
105     """
106     if not verifier_id:
107         verifier_id = get_verifier_id()
108
109     if not deployment_id:
110         deployment_id = get_verifier_deployment_id()
111
112     return os.path.join(CONST.__getattribute__('dir_rally_inst'),
113                         'verification',
114                         'verifier-{}'.format(verifier_id),
115                         'for-deployment-{}'.format(deployment_id))
116
117
118 def get_repo_tag(repo):
119     """
120     Returns last tag of current branch
121     """
122     cmd = ("git -C {0} describe --abbrev=0 HEAD".format(repo))
123     p = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
124     tag = p.stdout.readline().rstrip()
125
126     return str(tag)
127
128
129 def backup_tempest_config(conf_file):
130     """
131     Copy config file to tempest results directory
132     """
133     shutil.copyfile(conf_file,
134                     os.path.join(TEMPEST_RESULTS_DIR, 'tempest.conf'))
135
136
137 def configure_tempest(deployment_dir, image_id=None, flavor_id=None,
138                       compute_cnt=None):
139     """
140     Calls rally verify and updates the generated tempest.conf with
141     given parameters
142     """
143     conf_file = configure_verifier(deployment_dir)
144     configure_tempest_update_params(conf_file, image_id, flavor_id,
145                                     compute_cnt)
146
147
148 def configure_tempest_defcore(deployment_dir, image_id, flavor_id,
149                               image_id_alt, flavor_id_alt, tenant_id):
150     """
151     Add/update needed parameters into tempest.conf file
152     """
153     conf_file = configure_verifier(deployment_dir)
154     configure_tempest_update_params(conf_file, image_id, flavor_id)
155
156     logger.debug("Updating selected tempest.conf parameters for defcore...")
157     config = ConfigParser.RawConfigParser()
158     config.read(conf_file)
159     config.set('DEFAULT', 'log_file', '{}/tempest.log'.format(deployment_dir))
160     config.set('oslo_concurrency', 'lock_path',
161                '{}/lock_files'.format(deployment_dir))
162     generate_test_accounts_file(tenant_id=tenant_id)
163     config.set('auth', 'test_accounts_file', TEST_ACCOUNTS_FILE)
164     config.set('scenario', 'img_dir', '{}'.format(deployment_dir))
165     config.set('scenario', 'img_file', 'tempest-image')
166     config.set('compute', 'image_ref', image_id)
167     config.set('compute', 'image_ref_alt', image_id_alt)
168     config.set('compute', 'flavor_ref', flavor_id)
169     config.set('compute', 'flavor_ref_alt', flavor_id_alt)
170
171     with open(conf_file, 'wb') as config_file:
172         config.write(config_file)
173
174     confpath = pkg_resources.resource_filename(
175         'functest',
176         'opnfv_tests/openstack/refstack_client/refstack_tempest.conf')
177     shutil.copyfile(conf_file, confpath)
178
179
180 def generate_test_accounts_file(tenant_id):
181     """
182     Add needed tenant and user params into test_accounts.yaml
183     """
184
185     logger.debug("Add needed params into test_accounts.yaml...")
186     accounts_list = [
187         {
188             'tenant_name':
189                 CONST.__getattribute__('tempest_identity_tenant_name'),
190             'tenant_id': str(tenant_id),
191             'username': CONST.__getattribute__('tempest_identity_user_name'),
192             'password':
193                 CONST.__getattribute__('tempest_identity_user_password')
194         }
195     ]
196
197     with open(TEST_ACCOUNTS_FILE, "w") as f:
198         yaml.dump(accounts_list, f, default_flow_style=False)
199
200
201 def configure_tempest_update_params(tempest_conf_file, image_id=None,
202                                     flavor_id=None, compute_cnt=1):
203     """
204     Add/update needed parameters into tempest.conf file
205     """
206     logger.debug("Updating selected tempest.conf parameters...")
207     config = ConfigParser.RawConfigParser()
208     config.read(tempest_conf_file)
209     config.set(
210         'compute',
211         'fixed_network_name',
212         CONST.__getattribute__('tempest_private_net_name'))
213     config.set('compute', 'volume_device_name',
214                CONST.__getattribute__('tempest_volume_device_name'))
215     if CONST.__getattribute__('tempest_use_custom_images'):
216         if image_id is not None:
217             config.set('compute', 'image_ref', image_id)
218         if IMAGE_ID_ALT is not None:
219             config.set('compute', 'image_ref_alt', IMAGE_ID_ALT)
220     if CONST.__getattribute__('tempest_use_custom_flavors'):
221         if flavor_id is not None:
222             config.set('compute', 'flavor_ref', flavor_id)
223         if FLAVOR_ID_ALT is not None:
224             config.set('compute', 'flavor_ref_alt', FLAVOR_ID_ALT)
225     if compute_cnt > 1:
226         # enable multinode tests
227         config.set('compute', 'min_compute_nodes', compute_cnt)
228
229     config.set('identity', 'region', 'RegionOne')
230     if os_utils.is_keystone_v3():
231         auth_version = 'v3'
232     else:
233         auth_version = 'v2'
234     config.set('identity', 'auth_version', auth_version)
235     config.set(
236         'validation', 'ssh_timeout',
237         CONST.__getattribute__('tempest_validation_ssh_timeout'))
238     config.set('object-storage', 'operator_role',
239                CONST.__getattribute__('tempest_object_storage_operator_role'))
240
241     if CONST.__getattribute__('OS_ENDPOINT_TYPE') is not None:
242         sections = config.sections()
243         if os_utils.is_keystone_v3():
244             config.set('identity', 'v3_endpoint_type',
245                        CONST.__getattribute__('OS_ENDPOINT_TYPE'))
246             if 'identity-feature-enabled' not in sections:
247                 config.add_section('identity-feature-enabled')
248                 config.set('identity-feature-enabled', 'api_v2', False)
249                 config.set('identity-feature-enabled', 'api_v2_admin', False)
250         services_list = ['compute',
251                          'volume',
252                          'image',
253                          'network',
254                          'data-processing',
255                          'object-storage',
256                          'orchestration']
257         for service in services_list:
258             if service not in sections:
259                 config.add_section(service)
260             config.set(service, 'endpoint_type',
261                        CONST.__getattribute__('OS_ENDPOINT_TYPE'))
262
263     logger.debug('Add/Update required params defined in tempest_conf.yaml '
264                  'into tempest.conf file')
265     with open(TEMPEST_CONF_YAML) as f:
266         conf_yaml = yaml.safe_load(f)
267     if conf_yaml:
268         sections = config.sections()
269         for section in conf_yaml:
270             if section not in sections:
271                 config.add_section(section)
272             sub_conf = conf_yaml.get(section)
273             for key, value in sub_conf.items():
274                 config.set(section, key, value)
275
276     with open(tempest_conf_file, 'wb') as config_file:
277         config.write(config_file)
278
279     backup_tempest_config(tempest_conf_file)
280
281
282 def configure_verifier(deployment_dir):
283     """
284     Execute rally verify configure-verifier, which generates tempest.conf
285     """
286     tempest_conf_file = os.path.join(deployment_dir, "tempest.conf")
287     if os.path.isfile(tempest_conf_file):
288         logger.debug("Verifier is already configured.")
289         logger.debug("Reconfiguring the current verifier...")
290         cmd = "rally verify configure-verifier --reconfigure"
291     else:
292         logger.info("Configuring the verifier...")
293         cmd = "rally verify configure-verifier"
294     ft_utils.execute_command(cmd)
295
296     logger.debug("Looking for tempest.conf file...")
297     if not os.path.isfile(tempest_conf_file):
298         logger.error("Tempest configuration file %s NOT found."
299                      % tempest_conf_file)
300         raise Exception("Tempest configuration file %s NOT found."
301                         % tempest_conf_file)
302     else:
303         return tempest_conf_file