Refactor qos spec unit test in volume v1
Refactor qos spec unit test with FakeQos class in volume v1. Change-Id: Ia76bf41a911457282b293dfaf0af311b1a9194ce
This commit is contained in:
Huanxuan Ao
@@ -305,6 +305,32 @@ class FakeQos(object):
|
||||
loaded=True)
|
||||
return qos
|
||||
|
||||
@staticmethod
|
||||
def create_one_qos_association(attrs=None):
|
||||
"""Create a fake Qos specification association.
|
||||
|
||||
:param Dictionary attrs:
|
||||
A dictionary with all attributes
|
||||
:return:
|
||||
A FakeResource object with id, name, association_type, etc.
|
||||
"""
|
||||
attrs = attrs or {}
|
||||
|
||||
# Set default attributes.
|
||||
qos_association_info = {
|
||||
"id": 'type-id-' + uuid.uuid4().hex,
|
||||
"name": 'type-name-' + uuid.uuid4().hex,
|
||||
"association_type": 'volume_type',
|
||||
}
|
||||
|
||||
# Overwrite default attributes.
|
||||
qos_association_info.update(attrs)
|
||||
|
||||
qos_association = fakes.FakeResource(
|
||||
info=copy.deepcopy(qos_association_info),
|
||||
loaded=True)
|
||||
return qos_association
|
||||
|
||||
@staticmethod
|
||||
def create_qoses(attrs=None, count=2):
|
||||
"""Create multiple fake Qos specifications.
|
||||
|
||||
@@ -13,14 +13,12 @@
|
||||
# under the License.
|
||||
#
|
||||
|
||||
import copy
|
||||
import mock
|
||||
from mock import call
|
||||
|
||||
from osc_lib import exceptions
|
||||
from osc_lib import utils
|
||||
|
||||
from openstackclient.tests.unit import fakes
|
||||
from openstackclient.tests.unit.volume.v1 import fakes as volume_fakes
|
||||
from openstackclient.volume.v1 import qos_specs
|
||||
|
||||
@@ -39,154 +37,124 @@ class TestQos(volume_fakes.TestVolumev1):
|
||||
|
||||
class TestQosAssociate(TestQos):
|
||||
|
||||
volume_type = volume_fakes.FakeType.create_one_type()
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosAssociate, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = self.qos_spec
|
||||
self.types_mock.get.return_value = self.volume_type
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.AssociateQos(self.app, None)
|
||||
|
||||
def test_qos_associate(self):
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS),
|
||||
loaded=True
|
||||
)
|
||||
self.types_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.TYPE),
|
||||
loaded=True
|
||||
)
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.type_id
|
||||
self.qos_spec.id,
|
||||
self.volume_type.id
|
||||
]
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id),
|
||||
('volume_type', volume_fakes.type_id)
|
||||
('qos_spec', self.qos_spec.id),
|
||||
('volume_type', self.volume_type.id)
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
result = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.associate.assert_called_with(
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.type_id
|
||||
self.qos_spec.id,
|
||||
self.volume_type.id
|
||||
)
|
||||
self.assertIsNone(result)
|
||||
|
||||
|
||||
class TestQosCreate(TestQos):
|
||||
|
||||
new_qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
columns = (
|
||||
'consumer',
|
||||
'id',
|
||||
'name',
|
||||
'specs'
|
||||
)
|
||||
datalist = (
|
||||
volume_fakes.qos_consumer,
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.qos_name
|
||||
new_qos_spec.consumer,
|
||||
new_qos_spec.id,
|
||||
new_qos_spec.name,
|
||||
new_qos_spec.specs
|
||||
)
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosCreate, self).setUp()
|
||||
|
||||
self.qos_mock.create.return_value = self.new_qos_spec
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.CreateQos(self.app, None)
|
||||
|
||||
def test_qos_create_without_properties(self):
|
||||
self.qos_mock.create.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS_DEFAULT_CONSUMER),
|
||||
loaded=True
|
||||
)
|
||||
|
||||
arglist = [
|
||||
volume_fakes.qos_name,
|
||||
self.new_qos_spec.name,
|
||||
]
|
||||
verifylist = [
|
||||
('name', volume_fakes.qos_name),
|
||||
('name', self.new_qos_spec.name),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
columns, data = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.create.assert_called_with(
|
||||
volume_fakes.qos_name,
|
||||
{'consumer': volume_fakes.qos_default_consumer}
|
||||
self.new_qos_spec.name,
|
||||
{'consumer': 'both'}
|
||||
)
|
||||
|
||||
self.assertEqual(self.columns, columns)
|
||||
datalist = (
|
||||
volume_fakes.qos_default_consumer,
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.qos_name
|
||||
)
|
||||
self.assertEqual(datalist, data)
|
||||
self.assertEqual(self.datalist, data)
|
||||
|
||||
def test_qos_create_with_consumer(self):
|
||||
self.qos_mock.create.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS),
|
||||
loaded=True
|
||||
)
|
||||
|
||||
arglist = [
|
||||
volume_fakes.qos_name,
|
||||
'--consumer', volume_fakes.qos_consumer
|
||||
'--consumer', self.new_qos_spec.consumer,
|
||||
self.new_qos_spec.name,
|
||||
]
|
||||
verifylist = [
|
||||
('name', volume_fakes.qos_name),
|
||||
('consumer', volume_fakes.qos_consumer)
|
||||
('consumer', self.new_qos_spec.consumer),
|
||||
('name', self.new_qos_spec.name),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
columns, data = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.create.assert_called_with(
|
||||
volume_fakes.qos_name,
|
||||
{'consumer': volume_fakes.qos_consumer}
|
||||
self.new_qos_spec.name,
|
||||
{'consumer': self.new_qos_spec.consumer}
|
||||
)
|
||||
self.assertEqual(self.columns, columns)
|
||||
self.assertEqual(self.datalist, data)
|
||||
|
||||
def test_qos_create_with_properties(self):
|
||||
self.qos_mock.create.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS_WITH_SPECS),
|
||||
loaded=True
|
||||
)
|
||||
|
||||
arglist = [
|
||||
volume_fakes.qos_name,
|
||||
'--consumer', volume_fakes.qos_consumer,
|
||||
'--consumer', self.new_qos_spec.consumer,
|
||||
'--property', 'foo=bar',
|
||||
'--property', 'iops=9001'
|
||||
'--property', 'iops=9001',
|
||||
self.new_qos_spec.name,
|
||||
]
|
||||
verifylist = [
|
||||
('name', volume_fakes.qos_name),
|
||||
('consumer', volume_fakes.qos_consumer),
|
||||
('property', volume_fakes.qos_specs)
|
||||
('consumer', self.new_qos_spec.consumer),
|
||||
('property', self.new_qos_spec.specs),
|
||||
('name', self.new_qos_spec.name),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
columns, data = self.cmd.take_action(parsed_args)
|
||||
|
||||
specs = volume_fakes.qos_specs.copy()
|
||||
specs.update({'consumer': volume_fakes.qos_consumer})
|
||||
self.new_qos_spec.specs.update(
|
||||
{'consumer': self.new_qos_spec.consumer})
|
||||
self.qos_mock.create.assert_called_with(
|
||||
volume_fakes.qos_name,
|
||||
specs
|
||||
self.new_qos_spec.name,
|
||||
self.new_qos_spec.specs
|
||||
)
|
||||
|
||||
columns = self.columns + (
|
||||
'specs',
|
||||
)
|
||||
self.assertEqual(columns, columns)
|
||||
datalist = self.datalist + (
|
||||
volume_fakes.qos_specs,
|
||||
)
|
||||
self.assertEqual(datalist, data)
|
||||
self.assertEqual(self.columns, columns)
|
||||
self.assertEqual(self.datalist, data)
|
||||
|
||||
|
||||
class TestQosDelete(TestQos):
|
||||
@@ -294,79 +262,62 @@ class TestQosDelete(TestQos):
|
||||
|
||||
class TestQosDisassociate(TestQos):
|
||||
|
||||
volume_type = volume_fakes.FakeType.create_one_type()
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosDisassociate, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = self.qos_spec
|
||||
self.types_mock.get.return_value = self.volume_type
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.DisassociateQos(self.app, None)
|
||||
|
||||
def test_qos_disassociate_with_volume_type(self):
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS),
|
||||
loaded=True
|
||||
)
|
||||
self.types_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.TYPE),
|
||||
loaded=True
|
||||
)
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
'--volume-type', volume_fakes.type_id
|
||||
'--volume-type', self.volume_type.id,
|
||||
self.qos_spec.id,
|
||||
]
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id),
|
||||
('volume_type', volume_fakes.type_id)
|
||||
('volume_type', self.volume_type.id),
|
||||
('qos_spec', self.qos_spec.id),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
result = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.disassociate.assert_called_with(
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.type_id
|
||||
self.qos_spec.id,
|
||||
self.volume_type.id
|
||||
)
|
||||
self.assertIsNone(result)
|
||||
|
||||
def test_qos_disassociate_with_all_volume_types(self):
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS),
|
||||
loaded=True
|
||||
)
|
||||
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
'--all'
|
||||
'--all',
|
||||
self.qos_spec.id,
|
||||
]
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id)
|
||||
('qos_spec', self.qos_spec.id)
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
result = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.disassociate_all.assert_called_with(volume_fakes.qos_id)
|
||||
self.qos_mock.disassociate_all.assert_called_with(self.qos_spec.id)
|
||||
self.assertIsNone(result)
|
||||
|
||||
|
||||
class TestQosList(TestQos):
|
||||
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
qos_association = volume_fakes.FakeQos.create_one_qos_association()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosList, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS_WITH_ASSOCIATIONS),
|
||||
loaded=True,
|
||||
)
|
||||
self.qos_mock.list.return_value = [self.qos_mock.get.return_value]
|
||||
self.qos_mock.get_associations.return_value = [fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.qos_association),
|
||||
loaded=True,
|
||||
)]
|
||||
self.qos_mock.list.return_value = [self.qos_spec]
|
||||
self.qos_mock.get_associations.return_value = [self.qos_association]
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.ListQos(self.app, None)
|
||||
@@ -389,81 +340,72 @@ class TestQosList(TestQos):
|
||||
)
|
||||
self.assertEqual(collist, columns)
|
||||
datalist = ((
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.qos_name,
|
||||
volume_fakes.qos_consumer,
|
||||
volume_fakes.type_name,
|
||||
utils.format_dict(volume_fakes.qos_specs),
|
||||
self.qos_spec.id,
|
||||
self.qos_spec.name,
|
||||
self.qos_spec.consumer,
|
||||
self.qos_association.name,
|
||||
utils.format_dict(self.qos_spec.specs),
|
||||
), )
|
||||
self.assertEqual(datalist, tuple(data))
|
||||
|
||||
|
||||
class TestQosSet(TestQos):
|
||||
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosSet, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = self.qos_spec
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.SetQos(self.app, None)
|
||||
|
||||
def test_qos_set_with_properties_with_id(self):
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS_WITH_SPECS),
|
||||
loaded=True
|
||||
)
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
'--property', 'foo=bar',
|
||||
'--property', 'iops=9001'
|
||||
'--property', 'iops=9001',
|
||||
self.qos_spec.id,
|
||||
]
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id),
|
||||
('property', volume_fakes.qos_specs)
|
||||
('property', self.qos_spec.specs),
|
||||
('qos_spec', self.qos_spec.id),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
result = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.set_keys.assert_called_with(
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.qos_specs
|
||||
self.qos_spec.id,
|
||||
self.qos_spec.specs
|
||||
)
|
||||
self.assertIsNone(result)
|
||||
|
||||
|
||||
class TestQosShow(TestQos):
|
||||
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
qos_association = volume_fakes.FakeQos.create_one_qos_association()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosShow, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS_WITH_ASSOCIATIONS),
|
||||
loaded=True,
|
||||
)
|
||||
self.qos_mock.get_associations.return_value = [fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.qos_association),
|
||||
loaded=True,
|
||||
)]
|
||||
|
||||
self.qos_mock.get.return_value = self.qos_spec
|
||||
self.qos_mock.get_associations.return_value = [self.qos_association]
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.ShowQos(self.app, None)
|
||||
|
||||
def test_qos_show(self):
|
||||
arglist = [
|
||||
volume_fakes.qos_id
|
||||
self.qos_spec.id
|
||||
]
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id)
|
||||
('qos_spec', self.qos_spec.id)
|
||||
]
|
||||
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
columns, data = self.cmd.take_action(parsed_args)
|
||||
self.qos_mock.get.assert_called_with(
|
||||
volume_fakes.qos_id
|
||||
self.qos_spec.id
|
||||
)
|
||||
|
||||
collist = (
|
||||
@@ -475,56 +417,53 @@ class TestQosShow(TestQos):
|
||||
)
|
||||
self.assertEqual(collist, columns)
|
||||
datalist = (
|
||||
volume_fakes.type_name,
|
||||
volume_fakes.qos_consumer,
|
||||
volume_fakes.qos_id,
|
||||
volume_fakes.qos_name,
|
||||
utils.format_dict(volume_fakes.qos_specs),
|
||||
self.qos_association.name,
|
||||
self.qos_spec.consumer,
|
||||
self.qos_spec.id,
|
||||
self.qos_spec.name,
|
||||
utils.format_dict(self.qos_spec.specs),
|
||||
)
|
||||
self.assertEqual(datalist, tuple(data))
|
||||
|
||||
|
||||
class TestQosUnset(TestQos):
|
||||
|
||||
qos_spec = volume_fakes.FakeQos.create_one_qos()
|
||||
|
||||
def setUp(self):
|
||||
super(TestQosUnset, self).setUp()
|
||||
|
||||
self.qos_mock.get.return_value = self.qos_spec
|
||||
# Get the command object to test
|
||||
self.cmd = qos_specs.UnsetQos(self.app, None)
|
||||
|
||||
def test_qos_unset_with_properties(self):
|
||||
self.qos_mock.get.return_value = fakes.FakeResource(
|
||||
None,
|
||||
copy.deepcopy(volume_fakes.QOS),
|
||||
loaded=True
|
||||
)
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
'--property', 'iops',
|
||||
'--property', 'foo'
|
||||
'--property', 'foo',
|
||||
self.qos_spec.id,
|
||||
]
|
||||
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id),
|
||||
('property', ['iops', 'foo'])
|
||||
('property', ['iops', 'foo']),
|
||||
('qos_spec', self.qos_spec.id),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
result = self.cmd.take_action(parsed_args)
|
||||
|
||||
self.qos_mock.unset_keys.assert_called_with(
|
||||
volume_fakes.qos_id,
|
||||
self.qos_spec.id,
|
||||
['iops', 'foo']
|
||||
)
|
||||
self.assertIsNone(result)
|
||||
|
||||
def test_qos_unset_nothing(self):
|
||||
arglist = [
|
||||
volume_fakes.qos_id,
|
||||
self.qos_spec.id,
|
||||
]
|
||||
|
||||
verifylist = [
|
||||
('qos_spec', volume_fakes.qos_id),
|
||||
('qos_spec', self.qos_spec.id),
|
||||
]
|
||||
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user