1 # Copyright (c) 2015 Intel Research and Development Ireland Ltd.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
15 from __future__ import absolute_import
17 import experimental_framework.heat_template_generation as heat_gen
19 __author__ = 'gpetralx'
22 class TestTreeNode(unittest.TestCase):
25 self.tree = heat_gen.TreeNode()
30 def test_add_child_for_success(self):
31 child = heat_gen.TreeNode()
32 self.tree.add_child(child)
33 self.assertIn(child, self.tree.down)
35 def test_get_parent_for_success(self):
36 self.assertIsNone(self.tree.get_parent())
37 child = heat_gen.TreeNode()
38 self.tree.add_child(child)
39 self.assertEqual(self.tree, child.get_parent())
41 def test_get_children_for_success(self):
42 self.assertListEqual(list(), self.tree.get_children())
43 child = heat_gen.TreeNode()
44 self.tree.add_child(child)
46 self.assertListEqual(children, self.tree.get_children())
48 def test_variable_name_for_success(self):
49 self.assertEqual('', self.tree.get_variable_name())
50 variable_name = 'test'
51 self.tree.set_variable_name(variable_name)
52 self.assertEqual(variable_name, self.tree.get_variable_name())
54 def test_variable_value_for_success(self):
55 self.assertEqual(0, self.tree.get_variable_value())
57 self.tree.set_variable_value(variable_value)
58 self.assertEqual(variable_value, self.tree.get_variable_value())
60 def test_get_path_for_success(self):
61 child_1 = heat_gen.TreeNode()
62 self.tree.add_child(child_1)
63 child_2 = heat_gen.TreeNode()
64 child_1.add_child(child_2)
65 child_3 = heat_gen.TreeNode()
66 child_2.add_child(child_3)
68 path = [self.tree, child_1, child_2, child_3]
70 self.assertListEqual(path, child_3.get_path())
72 def test_str_for_success(self):
75 self.tree.set_variable_name(name)
76 self.tree.set_variable_value(value)
77 self.assertEqual(name + " --> " + str(value), str(self.tree))
79 def test_repr_for_success(self):
82 self.tree.set_variable_name(name)
83 self.tree.set_variable_value(value)
84 self.assertEqual(name + " = " + str(value), repr(self.tree))
86 def test_get_leaves_for_success(self):
87 child_1 = heat_gen.TreeNode()
88 self.tree.add_child(child_1)
89 child_2 = heat_gen.TreeNode()
90 child_1.add_child(child_2)
91 child_3 = heat_gen.TreeNode()
92 child_2.add_child(child_3)
93 child_4 = heat_gen.TreeNode()
94 child_2.add_child(child_4)
95 child_5 = heat_gen.TreeNode()
96 child_2.add_child(child_5)
97 leaves = [child_3, child_4, child_5]
98 self.assertListEqual(leaves, heat_gen.TreeNode.get_leaves(self.tree))