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.
16 from experimental_framework import heat_manager
19 __author__ = 'gpetralx'
22 def get_mock_heat(version, *args, **kwargs):
26 class MockStacks(object):
27 def __init__(self, stacks):
32 for stack in self.stacks:
33 list_name.append(stack.stack_name)
37 def validate(self, template=None):
41 for stack in self.stacks:
43 return self.stacks.remove(stack)
45 def create(self, stack_name=None, files=None, template=None,
48 self.stacks.append(MockStack(stack_name))
51 class MockStacks_2(object):
52 def __init__(self, stacks):
59 class MockStack(object):
60 def __init__(self, stack_name):
61 self.name = stack_name
64 def stack_status(self):
65 return self.stack_name + '_status'
75 def __eq__(self, other):
76 return self.name == other
79 class MockHeat(object):
81 stacks = [MockStack('stack_1'), MockStack('stack_2')]
82 self.stacks_list = MockStacks(stacks)
86 return self.stacks_list
89 class MockHeat_2(MockHeat):
91 stacks = [MockStack('stack_1'), MockStack('stack_2')]
92 self.stacks_list = MockStacks_2(stacks)
95 class HeatManagerMock(heat_manager.HeatManager):
98 self.heat = MockHeat()
101 class HeatManagerMock_2(heat_manager.HeatManager):
103 if self.heat is None:
104 self.heat = MockHeat_2()
107 class TestHeatManager(unittest.TestCase):
111 credentials['ip_controller'] = '1.1.1.1'
112 credentials['heat_url'] = 'http://heat_url'
113 credentials['user'] = 'user'
114 credentials['password'] = 'password'
115 credentials['auth_uri'] = 'auth_uri'
116 credentials['project'] = 'project'
117 self.heat_manager = HeatManagerMock(credentials)
118 self.heat_manager.init_heat()
123 def test_is_stack_deployed_for_success(self):
124 self.assertTrue(self.heat_manager.is_stack_deployed('stack_1'))
125 self.assertFalse(self.heat_manager.is_stack_deployed('stack_n'))
127 def test_check_status_for_success(self):
128 self.assertEqual('stack_1_status',
129 self.heat_manager.check_stack_status('stack_1'))
130 self.assertEqual('NOT_FOUND',
131 self.heat_manager.check_stack_status('stack_x'))
133 def test_validate_template_for_success(self):
135 'tests/data/test_templates/VTC_base_single_vm_wait_1.yaml'
136 with self.assertRaises(ValueError):
137 self.heat_manager.validate_heat_template(template_file)
139 def test_delete_stack_for_success(self):
140 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
141 self.assertFalse(self.heat_manager.delete_stack('stack_x'))
143 def test_delete_stack_for_success_2(self):
144 self.assertTrue(self.heat_manager.delete_stack('stack_1'))
146 @mock.patch('experimental_framework.common.LOG')
147 @mock.patch('heatclient.common.template_utils.get_template_contents')
148 @mock.patch('heatclient.client.Client')
149 # @mock.patch('heatclient.client.Client', side_effect=DummyHeatClient)
150 def test_create_stack_for_success(self, mock_stack_create,
151 mock_get_template_contents,
153 return_value = ({'template': 'template'}, 'template')
154 mock_get_template_contents.return_value = return_value
155 self.heat_manager.create_stack('template', 'stack_n', 'parameters')
156 self.assertTrue(self.heat_manager.is_stack_deployed('stack_n'))
159 class TestHeatManager_2(unittest.TestCase):
163 credentials['ip_controller'] = '1.1.1.1'
164 credentials['heat_url'] = 'http://heat_url'
165 credentials['user'] = 'user'
166 credentials['password'] = 'password'
167 credentials['auth_uri'] = 'auth_uri'
168 credentials['project'] = 'project'
169 self.heat_manager = HeatManagerMock_2(credentials)
174 def test_delete_stack_for_success_2(self):
175 self.assertFalse(self.heat_manager.delete_stack('stack_1'))
178 class ServiceCatalog():
179 def url_for(self, service_type):
180 return 'http://heat_url'
183 class KeystoneMock(object):
185 def auth_token(self):
188 service_catalog = ServiceCatalog()
191 class TestHeatInit(unittest.TestCase):
194 credentials['ip_controller'] = '1.1.1.1'
195 credentials['heat_url'] = 'http://heat_url'
196 credentials['user'] = 'user'
197 credentials['password'] = 'password'
198 credentials['auth_uri'] = 'auth_uri'
199 credentials['project'] = 'project'
200 self.heat_manager = heat_manager.HeatManager(credentials)
205 @mock.patch('heatclient.client.Client')
206 @mock.patch('keystoneclient.v2_0.client.Client')
207 def test_heat_init_for_sanity(self, keystone_client, heat_client):
208 keystone_client.return_value = KeystoneMock()
209 heat_client.return_value = MockHeat()
210 self.heat_manager.init_heat()
211 keystone_client.assert_called_once_with(username='user',
212 tenant_name='project',
215 heat_client.assert_called_once_with('1', endpoint='http://heat_url',