def testA(self):
messages = []
@ndb.tasklet
def work(val):
messages.append(val)
raise ndb.Return(val)
fut = FutureFuture(lambda: work('A'))
fut.start()
assert fut.get_result() == 'A'
def testA(self):
messages = []
@ndb.tasklet
def work(val):
messages.append(val)
raise ndb.Return(val)
fut = FutureFuture(lambda: work('A'))
fut.start()
assert fut.get_result() == 'A'
def testDeferredWorker(self):
messages = []
@ndb.tasklet
def work(val):
messages.append(val)
raise ndb.Return(val)
fut = FutureFuture(lambda: work('A'))
with pytest.raises(RuntimeError):
fut.get_result()
assert messages == []
def testDeferredWorker(self):
messages = []
@ndb.tasklet
def work(val):
messages.append(val)
raise ndb.Return(val)
fut = FutureFuture(lambda: work('A'))
with pytest.raises(RuntimeError):
fut.get_result()
assert messages == []
def testEnsure(self):
futures = map(ndb.Future, range(5))
messages = []
@ndb.tasklet
def work(val):
print "Enter %s" % val
val = yield futures[val]
messages.append(val)
print "Wake up %s" % val
raise ndb.Return(val)
fut = DeferredFuturesQueue(3)
dependents = map(lambda i: FutureFuture(lambda: work(i)), range(5))
map(fut.add_dependent, dependents)
fut.complete()
for i in [2, 0, 4, 3, 1]:
futures[i].set_result(i)
run_loop()
assert fut.done()
assert fut.get_result() == [0, 1, 2, 3, 4]
assert messages == [2, 0, 4, 3, 1]
