25 lines
840 B
Python
25 lines
840 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from libp2p.identity.identify.pb.identify_pb2 import Identify
|
||
|
|
from libp2p.identity.identify.protocol import ID, _mk_identify_protobuf
|
||
|
|
from libp2p.peer.peerinfo import info_from_p2p_addr
|
||
|
|
from tests.utils import set_up_nodes_by_transport_opt
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_identify_protocol():
|
||
|
|
transport_opt_list = [["/ip4/127.0.0.1/tcp/0"], ["/ip4/127.0.0.1/tcp/0"]]
|
||
|
|
(host_a, host_b) = await set_up_nodes_by_transport_opt(transport_opt_list)
|
||
|
|
|
||
|
|
addr = host_a.get_addrs()[0]
|
||
|
|
info = info_from_p2p_addr(addr)
|
||
|
|
await host_b.connect(info)
|
||
|
|
|
||
|
|
stream = await host_b.new_stream(host_a.get_id(), (ID,))
|
||
|
|
response = await stream.read()
|
||
|
|
await stream.close()
|
||
|
|
|
||
|
|
identify_response = Identify()
|
||
|
|
identify_response.ParseFromString(response)
|
||
|
|
assert identify_response == _mk_identify_protobuf(host_a)
|