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.
17 import experimental_framework.common as common
19 from experimental_framework import heat_manager
22 __author__ = 'gpetralx'
25 def get_mock_heat(version, *args, **kwargs):
29 class MockStacks(object):
30 def __init__(self, stacks):
35 for stack in self.stacks:
36 list_name.append(stack.stack_name)
40 def validate(self, template=None):
44 for stack in self.stacks:
46 return self.stacks.remove(stack)
48 def create(self, stack_name=None, files=None, template=None,
51 self.stacks.append(MockStack(stack_name))
54 class MockStacks_2(object):
55 def __init__(self, stacks):
62 class MockStack(object):
63 def __init__(self, stack_name):
64 self.name = stack_name
67 def stack_status(self):
68 return self.stack_name + '_status'
78 def __eq__(self, other):
79 return self.name == other
82 class MockHeat(object):
84 stacks = [MockStack('stack_1'), MockStack('stack_2')]
85 self.stacks_list = MockStacks(stacks)
89 return self.stacks_list
92 class MockHeat_2(MockHeat):
94 stacks = [MockStack('stack_1'), MockStack('stack_2')]
95 self.stacks_list = MockStacks_2(stacks)
98 class HeatManagerMock(heat_manager.HeatManager):
100 if self.heat is None:
101 self.heat = MockHeat()
104 class HeatManagerMock_2(heat_manager.HeatManager):
106 if self.heat is None:
107 self.heat = MockHeat_2()
110 class TestHeatManager(unittest.TestCase):
114 credentials['ip_controller'] = '1.1.1.1'
115 credentials['heat_url'] = 'http://heat_url'
116 credentials['user'] = 'user'
117 credentials['password'] = 'password'
118 credentials['auth_uri'] = 'auth_uri'
119 credentials['project'] = 'project'
120 self.heat_manager = HeatManagerMock(credentials)
121 self.heat_manager.init_heat()
126 def test_is_stack_deployed_for_success(self):
127 self.assertTrue(self.heat_manager.is_stack_deployed('stack_1'))
128 self.assertFalse(self.heat_manager.is_stack_deployed('stack_n'))
130 def test_check_status_for_success(self):
131 self.assertEqual('stack_1_status',
132 self.heat_manager.check_stack_status('stack_1'))
133 self.assertEqual('NOT_FOUND',
134 self.heat_manager.check_stack_status('stack_x'))
136 def test_validate_template_for_success(self):
138 'tests/data/test_templates/VTC_base_single_vm_wait_1.yaml'
139 with self.assertRaises(ValueError):
140 self.heat_manager.validate_heat_template(template_file)
142 def test_delete_stack_for_success(self):
143 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
144 self.assertFalse(self.heat_manager.delete_stack('stack_x'))
146 def test_delete_stack_for_success_2(self):
147 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
149 @mock.patch('experimental_framework.common.LOG')
150 @mock.patch('heatclient.common.template_utils.get_template_contents')
151 @mock.patch('heatclient.client.Client')
152 # @mock.patch('heatclient.client.Client', side_effect=DummyHeatClient)
153 def test_create_stack_for_success(self, mock_stack_create,
154 mock_get_template_contents,
156 return_value = ({'template': 'template'}, 'template')
157 mock_get_template_contents.return_value = return_value
158 self.heat_manager.create_stack('template', 'stack_n', 'parameters')
159 self.assertTrue(self.heat_manager.is_stack_deployed('stack_n'))
162 class TestHeatManager_2(unittest.TestCase):
166 credentials['ip_controller'] = '1.1.1.1'
167 credentials['heat_url'] = 'http://heat_url'
168 credentials['user'] = 'user'
169 credentials['password'] = 'password'
170 credentials['auth_uri'] = 'auth_uri'
171 credentials['project'] = 'project'
172 self.heat_manager = HeatManagerMock_2(credentials)
177 def test_delete_stack_for_success_2(self):
178 common.LOG = logging.getLogger()
179 self.assertFalse(self.heat_manager.delete_stack('stack_1'))
182 class ServiceCatalog():
183 def url_for(self, service_type):
184 return 'http://heat_url'
187 class KeystoneMock(object):
189 def auth_token(self):
192 service_catalog = ServiceCatalog()
195 class TestHeatInit(unittest.TestCase):
198 credentials['ip_controller'] = '1.1.1.1'
199 credentials['heat_url'] = 'http://heat_url'
200 credentials['user'] = 'user'
201 credentials['password'] = 'password'
202 credentials['auth_uri'] = 'auth_uri'
203 credentials['project'] = 'project'
204 self.heat_manager = heat_manager.HeatManager(credentials)
209 @mock.patch('heatclient.client.Client')
210 @mock.patch('keystoneclient.v2_0.client.Client')
211 def test_heat_init_for_sanity(self, keystone_client, heat_client):
212 keystone_client.return_value = KeystoneMock()
213 heat_client.return_value = MockHeat()
214 self.heat_manager.init_heat()
215 keystone_client.assert_called_once_with(username='user',
216 tenant_name='project',
219 heat_client.assert_called_once_with('1', endpoint='http://heat_url',