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

Fix: StreamReset in the stream handlers

Browse Source 
Since we don't catch `Exception` in the stream handlers, catch them in
the stream handlers in the tests.
This commit is contained in:
mhchia
2020-02-04 20:46:40 +08:00
parent a7ba59bf9f
commit 0548d28568
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/libp2p/test_libp2p.py
View File

@ -1,6 +1,7 @@
import multiaddr
import pytest
from libp2p.network.stream.exceptions import StreamError
from libp2p.tools.constants import MAX_READ_LEN
from libp2p.tools.factories import HostFactory
from libp2p.tools.utils import connect, create_echo_stream_handler
@ -42,13 +43,22 @@ async def test_double_response(is_host_secure):
async def double_response_stream_handler(stream):
while True:
read_string = (await stream.read(MAX_READ_LEN)).decode()
try:
read_string = (await stream.read(MAX_READ_LEN)).decode()
except StreamError:
break
response = ACK_STR_0 + read_string
await stream.write(response.encode())
try:
await stream.write(response.encode())
except StreamError:
break
response = ACK_STR_1 + read_string
await stream.write(response.encode())
try:
await stream.write(response.encode())
except StreamError:
break
hosts[1].set_stream_handler(PROTOCOL_ID_0, double_response_stream_handler)