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

Use the real get_unused_tcp_port

Browse Source 
To get rid of the fake one
This commit is contained in:
mhchia 2020-01-07 16:45:59 +08:00
parent 45eeb4fba3
commit eab59482c0
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests_interop/test_echo.py
View File

@ -1,7 +1,7 @@
import random
import re import re
from multiaddr import Multiaddr from multiaddr import Multiaddr
from p2pclient.utils import get_unused_tcp_port
import pytest import pytest
import trio import trio
@ -15,11 +15,6 @@ ECHO_PATH = GO_BIN_PATH / "echo"
ECHO_PROTOCOL_ID = TProtocol("/echo/1.0.0") ECHO_PROTOCOL_ID = TProtocol("/echo/1.0.0")
# FIXME: Change to a reasonable implementation
def unused_tcp_port_factory():
return random.randint(1024, 65535)
class EchoProcess(BaseInteractiveProcess): class EchoProcess(BaseInteractiveProcess):
port: int port: int
_peer_info: PeerInfo _peer_info: PeerInfo
@ -68,7 +63,7 @@ class EchoProcess(BaseInteractiveProcess):
@pytest.mark.trio @pytest.mark.trio
async def test_insecure_conn_py_to_go(is_host_secure): async def test_insecure_conn_py_to_go(is_host_secure):
async with HostFactory.create_batch_and_listen(is_host_secure, 1) as hosts: async with HostFactory.create_batch_and_listen(is_host_secure, 1) as hosts:
go_proc = EchoProcess(unused_tcp_port_factory(), is_host_secure) go_proc = EchoProcess(get_unused_tcp_port(), is_host_secure)
await go_proc.start() await go_proc.start()
host = hosts[0] host = hosts[0]
@ -99,6 +94,6 @@ async def test_insecure_conn_go_to_py(is_host_secure):
host.set_stream_handler(ECHO_PROTOCOL_ID, _handle_echo) host.set_stream_handler(ECHO_PROTOCOL_ID, _handle_echo)
py_maddr = host.get_addrs()[0] py_maddr = host.get_addrs()[0]
go_proc = EchoProcess(unused_tcp_port_factory(), is_host_secure, py_maddr) go_proc = EchoProcess(get_unused_tcp_port(), is_host_secure, py_maddr)
await go_proc.start() await go_proc.start()
await event_handler_finished.wait() await event_handler_finished.wait()