forked from goodboy/tractor
1
0
Fork 0

Adjust cancelled test to expect raised overrun error

expected_ctx_cancelled
Tyler Goodlet 2021-12-15 17:35:28 -05:00
parent 98a830ccba
commit 916e27eedc
1 changed files with 3 additions and 17 deletions

View File

@ -608,33 +608,19 @@ def test_one_end_stream_not_opened(overrun_by):
# 2 overrun cases and the no overrun case (which pushes right up to # 2 overrun cases and the no overrun case (which pushes right up to
# the msg limit) # the msg limit)
if overrunner == 'caller': if overrunner == 'caller' or 'cance' in overrunner:
with pytest.raises(tractor.RemoteActorError) as excinfo: with pytest.raises(tractor.RemoteActorError) as excinfo:
trio.run(main) trio.run(main)
assert excinfo.value.type == 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 # TODO: embedded remote errors so that we can verify the source
# error? # error? the callee delivers an error which is an overrun
# the callee delivers an error which is an overrun wrapped # wrapped in a remote actor error.
# in a remote actor error.
assert excinfo.value.type == tractor.RemoteActorError assert excinfo.value.type == tractor.RemoteActorError
else: else: