Mplex: catch exceptions from channel.send

This commit is contained in:
mhchia
2020-02-05 17:05:30 +08:00
parent 12cb0d9ac4
commit 996b5cf15d
3 changed files with 8 additions and 9 deletions

View File

@ -31,12 +31,10 @@ async def test_mplex_conn(mplex_conn_pair):
assert stream_0.event_remote_closed.is_set()
assert stream_0.event_reset.is_set()
assert stream_0.event_local_closed.is_set()
assert conn_0.streams is None
# Test: All streams on the other side are also closed.
assert stream_1.event_remote_closed.is_set()
assert stream_1.event_reset.is_set()
assert stream_1.event_local_closed.is_set()
assert conn_1.streams is None
# Test: No effect to close more than once between two side.
await conn_0.close()

View File

@ -69,9 +69,7 @@ async def test_mplex_stream_read_after_remote_closed(mplex_stream_pair):
await stream_0.close()
assert stream_0.event_local_closed.is_set()
await trio.sleep(0.01)
# await trio.sleep(100000)
await wait_all_tasks_blocked()
# await trio.sleep_forever()
assert stream_1.event_remote_closed.is_set()
assert (await stream_1.read(MAX_READ_LEN)) == DATA
with pytest.raises(MplexStreamEOF):