tests: Reorder compute fakes
This will make future changes slightly less complex. Change-Id: Ia69fd9d05796cce5ed48a8d12902cb54caec9779 Signed-off-by: Stephen Finucane <sfinucan@redhat.com>
This commit is contained in:
parent
3c9afe69bb
commit
347b444feb
@ -75,77 +75,6 @@ QUOTA_columns = tuple(sorted(QUOTA))
|
|||||||
QUOTA_data = tuple(QUOTA[x] for x in sorted(QUOTA))
|
QUOTA_data = tuple(QUOTA[x] for x in sorted(QUOTA))
|
||||||
|
|
||||||
|
|
||||||
class FakeAggregate(object):
|
|
||||||
"""Fake one aggregate."""
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def create_one_aggregate(attrs=None):
|
|
||||||
"""Create a fake aggregate.
|
|
||||||
|
|
||||||
:param dict attrs:
|
|
||||||
A dictionary with all attributes
|
|
||||||
:return:
|
|
||||||
A FakeResource object, with id and other attributes
|
|
||||||
"""
|
|
||||||
attrs = attrs or {}
|
|
||||||
|
|
||||||
# Set default attribute
|
|
||||||
aggregate_info = {
|
|
||||||
"name": "aggregate-name-" + uuid.uuid4().hex,
|
|
||||||
"availability_zone": "ag_zone",
|
|
||||||
"hosts": [],
|
|
||||||
"id": "aggregate-id-" + uuid.uuid4().hex,
|
|
||||||
"metadata": {
|
|
||||||
"availability_zone": "ag_zone",
|
|
||||||
"key1": "value1",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
# Overwrite default attributes.
|
|
||||||
aggregate_info.update(attrs)
|
|
||||||
|
|
||||||
aggregate = fakes.FakeResource(
|
|
||||||
info=copy.deepcopy(aggregate_info), loaded=True
|
|
||||||
)
|
|
||||||
return aggregate
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def create_aggregates(attrs=None, count=2):
|
|
||||||
"""Create multiple fake aggregates.
|
|
||||||
|
|
||||||
:param dict attrs:
|
|
||||||
A dictionary with all attributes
|
|
||||||
:param int count:
|
|
||||||
The number of aggregates to fake
|
|
||||||
:return:
|
|
||||||
A list of FakeResource objects faking the aggregates
|
|
||||||
"""
|
|
||||||
aggregates = []
|
|
||||||
for i in range(0, count):
|
|
||||||
aggregates.append(FakeAggregate.create_one_aggregate(attrs))
|
|
||||||
|
|
||||||
return aggregates
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def get_aggregates(aggregates=None, count=2):
|
|
||||||
"""Get an iterable MagicMock object with a list of faked aggregates.
|
|
||||||
|
|
||||||
If aggregates list is provided, then initialize the Mock object
|
|
||||||
with the list. Otherwise create one.
|
|
||||||
|
|
||||||
:param List aggregates:
|
|
||||||
A list of FakeResource objects faking aggregates
|
|
||||||
:param int count:
|
|
||||||
The number of aggregates to fake
|
|
||||||
:return:
|
|
||||||
An iterable Mock object with side_effect set to a list of faked
|
|
||||||
aggregates
|
|
||||||
"""
|
|
||||||
if aggregates is None:
|
|
||||||
aggregates = FakeAggregate.create_aggregates(count)
|
|
||||||
return mock.Mock(side_effect=aggregates)
|
|
||||||
|
|
||||||
|
|
||||||
class FakeComputev2Client(object):
|
class FakeComputev2Client(object):
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
self.agents = mock.Mock()
|
self.agents = mock.Mock()
|
||||||
@ -255,6 +184,77 @@ class TestComputev2(utils.TestCommand):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class FakeAggregate(object):
|
||||||
|
"""Fake one aggregate."""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def create_one_aggregate(attrs=None):
|
||||||
|
"""Create a fake aggregate.
|
||||||
|
|
||||||
|
:param dict attrs:
|
||||||
|
A dictionary with all attributes
|
||||||
|
:return:
|
||||||
|
A FakeResource object, with id and other attributes
|
||||||
|
"""
|
||||||
|
attrs = attrs or {}
|
||||||
|
|
||||||
|
# Set default attribute
|
||||||
|
aggregate_info = {
|
||||||
|
"name": "aggregate-name-" + uuid.uuid4().hex,
|
||||||
|
"availability_zone": "ag_zone",
|
||||||
|
"hosts": [],
|
||||||
|
"id": "aggregate-id-" + uuid.uuid4().hex,
|
||||||
|
"metadata": {
|
||||||
|
"availability_zone": "ag_zone",
|
||||||
|
"key1": "value1",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
# Overwrite default attributes.
|
||||||
|
aggregate_info.update(attrs)
|
||||||
|
|
||||||
|
aggregate = fakes.FakeResource(
|
||||||
|
info=copy.deepcopy(aggregate_info), loaded=True
|
||||||
|
)
|
||||||
|
return aggregate
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def create_aggregates(attrs=None, count=2):
|
||||||
|
"""Create multiple fake aggregates.
|
||||||
|
|
||||||
|
:param dict attrs:
|
||||||
|
A dictionary with all attributes
|
||||||
|
:param int count:
|
||||||
|
The number of aggregates to fake
|
||||||
|
:return:
|
||||||
|
A list of FakeResource objects faking the aggregates
|
||||||
|
"""
|
||||||
|
aggregates = []
|
||||||
|
for i in range(0, count):
|
||||||
|
aggregates.append(FakeAggregate.create_one_aggregate(attrs))
|
||||||
|
|
||||||
|
return aggregates
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def get_aggregates(aggregates=None, count=2):
|
||||||
|
"""Get an iterable MagicMock object with a list of faked aggregates.
|
||||||
|
|
||||||
|
If aggregates list is provided, then initialize the Mock object
|
||||||
|
with the list. Otherwise create one.
|
||||||
|
|
||||||
|
:param List aggregates:
|
||||||
|
A list of FakeResource objects faking aggregates
|
||||||
|
:param int count:
|
||||||
|
The number of aggregates to fake
|
||||||
|
:return:
|
||||||
|
An iterable Mock object with side_effect set to a list of faked
|
||||||
|
aggregates
|
||||||
|
"""
|
||||||
|
if aggregates is None:
|
||||||
|
aggregates = FakeAggregate.create_aggregates(count)
|
||||||
|
return mock.Mock(side_effect=aggregates)
|
||||||
|
|
||||||
|
|
||||||
class FakeAgent(object):
|
class FakeAgent(object):
|
||||||
"""Fake one or more agent."""
|
"""Fake one or more agent."""
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user