Merge "Add save aggregate metadata test case"

This commit is contained in:
Zuul 2018-12-18 11:26:58 +00:00 committed by Gerrit Code Review
commit 1de1107bf7
1 changed files with 27 additions and 0 deletions

View File

@ -67,3 +67,30 @@ class TestAggregateObject(base.DbTestCase):
mock_aggregate_update.return_value = self.fake_aggregate
mock_aggregate_update.assert_called_once_with(
self.context, id, updates)
def test_save_metadata(self):
add_meta = {'add_meta': 'addval'}
del_meta = {'del_meta': 'delval'}
id = self.fake_aggregate['id']
with mock.patch.object(self.dbapi, 'aggregate_update',
autospec=True) as mock_aggregate_update:
with mock.patch.object(
self.dbapi, 'aggregate_metadata_update_or_create',
autospec=True) as mock_aggregate_upcreate:
with mock.patch.object(
self.dbapi, 'aggregate_metadata_delete',
autospec=True) as mock_aggregate_delete:
aggregate = objects.Aggregate(self.context,
**self.fake_aggregate)
aggregate._orig_metadata = del_meta
aggregate.metadata = add_meta
updates = aggregate.obj_get_changes()
del updates['metadata']
aggregate.save(self.context)
mock_aggregate_update.return_value = self.fake_aggregate
mock_aggregate_upcreate.assert_called_once_with(
self.context, id, add_meta)
mock_aggregate_delete.assert_called_once_with(
self.context, id, 'del_meta')
mock_aggregate_update.assert_called_once_with(
self.context, id, updates)