py-libp2p/tests/transport/test_tcp.py
2019-01-09 21:38:56 +03:00

21 lines
589 B
Python

import asyncio
import pytest
from libp2p.transport.tcp.tcp import _multiaddr_from_socket
@pytest.mark.asyncio
async def test_multiaddr_from_socket():
def handler(r, w):
pass
server = await asyncio.start_server(handler, '127.0.0.1', 8000)
assert str(_multiaddr_from_socket(server.sockets[0])) == '/ip4/127.0.0.1/tcp/8000'
server = await asyncio.start_server(handler, '127.0.0.1', 0)
addr = _multiaddr_from_socket(server.sockets[0])
assert addr.value_for_protocol('ip4') == '127.0.0.1'
port = addr.value_for_protocol('tcp')
assert int(port) > 0