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)