diff options
author | Mickael JEZEQUEL <mickael.jezequel@orange.com> | 2018-02-28 15:47:27 +0100 |
---|---|---|
committer | Mickael JEZEQUEL <mickael.jezequel@orange.com> | 2018-03-01 14:59:10 +0100 |
commit | e2bdf848d6d798d54f94f8e3a9e130910159f1d9 (patch) | |
tree | 2a9d49fb7557cbde763bf841fc633c1c5be53a31 /validation-scripts/ice_validator/tests/utils/vm_types.py | |
parent | 6027a1ad84f129cf9f8996b3020849ec1461085b (diff) |
Import ICE tests from vvp/validation-scripts
Change-Id: Ia5de2787792dcc61d85f297a3262a80faa1f6d56
Issue-ID: VNFSDK-39
Signed-off-by: Mickael JEZEQUEL <mickael.jezequel@orange.com>
Diffstat (limited to 'validation-scripts/ice_validator/tests/utils/vm_types.py')
-rw-r--r-- | validation-scripts/ice_validator/tests/utils/vm_types.py | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/validation-scripts/ice_validator/tests/utils/vm_types.py b/validation-scripts/ice_validator/tests/utils/vm_types.py new file mode 100644 index 0000000..5bd447b --- /dev/null +++ b/validation-scripts/ice_validator/tests/utils/vm_types.py @@ -0,0 +1,117 @@ +# -*- coding: utf8 -*- +# ============LICENSE_START======================================================= +# org.onap.vvp/validation-scripts +# =================================================================== +# Copyright © 2017 AT&T Intellectual Property. All rights reserved. +# =================================================================== +# +# Unless otherwise specified, all software contained herein is licensed +# under the Apache License, Version 2.0 (the “License”); +# you may not use this software except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# +# +# Unless otherwise specified, all documentation contained herein is licensed +# under the Creative Commons License, Attribution 4.0 Intl. (the “License”); +# you may not use this documentation except in compliance with the License. +# You may obtain a copy of the License at +# +# https://creativecommons.org/licenses/by/4.0/ +# +# Unless required by applicable law or agreed to in writing, documentation +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# ============LICENSE_END============================================ +# +# ECOMP is a trademark and service mark of AT&T Intellectual Property. +# + +import re + + +def get_vm_types_for_resource(resource): + ''' + Get all unique vm_types for a resource + Notes: + - Returns set([]) if the resource is not formatted + properly, the passed resource is not a nova server + - If more than one vm_type is detected all vm_types will + be returned + ''' + if not isinstance(resource, dict): + return set() + if 'type' not in resource: + return set() + if resource['type'] != 'OS::Nova::Server': + return set() + if 'properties' not in resource: + return set() + + key_values = ["name", "flavor", "image"] + key_value_formats = [ + ["name", "string", + re.compile(r'(.+?)_name_\d+')], + ["name", "comma_delimited_list", + re.compile(r'(.+?)_names')], + ["flavor", "string", + re.compile(r'(.+?)_flavor_name')], + ["image", "string", + re.compile(r'(.+?)_image_name')], + ] + + vm_types = [] + for k2, v2 in resource['properties'].items(): + if k2 not in key_values: + continue + if "get_param" not in v2: + continue + formats = [v for v in key_value_formats if v[0] == k2] + for v3 in formats: + param = v2["get_param"] + if isinstance(param, list): + param = param[0] + m = v3[2].match(param) + if m and m.group(1): + vm_types.append(m.group(1)) + + return set(vm_types) + + +def get_vm_type_for_nova_server(resource): + ''' + Get the vm_type for a resource + Note: Returns None if not exactly one vm_type + is detected, if the resource is not formatted properly, or + the passed resource is not a nova server + ''' + vm_types = get_vm_types_for_resource(resource) + + # if more than one vm_type was identified, return None + if len(vm_types) > 1: + return None + + return vm_types.pop() + + +def get_vm_types(resources): + ''' + Get all vm_types for a list of heat resources, do note that + some of the values retrieved may be invalid + ''' + vm_types = [] + for v in resources.values(): + vm_types.extend(list(get_vm_types_for_resource(v))) + + return set(vm_types) |