def txn(): calls = memcache.incr('calls') result = ResultModel.get_by_key_name('test') if not result: result = ResultModel(key_name='test', total=0) result.total += len(keys) result.put() if memcache.get('raise') and (calls % 2): raise db.Error()
def test_run_twice(self): self.setUpContext() self.context.dispatch('ok', self.obj) self.assertEqual(1, ResultModel.get_by_key_name('test').total) self.assertEqual(2, _FantasmTaskSemaphore.all().count()) self.setUpContext(retryCount=1) # assumes retry count is set correctly self.context.dispatch('ok', self.obj) self.assertEqual(1, ResultModel.get_by_key_name('test').total) self.assertEqual(2, _FantasmTaskSemaphore.all().count())
def test(self): self.context.initialize() # queues the first task self.assertEqual(20, SimpleModel.all().count()) runQueuedTasks(speedup=False) result = ResultModel.get_by_key_name('test') self.assertEqual(20, result.total)