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_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_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_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_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_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_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_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_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_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(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_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_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_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_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'])