py-libp2p/tests/transport/test_tcp.py

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