Accepting tuple from get_or_create in _process_delete

This commit is contained in:
Andrew Melton 2013-05-29 10:54:27 -04:00
parent f601dff52f
commit b88dfffd94
2 changed files with 3 additions and 3 deletions

View File

@ -288,7 +288,7 @@ def _process_delete(raw, body):
'instance': instance_id,
'deleted_at': deleted_at,
}
delete = STACKDB.get_or_create_instance_delete(**values)
(delete, new) = STACKDB.get_or_create_instance_delete(**values)
delete.raw = raw
launched_at = payload.get('launched_at')

View File

@ -605,7 +605,7 @@ class StacktachUsageParsingTestCase(unittest.TestCase):
delete.deleted_at = delete_decimal
views.STACKDB.get_or_create_instance_delete(instance=INSTANCE_ID_1,
deleted_at=delete_decimal)\
.AndReturn(delete)
.AndReturn((delete, True))
views.STACKDB.save(delete)
self.mox.ReplayAll()
@ -629,7 +629,7 @@ class StacktachUsageParsingTestCase(unittest.TestCase):
delete.deleted_at = delete_decimal
views.STACKDB.get_or_create_instance_delete(instance=INSTANCE_ID_1,
deleted_at=delete_decimal)\
.AndReturn(delete)
.AndReturn((delete, True))
views.STACKDB.save(delete)
self.mox.ReplayAll()