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

Don't cancel the context on overrun cases

stricter_context_starting
Tyler Goodlet 2021-12-06 11:53:15 -05:00
parent 318027ebd1
commit 142083d81b
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_context_streams.py
View File

@ -97,22 +97,28 @@ def test_one_end_stream_not_opened(overrun_by):
enable_modules=[__name__],
)
async with portal.open_context(entrypoint) as (ctx, sent):
async with portal.open_context(
entrypoint,
) as (ctx, sent):
assert sent is None
if overrunner in (None, 'caller'):
async with ctx.open_stream() as stream:
for i in range(buf_size - 1):
await stream.send(i)
if overrunner is None:
# without this we block waiting on the child side
await ctx.cancel()
else:
await stream.send('yo')
else:
# callee overruns caller case so we do nothing here
await trio.sleep_forever()
# without this we block waiting on the child side
await ctx.cancel()
await portal.cancel_actor()
# 2 overrun cases and the no overrun case (which pushes right up to