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 __author__ = 'gpetralx'
19 from experimental_framework import heat_manager
23 def get_mock_heat(version, *args, **kwargs):
27 class MockStacks(object):
28 def __init__(self, stacks):
33 for stack in self.stacks:
34 list_name.append(stack.stack_name)
38 def validate(self, template=None):
42 for stack in self.stacks:
44 return self.stacks.remove(stack)
46 def create(self, stack_name=None, files=None, template=None,
49 self.stacks.append(MockStack(stack_name))
52 class MockStacks_2(object):
53 def __init__(self, stacks):
60 class MockStack(object):
61 def __init__(self, stack_name):
62 self.name = stack_name
65 def stack_status(self):
66 return self.stack_name + '_status'
76 def __eq__(self, other):
77 return self.name == other
80 class MockHeat(object):
82 stacks = [MockStack('stack_1'), MockStack('stack_2')]
83 self.stacks_list = MockStacks(stacks)
87 return self.stacks_list
90 class MockHeat_2(MockHeat):
92 stacks = [MockStack('stack_1'), MockStack('stack_2')]
93 self.stacks_list = MockStacks_2(stacks)
96 class HeatManagerMock(heat_manager.HeatManager):
99 self.heat = MockHeat()
102 class HeatManagerMock_2(heat_manager.HeatManager):
104 if self.heat is None:
105 self.heat = MockHeat_2()
108 class TestHeatManager(unittest.TestCase):
112 credentials['ip_controller'] = '1.1.1.1'
113 credentials['heat_url'] = 'http://heat_url'
114 credentials['user'] = 'user'
115 credentials['password'] = 'password'
116 credentials['auth_uri'] = 'auth_uri'
117 credentials['project'] = 'project'
118 self.heat_manager = HeatManagerMock(credentials)
119 self.heat_manager.init_heat()
124 def test_is_stack_deployed_for_success(self):
125 self.assertTrue(self.heat_manager.is_stack_deployed('stack_1'))
126 self.assertFalse(self.heat_manager.is_stack_deployed('stack_n'))
128 def test_check_status_for_success(self):
129 self.assertEqual('stack_1_status',
130 self.heat_manager.check_stack_status('stack_1'))
131 self.assertEqual('NOT_FOUND',
132 self.heat_manager.check_stack_status('stack_x'))
134 def test_validate_template_for_success(self):
136 'tests/data/test_templates/VTC_base_single_vm_wait_1.yaml'
137 with self.assertRaises(ValueError):
138 self.heat_manager.validate_heat_template(template_file)
140 def test_delete_stack_for_success(self):
141 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
142 self.assertFalse(self.heat_manager.delete_stack('stack_x'))
144 def test_delete_stack_for_success_2(self):
145 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
147 @mock.patch('experimental_framework.common.LOG')
148 @mock.patch('heatclient.common.template_utils.get_template_contents')
149 @mock.patch('heatclient.client.Client')
150 # @mock.patch('heatclient.client.Client', side_effect=DummyHeatClient)
151 def test_create_stack_for_success(self, mock_stack_create,
152 mock_get_template_contents,
154 return_value = ({'template': 'template'}, 'template')
155 mock_get_template_contents.return_value = return_value
156 self.heat_manager.create_stack('template', 'stack_n', 'parameters')
157 self.assertTrue(self.heat_manager.is_stack_deployed('stack_n'))
160 class TestHeatManager_2(unittest.TestCase):
164 credentials['ip_controller'] = '1.1.1.1'
165 credentials['heat_url'] = 'http://heat_url'
166 credentials['user'] = 'user'
167 credentials['password'] = 'password'
168 credentials['auth_uri'] = 'auth_uri'
169 credentials['project'] = 'project'
170 self.heat_manager = HeatManagerMock_2(credentials)
175 def test_delete_stack_for_success_2(self):
176 self.assertFalse(self.heat_manager.delete_stack('stack_1'))
179 class KeystoneMock(object):
181 def auth_token(self):
185 class TestHeatInit(unittest.TestCase):
188 credentials['ip_controller'] = '1.1.1.1'
189 credentials['heat_url'] = 'http://heat_url'
190 credentials['user'] = 'user'
191 credentials['password'] = 'password'
192 credentials['auth_uri'] = 'auth_uri'
193 credentials['project'] = 'project'
194 self.heat_manager = heat_manager.HeatManager(credentials)
199 @mock.patch('heatclient.client.Client')
200 @mock.patch('keystoneclient.v2_0.client.Client')
201 def test_heat_init_for_sanity(self, keystone_client, heat_client):
202 keystone_client.return_value = KeystoneMock()
203 heat_client.return_value = MockHeat()
204 self.heat_manager.init_heat()
205 keystone_client.assert_called_once_with(username='user',
206 tenant_name='project',
209 heat_client.assert_called_once_with('1', endpoint='http://heat_url',