cb8712281d
Use parentheses instead of backslash for line continuation. Plus, usage of escaped quotes in long strings is avoided where it makes sense. Change-Id: If2e78012b85a4430c6f03f65784cac2d032cf116
124 lines
4.4 KiB
Python
124 lines
4.4 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.
|
|
"""Tests for :module:'heat.engine.clients.os.cinder'."""
|
|
|
|
import uuid
|
|
|
|
import mock
|
|
|
|
from heat.common import exception
|
|
from heat.engine.clients.os import cinder
|
|
from heat.tests import common
|
|
from heat.tests import utils
|
|
|
|
|
|
class CinderClientPluginTests(common.HeatTestCase):
|
|
"""
|
|
Basic tests for the helper methods in
|
|
:module:'heat.engine.clients.os.cinder'.
|
|
"""
|
|
def setUp(self):
|
|
super(CinderClientPluginTests, self).setUp()
|
|
self.cinder_client = self.m.CreateMockAnything()
|
|
con = utils.dummy_context()
|
|
c = con.clients
|
|
self.cinder_plugin = c.client_plugin('cinder')
|
|
self.cinder_plugin._client = self.cinder_client
|
|
|
|
def test_get_volume(self):
|
|
"""Tests the get_volume function."""
|
|
volume_id = str(uuid.uuid4())
|
|
my_volume = self.m.CreateMockAnything()
|
|
self.cinder_client.volumes = self.m.CreateMockAnything()
|
|
self.cinder_client.volumes.get(
|
|
volume_id).MultipleTimes().AndReturn(my_volume)
|
|
self.m.ReplayAll()
|
|
|
|
self.assertEqual(my_volume, self.cinder_plugin.get_volume(volume_id))
|
|
|
|
self.m.VerifyAll()
|
|
|
|
def test_get_snapshot(self):
|
|
"""Tests the get_volume_snapshot function."""
|
|
snapshot_id = str(uuid.uuid4())
|
|
my_snapshot = self.m.CreateMockAnything()
|
|
self.cinder_client.volume_snapshots = self.m.CreateMockAnything()
|
|
self.cinder_client.volume_snapshots.get(
|
|
snapshot_id).MultipleTimes().AndReturn(my_snapshot)
|
|
self.m.ReplayAll()
|
|
|
|
self.assertEqual(my_snapshot,
|
|
self.cinder_plugin.get_volume_snapshot(snapshot_id))
|
|
|
|
self.m.VerifyAll()
|
|
|
|
|
|
class VolumeConstraintTest(common.HeatTestCase):
|
|
|
|
def setUp(self):
|
|
super(VolumeConstraintTest, self).setUp()
|
|
self.ctx = utils.dummy_context()
|
|
self.mock_get_volume = mock.Mock()
|
|
self.ctx.clients.client_plugin(
|
|
'cinder').get_volume = self.mock_get_volume
|
|
self.constraint = cinder.VolumeConstraint()
|
|
|
|
def test_validation(self):
|
|
self.mock_get_volume.return_value = None
|
|
self.assertTrue(self.constraint.validate("foo", self.ctx))
|
|
|
|
def test_validation_error(self):
|
|
self.mock_get_volume.side_effect = exception.VolumeNotFound(
|
|
volume='bar')
|
|
self.assertFalse(self.constraint.validate("bar", self.ctx))
|
|
|
|
|
|
class VolumeSnapshotConstraintTest(common.HeatTestCase):
|
|
|
|
def setUp(self):
|
|
super(VolumeSnapshotConstraintTest, self).setUp()
|
|
self.ctx = utils.dummy_context()
|
|
self.mock_get_snapshot = mock.Mock()
|
|
self.ctx.clients.client_plugin(
|
|
'cinder').get_volume_snapshot = self.mock_get_snapshot
|
|
self.constraint = cinder.VolumeSnapshotConstraint()
|
|
|
|
def test_validation(self):
|
|
self.mock_get_snapshot.return_value = 'snapshot'
|
|
self.assertTrue(self.constraint.validate("foo", self.ctx))
|
|
|
|
def test_validation_error(self):
|
|
self.mock_get_snapshot.side_effect = exception.VolumeSnapshotNotFound(
|
|
snapshot='bar')
|
|
self.assertFalse(self.constraint.validate("bar", self.ctx))
|
|
|
|
|
|
class VolumeTypeConstraintTest(common.HeatTestCase):
|
|
|
|
def setUp(self):
|
|
super(VolumeTypeConstraintTest, self).setUp()
|
|
self.ctx = utils.dummy_context()
|
|
self.mock_get_volume_type = mock.Mock()
|
|
self.ctx.clients.client_plugin(
|
|
'cinder').get_volume_type = self.mock_get_volume_type
|
|
self.constraint = cinder.VolumeTypeConstraint()
|
|
|
|
def test_validation(self):
|
|
self.mock_get_volume_type.return_value = 'volume_type'
|
|
self.assertTrue(self.constraint.validate("foo", self.ctx))
|
|
|
|
def test_validation_error(self):
|
|
self.mock_get_volume_type.side_effect = exception.VolumeTypeNotFound(
|
|
volume_type='bar')
|
|
self.assertFalse(self.constraint.validate("bar", self.ctx))
|