StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add additional sleep

Browse Source 
To be more confident that the stream handler is registered in the
daemon.
This commit is contained in:
mhchia 2019-09-24 13:28:14 +08:00
parent 1bfc6b41e4
commit e217acf4ac
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests_interop/conftest.py
View File

@ -165,6 +165,8 @@ async def py_to_daemon_stream_pair(hosts, p2pds, is_to_fail_daemon_stream):
event_stream_handled.set() event_stream_handled.set()
await p2pd.control.stream_handler(protocol_id, daemon_stream_handler) await p2pd.control.stream_handler(protocol_id, daemon_stream_handler)
# Sleep for a while to wait for the handler being registered.
await asyncio.sleep(0.01)
if is_to_fail_daemon_stream: if is_to_fail_daemon_stream:
# FIXME: This is a workaround to make daemon reset the stream. # FIXME: This is a workaround to make daemon reset the stream.
@ -180,5 +182,5 @@ async def py_to_daemon_stream_pair(hosts, p2pds, is_to_fail_daemon_stream):
stream_py = await host.new_stream(p2pd.peer_id, [protocol_id]) stream_py = await host.new_stream(p2pd.peer_id, [protocol_id])
if not is_to_fail_daemon_stream: if not is_to_fail_daemon_stream:
await event_stream_handled.wait() await event_stream_handled.wait()
# NOTE: If `is_to_fail_daemon_stream == True`, `stream_daemon == None`. # NOTE: If `is_to_fail_daemon_stream == True`, then `stream_daemon == None`.
yield stream_py, stream_daemon yield stream_py, stream_daemon