Matteo Scandolo | 920e8fd | 2017-08-08 13:05:24 -0700 | [diff] [blame] | 1 | # Copyright 2017-present Open Networking Foundation |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 15 | from __future__ import absolute_import |
| 16 | from . import helpers # noqa: F401 |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 17 | import unittest |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 18 | from tosca.parser import TOSCA_Parser |
| 19 | |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 20 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 21 | class TOSCA_Parser_Test(unittest.TestCase): |
Matteo Scandolo | 485b713 | 2017-06-30 11:46:47 -0700 | [diff] [blame] | 22 | def test_get_tosca_models_by_name(self): |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 23 | """ |
Matteo Scandolo | 485b713 | 2017-06-30 11:46:47 -0700 | [diff] [blame] | 24 | [TOSCA_Parser] get_tosca_models_by_name: should extract models from the TOSCA recipe and store them in a dict |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 25 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 26 | |
Matteo Scandolo | 485b713 | 2017-06-30 11:46:47 -0700 | [diff] [blame] | 27 | class FakeNode: |
| 28 | def __init__(self, name): |
| 29 | self.name = name |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 30 | |
Matteo Scandolo | 485b713 | 2017-06-30 11:46:47 -0700 | [diff] [blame] | 31 | class FakeTemplate: |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 32 | nodetemplates = [FakeNode("model1"), FakeNode("model2")] |
Matteo Scandolo | 9ce1825 | 2017-06-22 10:48:25 -0700 | [diff] [blame] | 33 | |
Matteo Scandolo | 485b713 | 2017-06-30 11:46:47 -0700 | [diff] [blame] | 34 | res = TOSCA_Parser.get_tosca_models_by_name(FakeTemplate) |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 35 | self.assertIsInstance(res["model1"], FakeNode) |
| 36 | self.assertIsInstance(res["model2"], FakeNode) |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 37 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 38 | self.assertEqual(res["model1"].name, "model1") |
| 39 | self.assertEqual(res["model2"].name, "model2") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 40 | |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 41 | def test_populate_dependencies(self): |
| 42 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 43 | [TOSCA_Parser] populate_dependencies: if a recipe has dependencies, it |
| 44 | should find the ID of the requirements and add it to the model |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 45 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 46 | |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 47 | class FakeRecipe: |
| 48 | requirements = [ |
| 49 | { |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 50 | "site": { |
| 51 | "node": "site_onlab", |
| 52 | "relationship": "tosca.relationship.BelongsToOne", |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 53 | } |
| 54 | } |
| 55 | ] |
| 56 | |
| 57 | class FakeSite: |
| 58 | id = 1 |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 59 | name = "onlab" |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 60 | |
| 61 | class FakeModel: |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 62 | name = "test@opencord.org" |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 63 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 64 | saved_models = {"site_onlab": FakeSite} |
Matteo Scandolo | 8bbb03a | 2017-07-05 14:03:33 -0700 | [diff] [blame] | 65 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 66 | model = TOSCA_Parser.populate_dependencies( |
| 67 | FakeModel, FakeRecipe.requirements, saved_models |
| 68 | ) |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 69 | self.assertEqual(model.site_id, 1) |
| 70 | |
| 71 | def test_get_ordered_models_template(self): |
| 72 | """ |
| 73 | [TOSCA_Parser] get_ordered_models_template: Create a list of templates based on topsorted models |
| 74 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 75 | ordered_models = ["foo", "bar"] |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 76 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 77 | templates = {"foo": "foo_template", "bar": "bar_template"} |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 78 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 79 | ordered_templates = TOSCA_Parser.get_ordered_models_template( |
| 80 | ordered_models, templates |
| 81 | ) |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 82 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 83 | self.assertEqual(ordered_templates[0], "foo_template") |
| 84 | self.assertEqual(ordered_templates[1], "bar_template") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 85 | |
| 86 | def test_topsort_dependencies(self): |
| 87 | """ |
| 88 | [TOSCA_Parser] topsort_dependencies: Create a list of models based on dependencies |
| 89 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 90 | |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 91 | class FakeTemplate: |
| 92 | def __init__(self, name, deps): |
| 93 | self.name = name |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 94 | self.dependencies_names = deps |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 95 | |
| 96 | templates = { |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 97 | "deps": FakeTemplate("deps", ["main"]), |
| 98 | "main": FakeTemplate("main", []), |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 99 | } |
| 100 | |
| 101 | sorted = TOSCA_Parser.topsort_dependencies(templates) |
| 102 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 103 | self.assertEqual(sorted[0], "main") |
| 104 | self.assertEqual(sorted[1], "deps") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 105 | |
| 106 | def test_compute_dependencies(self): |
| 107 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 108 | [TOSCA_Parser] compute_dependencies: augment the TOSCA nodetemplate |
| 109 | with information on requirements (aka related models) |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 110 | """ |
| 111 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 112 | parser = TOSCA_Parser("", "user", "pass") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 113 | |
| 114 | class FakeNode: |
| 115 | def __init__(self, name, requirements): |
| 116 | self.name = name |
| 117 | self.requirements = requirements |
| 118 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 119 | main = FakeNode("main", []) |
| 120 | dep = FakeNode("dep", [{"relation": {"node": "main"}}]) |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 121 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 122 | models_by_name = {"main": main, "dep": dep} |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 123 | |
| 124 | class FakeTemplate: |
| 125 | nodetemplates = [dep, main] |
| 126 | |
| 127 | parser.compute_dependencies(FakeTemplate, models_by_name) |
| 128 | |
| 129 | templates = FakeTemplate.nodetemplates |
| 130 | augmented_dep = templates[0] |
| 131 | augmented_main = templates[1] |
| 132 | |
| 133 | self.assertIsInstance(augmented_dep.dependencies[0], FakeNode) |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 134 | self.assertEqual(augmented_dep.dependencies[0].name, "main") |
| 135 | self.assertEqual(augmented_dep.dependencies_names[0], "main") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 136 | |
| 137 | self.assertEqual(len(augmented_main.dependencies), 0) |
| 138 | self.assertEqual(len(augmented_main.dependencies_names), 0) |
| 139 | |
| 140 | def test_populate_model(self): |
| 141 | """ |
| 142 | [TOSCA_Parser] populate_model: augment the GRPC model with data from TOSCA |
| 143 | """ |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 144 | |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 145 | class FakeModel: |
| 146 | pass |
| 147 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 148 | data = {"name": "test", "foo": "bar", "number": 1} |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 149 | |
| 150 | model = TOSCA_Parser.populate_model(FakeModel, data) |
| 151 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 152 | self.assertEqual(model.name, "test") |
| 153 | self.assertEqual(model.foo, "bar") |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 154 | self.assertEqual(model.number, 1) |
| 155 | |
Matteo Scandolo | 5ccdb13 | 2018-01-17 14:02:12 -0800 | [diff] [blame] | 156 | def test_populate_model_error(self): |
| 157 | """ |
| 158 | [TOSCA_Parser] populate_model: should print a meaningful error message |
| 159 | """ |
| 160 | |
| 161 | class FakeModel: |
| 162 | |
| 163 | model_name = "FakeModel" |
| 164 | |
| 165 | def __setattr__(self, name, value): |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 166 | if name == "foo": |
| 167 | raise TypeError("reported exception") |
Matteo Scandolo | 5ccdb13 | 2018-01-17 14:02:12 -0800 | [diff] [blame] | 168 | else: |
| 169 | super(FakeModel, self).__setattr__(name, value) |
| 170 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 171 | data = {"name": "test", "foo": None, "number": 1} |
Matteo Scandolo | 5ccdb13 | 2018-01-17 14:02:12 -0800 | [diff] [blame] | 172 | |
| 173 | model = FakeModel() |
| 174 | |
| 175 | with self.assertRaises(Exception) as e: |
| 176 | model = TOSCA_Parser.populate_model(model, data) |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 177 | self.assertEqual( |
| 178 | str(e.exception), |
| 179 | 'Failed to set None on field foo for class FakeModel, Exception was: "reported exception"', |
| 180 | ) |
Matteo Scandolo | 5ccdb13 | 2018-01-17 14:02:12 -0800 | [diff] [blame] | 181 | |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 182 | def test_translate_exception(self): |
| 183 | """ |
| 184 | [TOSCA_Parser] translate_exception: convert a TOSCA Parser exception in a user readable string |
| 185 | """ |
| 186 | e = TOSCA_Parser._translate_exception("Non tosca exception") |
| 187 | self.assertEqual(e, "Non tosca exception") |
| 188 | |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 189 | e = TOSCA_Parser._translate_exception( |
| 190 | """ |
Matteo Scandolo | df2600b | 2017-07-05 17:01:29 -0700 | [diff] [blame] | 191 | MissingRequiredFieldError: some message |
| 192 | followed by unreadable |
| 193 | and mystic |
| 194 | python error |
| 195 | starting at line |
| 196 | 38209834 of some file |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 197 | UnknownFieldError: with some message |
| 198 | followed by useless things |
| 199 | ImportError: with some message |
| 200 | followed by useless things |
| 201 | InvalidTypeError: with some message |
| 202 | followed by useless things |
| 203 | TypeMismatchError: with some message |
| 204 | followed by useless things |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 205 | """ |
| 206 | ) |
| 207 | self.assertEqual( |
| 208 | e, |
| 209 | """MissingRequiredFieldError: some message |
Matteo Scandolo | 1fedfae | 2017-10-09 13:57:00 -0700 | [diff] [blame] | 210 | UnknownFieldError: with some message |
| 211 | ImportError: with some message |
| 212 | InvalidTypeError: with some message |
| 213 | TypeMismatchError: with some message |
Zack Williams | 6bb2cfe | 2019-03-27 15:01:45 -0700 | [diff] [blame^] | 214 | """, |
| 215 | ) |