async def test_stream_happy():
xs = AsyncSingleStream()
sink = AnonymousAsyncObserver()
await subscribe(xs, sink)
await xs.asend_later(1, 10)
await xs.asend_later(1, 20)
await xs.asend_later(1, 30)
assert sink.values == [(1, 10), (2, 20), (3, 30)]
async def test_stream_cancel_context():
xs = AsyncSingleStream()
sink = AnonymousAsyncObserver()
with await subscribe(xs, sink):
pass
await xs.asend_later(1, 10)
await xs.asend_later(1, 20)
assert sink.values == []
async def test_stream_send_after_close():
xs = AsyncSingleStream()
sink = AnonymousAsyncObserver()
await subscribe(xs, sink)
await xs.asend_later(1, 10)
await xs.asend_later(1, 20)
await xs.asend_later(1, 30)
await xs.aclose_later(2)
await xs.asend_later(1, 40)
assert sink.values == [(1, 10), (2, 20), (3, 30), (5, )]
async def test_stream_cold_close():
xs = AsyncSingleStream()
sink = AnonymousAsyncObserver()
async def aclose():
await xs.aclose()
asyncio.ensure_future(aclose())
await asyncio.sleep(10)
with await subscribe(xs, sink):
await xs.asend_later(1, 20)
assert sink.values == [(10, )]
async def test_stream_throws():
ex = MyException("ex")
xs = AsyncSingleStream()
sink = AnonymousAsyncObserver()
with pytest.raises(MyException):
sub = await subscribe(xs, sink)
await xs.asend_later(1, 10)
await xs.asend_later(1, 20)
await xs.asend_later(1, 30)
await xs.athrow_later(1, ex)
await xs.asend_later(1, 40)
await sub
assert sink.values == [(1, 10), (2, 20), (3, 30), (4, ex)]
async def test_stream_cold_throw():
xs = AsyncSingleStream()
sink = AsyncAnonymousObserver()
async def athrow():
await xs.athrow(MyException)
asyncio.ensure_future(athrow())
await asyncio.sleep(10)
async with subscribe(xs, sink):
await xs.asend_later(1, 20)
assert sink.values == [(10, MyException)]
async def test_stream_cold_send():
xs = AsyncSingleStream()
sink = AsyncAnonymousObserver()
async def asend(value):
await xs.asend(value)
asyncio.ensure_future(asend(42))
await asyncio.sleep(10)
async with subscribe(xs, sink):
await xs.asend_later(1, 20)
assert sink.values == [(10, 42), (11, 20)]
async def test_stream_cancel():
xs = AsyncSingleStream()
sub = None
def mapper(value):
return value * 10
ys = map(mapper, xs)
sink = AnonymousAsyncObserver()
sub = await subscribe(ys, sink)
await xs.asend_later(1, 10)
sub.cancel()
await xs.asend_later(1, 20)
assert sink.values == [(1, 100)]
async def test_stream_cancel_asend():
xs = AsyncSingleStream()
sub = None
async def asend(value):
sub.cancel()
await asyncio.sleep(0)
def mapper(value):
return value * 10
ys = map(mapper, xs)
sink = AnonymousAsyncObserver(asend)
async with subscribe(ys, sink) as sub:
await xs.asend_later(1, 10)
await xs.asend_later(1, 20)
assert sink.values == [(1, 100)]