def test_blowup(self): cell = IOCell() @asyncio.coroutine def f(route): raise RuntimeError() cell.add_tier(f) self.assertRaises(RuntimeError, list, cell)
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_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_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_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_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_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_one_tier_no_emit(self): cell = IOCell() refcnt = 0 @asyncio.coroutine def f(route): nonlocal refcnt refcnt += 1 cell.add_tier(f) self.assertFalse(list(cell)) self.assertEqual(refcnt, 1)
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_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_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_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_one_tier_no_emit(self): cell = IOCell() refcnt = 0 @asyncio.coroutine def f(route): nonlocal refcnt refcnt += 1 cell.add_tier(f) 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_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_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_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 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_multi_emitter_multi_source(self): cell = IOCell() @asyncio.coroutine def a1(route): yield from route.emit('a1-1') yield from route.emit('a1-2') a1t = cell.add_tier(a1) @asyncio.coroutine def a2(route): yield from route.emit('a2-1') yield from route.emit('a2-2') a2t = cell.add_tier(a2) @asyncio.coroutine def b(route, value): yield from route.emit(value) cell.add_tier(b, source=[a1t, a2t]) self.assertEqual(list(cell), ['a1-1', 'a2-1', 'a1-2', 'a2-2'])
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_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_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 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_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_uncalled_tier_coro_decor(self): cell = IOCell() def setup(): @cell.tier def fn(route): pass self.assertRaises(TypeError, setup)
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_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_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 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_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_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_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_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_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 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)
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.