def test_finish_callback_deferred(self):
task = ImportTask(shard_count=1, shards_processed=1)
with mock.patch('django.db.models.Model.save') as save_patch:
with mock.patch('google.appengine.ext.deferred.defer') as def_patch:
task.save()
self.assertTrue(save_patch.called)
task.status = ImportStatus.IN_PROGRESS
task.save()
self.assertTrue(def_patch.called)
def test_finish_callback_deferred(self):
task = ImportTask(shard_count=1, shards_processed=1)
with mock.patch('django.db.models.Model.save') as save_patch:
with mock.patch(
'google.appengine.ext.deferred.defer') as def_patch:
task.save()
self.assertTrue(save_patch.called)
task.status = ImportStatus.IN_PROGRESS
task.save()
self.assertTrue(def_patch.called)
