diff options
Diffstat (limited to 'ice_validator/tests/utils/vm_types.py')
| -rw-r--r-- | ice_validator/tests/utils/vm_types.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/ice_validator/tests/utils/vm_types.py b/ice_validator/tests/utils/vm_types.py index 78006b9..327b75b 100644 --- a/ice_validator/tests/utils/vm_types.py +++ b/ice_validator/tests/utils/vm_types.py @@ -2,7 +2,7 @@ # ============LICENSE_START======================================================= # org.onap.vvp/validation-scripts # =================================================================== -# Copyright © 2017 AT&T Intellectual Property. All rights reserved. +# Copyright © 2019 AT&T Intellectual Property. All rights reserved. # =================================================================== # # Unless otherwise specified, all software contained herein is licensed @@ -39,6 +39,7 @@ # import re +from tests import cached_yaml as yaml def get_vm_types_for_resource(resource): @@ -111,3 +112,20 @@ def get_vm_types(resources): vm_types.extend(list(get_vm_types_for_resource(v))) return set(vm_types) + + +def get_all_vm_types(yaml_files): + """ + Get all vm_types for a list of yaml files + """ + vm_types = [] + for yaml_file in yaml_files: + with open(yaml_file, "r") as f: + yml = yaml.load(f) + + if "resources" not in yml: + continue + + vm_types.extend(get_vm_types(yml["resources"])) + + return set(vm_types) |