varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for MplexStream

Browse Source
This commit is contained in:
mhchia
2019-09-18 21:51:09 +08:00
parent 02c55e5d14
commit 313ae45b45
4 changed files with 212 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/network/test_net_stream.py
View File

@ -53,11 +53,9 @@ async def test_net_stream_read_until_eof(net_stream_pair):
@pytest.mark.asyncio
async def test_net_stream_read_after_remote_closed(net_stream_pair):
stream_0, stream_1 = net_stream_pair
assert not stream_1.muxed_stream.event_remote_closed.is_set()
await stream_0.write(DATA)
await stream_0.close()
await asyncio.sleep(0.01)
assert stream_1.muxed_stream.event_remote_closed.is_set()
assert (await stream_1.read(MAX_READ_LEN)) == DATA
with pytest.raises(StreamEOF):
await stream_1.read(MAX_READ_LEN)