openstack/nova
Code Issues Proposed changes

Merge "Add missing key attribute to AggregateList.get_by_host()"

Browse Source
This commit is contained in:
Jenkins 2013-10-15 01:22:23 +00:00 committed by Gerrit Code Review
commit fd73c50534
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nova/objects/aggregate.py
View File

@ -145,6 +145,10 @@ class Aggregate(base.NovaPersistentObject, base.NovaObject):
class AggregateList(base.ObjectListBase, base.NovaObject):
# Version 1.0: Initial version
# Version 1.1: Added key argument to get_by_host()
VERSION = '1.1'
@base.remotable_classmethod
def get_all(cls, context):
db_aggregates = db.aggregate_get_all(context)
@ -152,7 +156,7 @@ class AggregateList(base.ObjectListBase, base.NovaObject):
db_aggregates)
@base.remotable_classmethod
def get_by_host(cls, context, host):
db_aggregates = db.aggregate_get_by_host(context, host)
def get_by_host(cls, context, host, key=None):
db_aggregates = db.aggregate_get_by_host(context, host, key=key)
return base.obj_make_list(context, AggregateList(), Aggregate,
db_aggregates)

2
nova/tests/objects/test_aggregate.py
View File

@ -133,7 +133,7 @@ class _TestAggregateObject(object):
def test_by_host(self):
self.mox.StubOutWithMock(db, 'aggregate_get_by_host')
db.aggregate_get_by_host(self.context, 'fake-host'
db.aggregate_get_by_host(self.context, 'fake-host', key=None,
).AndReturn([fake_aggregate])
self.mox.ReplayAll()
aggs = aggregate.AggregateList.get_by_host(self.context, 'fake-host')