def test_one_tier_coro(self):
cell = IOCell()
@cell.tier()
def f(route):
yield from route.emit(123)
self.assertEqual(list(cell), [123])
def test_blowup(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
raise RuntimeError()
cell.add_tier(f)
self.assertRaises(RuntimeError, list, cell)
def test_one_tier_append(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(123)
cell.append_tier(f)
self.assertEqual(list(cell), [123])
def test_from_cell_generator(self):
inner = IOCell()
@inner.tier()
def inner_tier(route):
for i in range(3):
yield from route.emit(i)
outer = IOCell()
@outer.tier()
def outer_tier(route):
yield from asyncio.sleep(0)
inner_it = iter(inner)
for i in range(3):
self.assertEqual(i, next(inner_it))
list(outer)
def test_uncalled_tier_coro_decor(self):
cell = IOCell()
def setup():
@cell.tier
def fn(route):
pass
self.assertRaises(TypeError, setup)
def test_buffer_1tier_0rem(self):
cell = IOCell()
@cell.tier(buffer=2)
def f(route):
for i in range(4):
yield from route.emit(i)
self.assertEqual(list(cell), list(range(4)))
def test_yield_from(self):
cell = IOCell()
@asyncio.coroutine
def coro(route):
yield from route.emit((yield from self.add(2, 3)))
cell.add_tier(coro)
self.assertEqual(list(cell), [5])
def test_varargs(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(*'abc')
yield from route.emit(*'def')
cell.add_tier(f)
self.assertEqual(list(cell), list('abcdef'))
def test_multi_emitter_1level(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(123)
yield from route.emit(321)
cell.add_tier(f)
self.assertEqual(list(cell), [123, 321])
def test_coro_noappend(self):
cell = IOCell()
@cell.tier()
def a1(route):
yield from route.emit(100)
@cell.tier(append=False)
def a2(route):
yield from route.emit(200)
self.assertEqual(list(cell), [100, 200])
def test_buffer_1tier_2rem_varargs(self):
cell = IOCell()
@cell.tier(buffer=4)
def f(route):
for i in range(10):
yield from route.emit(i, str(i))
expect = []
for i in range(10):
expect.extend((i, str(i)))
self.assertEqual(list(cell), expect)
def test_two_tier_coro(self):
cell = IOCell()
@cell.tier()
def f(route):
yield from route.emit(100)
@cell.tier()
def f2(route, i):
yield from route.emit(i + 50)
self.assertEqual(list(cell), [150])
def test_one_tier_no_emit(self):
cell = IOCell()
refcnt = 0
@cell.tier()
@asyncio.coroutine
def f(route):
nonlocal refcnt
refcnt += 1
self.assertFalse(list(cell))
self.assertEqual(refcnt, 1)
def wrap():
cell = IOCell()
class Thing(object):
pass
thing = Thing()
@cell.tier()
def tier(r):
yield from r.emit(thing)
return cell
def test_background_work_exception(self):
cell = IOCell()
@asyncio.coroutine
def bg():
raise Exception()
@cell.tier()
def coro(route):
cell.loop.create_task(bg())
yield # allow bg to run.
self.assertRaises(Exception, list, cell)
def test_cascaded_tiers(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(123)
@asyncio.coroutine
def f2(route, foo):
self.assertEqual(foo, 123)
t1 = cell.add_tier(f)
cell.add_tier(f2, source=[t1])
self.assertFalse(list(cell))
def test_single_final(self):
cell = IOCell()
refcnt = 0
@asyncio.coroutine
def f(route):
nonlocal refcnt
refcnt += 1
yield from route.emit(123)
cell.add_tier(f)
results = list(cell)
self.assertEqual(refcnt, 1)
self.assertEqual(results, [123])
def test_trailing_work(self):
cell = IOCell()
fullrun = False
@cell.tier()
def coro(route):
nonlocal fullrun
for i in range(3):
yield from route.emit(i)
for ii in range(3):
yield from asyncio.sleep(0)
fullrun = True
self.assertEqual(list(cell), list(range(3)))
self.assertTrue(fullrun)
def test_varargs(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(123, 345)
@asyncio.coroutine
def f2(route, foo, bar):
self.assertEqual(foo, 123)
self.assertEqual(bar, 345)
t1 = cell.add_tier(f)
cell.add_tier(f2, source=[t1])
self.assertFalse(list(cell))
def test_cascaded_tiers_no_emit_tier_coro_deco(self):
cell = IOCell()
refcnt = 0
@cell.tier()
def f(route):
nonlocal refcnt
refcnt += 1
@cell.tier(source=f)
def f2(route):
nonlocal refcnt
refcnt += 1
self.assertFalse(list(cell))
self.assertEqual(refcnt, 1)
def test_coro_source_override(self):
cell = IOCell()
@cell.tier()
def a1(route):
yield from route.emit(100)
@cell.tier(append=False)
def a2(route):
yield from route.emit(200)
@cell.tier(source=a1)
def b1(route, i):
yield from route.emit(i + 1)
self.assertEqual(list(cell), [200, 101])
def test_source_from_coroutine_sequence(self):
cell = IOCell()
refcnt = 0
@cell.tier()
def f(route):
nonlocal refcnt
refcnt += 1
@cell.tier(source=[f])
def f2(route):
nonlocal refcnt
refcnt += 1
self.assertFalse(list(cell))
self.assertEqual(refcnt, 1)
def test_multi_emitter_2level(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
yield from route.emit(123)
yield from route.emit(321)
t = cell.add_tier(f)
@asyncio.coroutine
def f2(route, number):
yield from route.emit(-number)
yield from route.emit(number + 1)
cell.add_tier(f2, source=t)
self.assertEqual(list(cell), [-123, 124, -321, 322])
def test_multi_blowup(self):
cell = IOCell()
@asyncio.coroutine
def f(route):
raise RuntimeError()
cell.add_tier(f)
@asyncio.coroutine
def f2(route):
raise ValueError()
cell.add_tier(f2)
it = iter(cell)
self.assertRaises(RuntimeError, next, it)
self.assertRaises(StopIteration, next, it) # ValueError is dropped.
def test_background_work_clean_exit(self):
cell = IOCell()
bg_done = False
@asyncio.coroutine
def bg():
nonlocal bg_done
for x in range(100):
yield
bg_done = True
@cell.tier()
def coro(route):
route.cell.loop.create_task(bg())
self.assertEqual(list(cell), [])
self.assertTrue(bg_done)
def test_cascaded_tiers_no_emit(self):
cell = IOCell()
refcnt = 0
@asyncio.coroutine
def f(route):
nonlocal refcnt
refcnt += 1
@asyncio.coroutine
def f2(route):
nonlocal refcnt
refcnt += 1
t1 = cell.add_tier(f)
cell.add_tier(f2, source=t1)
self.assertFalse(list(cell))
self.assertEqual(refcnt, 1)
def wrap():
cell = IOCell()
class Thing(object):
pass
thing1 = Thing()
thing1.thing2 = Thing()
thing1.thing2.thing1 = thing1
@cell.tier()
def tier1(r):
thing1.hello = 'world'
yield from r.emit(thing1)
@cell.tier(source=tier1)
def tier2(r, thing):
yield from r.emit(thing)
return cell
def test_enter_exit_bookends(self):
states = {}
class Coord(AbstractCellCoordinator):
@asyncio.coroutine
def start(self, tier):
states['enter'] = time.time()
@asyncio.coroutine
def finish(self, tier):
states['exit'] = time.time()
cell = IOCell(coord=Coord(), debug=True)
@cell.tier()
def t(route):
states['proc'] = time.time()
list(cell)
self.assertSetEqual(set(states), {'enter', 'proc', 'exit'})
self.assertTrue(states['enter'] < states['proc'] < states['exit'])
def test_background_work_gc(self):
cell = IOCell()
loop = cell.loop
cancel = mock.Mock()
@asyncio.coroutine
def bg():
yield from asyncio.sleep(1 << 20)
@cell.tier()
def coro(route):
task = loop.create_task(bg())
task.cancel = cancel
for i in range(10):
yield from route.emit(i)
l = iter(cell)
next(l)
self.assertFalse(cancel.called)
del l
self.assertTrue(cancel.called)
def test_buffer_2tier_0rem(self):
cell = IOCell()
cnt = 0
@cell.tier(buffer=2)
def f(route):
yield from route.emit(1)
yield from route.emit(2)
yield from route.emit(1)
yield from route.emit(2)
@cell.tier(source=f)
def f2(route, a, b):
nonlocal cnt
cnt += 1
self.assertEqual(a, 1)
self.assertEqual(b, 2)
yield from route.emit(a)
yield from route.emit(b)
self.assertEqual(list(cell), [1, 2, 1, 2])
self.assertEqual(cnt, 2)