jc211
/
tractor
forked from goodboy/tractor
1
0
Fork 0
Code Pull Requests Activity

Make cancel case expect multi-error

stricter_context_starting
Tyler Goodlet 2021-12-06 16:00:21 -05:00
parent c9132de7dc
commit a79cdc7b44
1 changed files with 28 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/test_context_streams.py
View File

@ -10,6 +10,7 @@ from itertools import count
import pytest import pytest
import trio import trio
import tractor import tractor
from tractor._exceptions import StreamOverrun
@tractor.context @tractor.context
@ -74,11 +75,11 @@ async def keep_sending_from_callee(
@pytest.mark.parametrize( @pytest.mark.parametrize(
'overrun_by', 'overrun_by',
[ [
(None, 0, never_open_stream), # use default settings
('caller', 1, never_open_stream), ('caller', 1, never_open_stream),
('cancel_caller_during_overrun', 1, never_open_stream),
('callee', 0, keep_sending_from_callee), ('callee', 0, keep_sending_from_callee),
], ],
ids='overrun_condition_by={}'.format, ids='overrun_condition={}'.format,
) )
def test_one_end_stream_not_opened(overrun_by): def test_one_end_stream_not_opened(overrun_by):
''' '''
@ -102,18 +103,21 @@ def test_one_end_stream_not_opened(overrun_by):
) as (ctx, sent): ) as (ctx, sent):
assert sent is None assert sent is None
if overrunner in (None, 'caller'): if 'caller' in overrunner:
async with ctx.open_stream() as stream: async with ctx.open_stream() as stream:
for i in range(buf_size - 1): for i in range(buf_size):
print(f'sending {i}')
await stream.send(i) await stream.send(i)
if overrunner is None: if 'cancel' in overrunner:
# without this we block waiting on the child side # without this we block waiting on the child side
await ctx.cancel() await ctx.cancel()
else: else:
await stream.send('yo') # expect overrun error to be relayed back
# and this sleep interrupted
await trio.sleep_forever()
else: else:
# callee overruns caller case so we do nothing here # callee overruns caller case so we do nothing here
@ -127,12 +131,29 @@ def test_one_end_stream_not_opened(overrun_by):
with pytest.raises(tractor.RemoteActorError) as excinfo: with pytest.raises(tractor.RemoteActorError) as excinfo:
trio.run(main) trio.run(main)
assert excinfo.value.type == tractor._exceptions.StreamOverrun assert excinfo.value.type == StreamOverrun
elif 'cancel' in overrunner:
with pytest.raises(trio.MultiError) as excinfo:
trio.run(main)
multierr = excinfo.value
for exc in multierr.exceptions:
etype = type(exc)
if etype == tractor.RemoteActorError:
assert exc.type == StreamOverrun
else:
assert etype == tractor.ContextCancelled
elif overrunner == 'callee': elif overrunner == 'callee':
with pytest.raises(tractor.RemoteActorError) as excinfo: with pytest.raises(tractor.RemoteActorError) as excinfo:
trio.run(main) trio.run(main)
# TODO: embedded remote errors so that we can verify the source
# error?
# the callee delivers an error which is an overrun wrapped
# in a remote actor error.
assert excinfo.value.type == tractor.RemoteActorError assert excinfo.value.type == tractor.RemoteActorError
else: else: