102 lines
3.7 KiB
Python
102 lines
3.7 KiB
Python
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
|
|
from heat.engine.clients.os import cinder as c_plugin
|
|
from heat.engine.resources.openstack.cinder import qos_specs
|
|
from heat.engine import stack
|
|
from heat.engine import template
|
|
from heat.tests import common
|
|
from heat.tests import utils
|
|
|
|
QOS_SPECS_TEMPLATE = {
|
|
'heat_template_version': '2015-10-15',
|
|
'description': 'Cinder QoS specs creation example',
|
|
'resources': {
|
|
'my_qos_specs': {
|
|
'type': 'OS::Cinder::QoSSpecs',
|
|
'properties': {
|
|
'name': 'foobar',
|
|
'specs': {"foo": "bar", "foo1": "bar1"}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
class QoSSpecsTest(common.HeatTestCase):
|
|
|
|
def setUp(self):
|
|
super(QoSSpecsTest, self).setUp()
|
|
self.ctx = utils.dummy_context()
|
|
self.patchobject(c_plugin.CinderClientPlugin, 'has_extension',
|
|
return_value=True)
|
|
self.stack = stack.Stack(
|
|
self.ctx, 'cinder_qos_spec_test_stack',
|
|
template.Template(QOS_SPECS_TEMPLATE)
|
|
)
|
|
self.my_qos_specs = self.stack['my_qos_specs']
|
|
cinder_client = mock.MagicMock()
|
|
self.cinderclient = mock.MagicMock()
|
|
self.my_qos_specs.client = cinder_client
|
|
cinder_client.return_value = self.cinderclient
|
|
self.qos_specs = self.cinderclient.qos_specs
|
|
self.value = mock.MagicMock()
|
|
self.value.id = '927202df-1afb-497f-8368-9c2d2f26e5db'
|
|
self.value.name = 'foobar'
|
|
self.value.specs = {"foo": "bar", "foo1": "bar1"}
|
|
self.qos_specs.create.return_value = self.value
|
|
|
|
def test_resource_mapping(self):
|
|
mapping = qos_specs.resource_mapping()
|
|
self.assertEqual(1, len(mapping))
|
|
self.assertEqual(qos_specs.QoSSpecs,
|
|
mapping['OS::Cinder::QoSSpecs'])
|
|
self.assertIsInstance(self.my_qos_specs,
|
|
qos_specs.QoSSpecs)
|
|
|
|
def _set_up_qos_specs_environment(self):
|
|
self.qos_specs.create.return_value = self.value
|
|
self.my_qos_specs.handle_create()
|
|
|
|
def test_qos_specs_handle_create_specs(self):
|
|
self._set_up_qos_specs_environment()
|
|
self.assertEqual(1, self.qos_specs.create.call_count)
|
|
self.assertEqual(self.value.id, self.my_qos_specs.resource_id)
|
|
|
|
def test_qos_specs_handle_update_specs(self):
|
|
self._set_up_qos_specs_environment()
|
|
resource_id = self.my_qos_specs.resource_id
|
|
prop_diff = {'specs': {"foo": "bar", "bar": "bar"}}
|
|
set_expected = {"bar": "bar"}
|
|
unset_expected = ["foo1"]
|
|
|
|
self.my_qos_specs.handle_update(
|
|
json_snippet=None, tmpl_diff=None, prop_diff=prop_diff
|
|
)
|
|
self.qos_specs.set_keys.assert_called_once_with(
|
|
resource_id,
|
|
set_expected
|
|
)
|
|
self.qos_specs.unset_keys.assert_called_once_with(
|
|
resource_id,
|
|
unset_expected
|
|
)
|
|
|
|
def test_qos_specs_handle_delete_specs(self):
|
|
self._set_up_qos_specs_environment()
|
|
resource_id = self.my_qos_specs.resource_id
|
|
self.my_qos_specs.handle_delete()
|
|
self.qos_specs.disassociate_all.assert_called_once_with(resource_id)
|