goodboy
/
tractor
1
0
Fork 1
Code Issues 1 Pull Requests 4 Packages Projects Releases Wiki Activity

Add timeout to streaming test

end_of_channel_fixes
Tyler Goodlet 2021-12-15 17:18:30 -05:00
parent f2ba961e81
commit 1652716574
1 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_streaming.py
View File

@ -79,6 +79,7 @@ async def stream_from_single_subactor(
seq = range(10) seq = range(10)
with trio.fail_after(5):
async with portal.open_stream_from( async with portal.open_stream_from(
stream_func, stream_func,
sequence=list(seq), # has to be msgpack serializable sequence=list(seq), # has to be msgpack serializable
@ -100,12 +101,14 @@ async def stream_from_single_subactor(
await trio.sleep(0.3) await trio.sleep(0.3)
# ensure EOC signalled-state translates
# XXX: not really sure this is correct,
# shouldn't it be a `ClosedResourceError`?
try: try:
await stream.__anext__() await stream.__anext__()
except StopAsyncIteration: except StopAsyncIteration:
# stop all spawned subactors # stop all spawned subactors
await portal.cancel_actor() await portal.cancel_actor()
# await nursery.cancel()
@pytest.mark.parametrize( @pytest.mark.parametrize(