diff options
Diffstat (limited to 'test/vcpe/preload.py')
-rwxr-xr-x | test/vcpe/preload.py | 98 |
1 files changed, 88 insertions, 10 deletions
diff --git a/test/vcpe/preload.py b/test/vcpe/preload.py index f99d8dee1..62c036c83 100755 --- a/test/vcpe/preload.py +++ b/test/vcpe/preload.py @@ -4,7 +4,6 @@ import requests import json import sys from datetime import datetime -from vcpecommon import * import csar_parser import logging import base64 @@ -27,14 +26,14 @@ class Preload: with open(template_file) as json_input: json_data = json.load(json_input) stk = [json_data] - while len(stk) > 0: + while stk: data = stk.pop() for k, v in data.items(): if type(v) is dict: stk.append(v) elif type(v) is list: stk.extend(v) - elif type(v) is str or type(v) is unicode: + elif type(v) is str or type(v) is unicode: # pylint: disable=E0602 if self.vcpecommon.template_variable_symbol in v: data[k] = self.replace(v, replace_dict) else: @@ -103,9 +102,9 @@ class Preload: self.logger.info('Preloading network ' + network_role) self.logger.info('template_file:' + template_file) if 'networkgra' in template_file: - return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_network_gra_url) + return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_network_gra_url) else: - return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_network_url) + return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_network_url) def preload(self, template_file, replace_dict, url): self.logger.debug('tempalte_file:'+ template_file) @@ -166,10 +165,10 @@ class Preload: '${suffix}': name_suffix} replace_dict.update(common_dict) self.logger.info('Preloading VF Module ' + vfmodule_name) - if gra_api_flag: - return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_gra_url) + if gra_api_flag: + return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_gra_url) else: - return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_vnf_url) + return self.preload(template_file, replace_dict, self.vcpecommon.sdnc_preload_vnf_url) def preload_all_networks(self, template_file, name_suffix): common_dict = {'${' + k + '}': v for k, v in self.vcpecommon.common_preload_config.items()} @@ -180,11 +179,90 @@ class Preload: return None return common_dict + def aai_region_query(self, req_method, json=None, verify=False): + """ + Perform actual AAI API request for region + :param req_method: request method ({'get','put'}) + :param json: Json payload + :param verify: SSL verify mode + :return: + """ + url, headers, auth = (self.vcpecommon.aai_region_query_url, + self.vcpecommon.aai_headers, + self.vcpecommon.aai_userpass) + try: + if req_method == 'get': + request = requests.get(url, headers=headers, auth=auth, + verify=verify) + elif req_method == 'put': + request = requests.put(url, headers=headers, auth=auth, + verify=verify, json=json) + else: + raise requests.exceptions.RequestException + except requests.exceptions.RequestException as e: + self.logger.error("Error connecting to AAI API. Error details: " + str(e.message)) + return False + try: + assert request.status_code == 200 + except AssertionError: + self.logger.error('AAI request failed. API returned http code ' + str(request.status_code)) + return False + try: + return request.json() + except ValueError as e: + if req_method == 'get': + self.logger.error('Unable to parse AAI response: ' + e.message) + return False + elif req_method == 'put': + return request.ok + else: + return False + + def preload_aai_data(self, template_aai_region_data): + """ + Update aai region data with identity-url + :param template_aai_region_data: path to region data template + :return: + """ + request = self.aai_region_query('get') + if request: + # Check if identity-url already updated (for idempotency) + self.logger.debug("Regiond data acquired from AAI:\n" + json.dumps(request,indent=4)) + try: + assert request['identity-url'] + except KeyError: + pass + else: + self.logger.info('Identity-url already present in {0} data, not updating'.format(self.vcpecommon.cloud['--os-region-name'])) + return + + # Get resource_version and relationship_list from region data + resource_version = request['resource-version'] + relationship_list = request['relationship-list'] + + replace_dict = {'${identity-url}': self.vcpecommon.cloud['--os-auth-url'], + '${identity_api_version}': self.vcpecommon.cloud['--os-identity-api-version'], + '${region_name}': self.vcpecommon.cloud['--os-region-name'], + '${resource_version}': resource_version + } + json_data = self.generate_json(template_aai_region_data, replace_dict) + json_data['relationship-list'] = relationship_list + self.logger.debug('Region update payload:\n' + json.dumps(json_data,indent=4)) + else: + sys.exit(1) + + # Update region data + request = self.aai_region_query('put', json_data) + if request: + self.logger.info('Successully updated identity-url in {0} ' + 'region'.format(self.vcpecommon.cloud['--os-region-name'])) + else: + sys.exit(1) + def test(self): # this is for testing purpose name_suffix = datetime.now().strftime('%Y%m%d%H%M') - vcpecommon = VcpeCommon() - preloader = Preload(vcpecommon) + preloader = Preload(self.vcpecommon) network_dict = {'${' + k + '}': v for k, v in self.vcpecommon.common_preload_config.items()} template_file = 'preload_templates/template.network.json' |