varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-10 15:10:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add agent version to identify protocol and improved tests

Browse Source
This commit is contained in:
acul71
2025-02-19 03:43:50 +01:00
committed by Paul Robinson
parent dc903460dc
commit bd8bd953ec
3 changed files with 56 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/core/identity/identify/test_protocol.py
View File

@ -4,8 +4,11 @@ from libp2p.identity.identify.pb.identify_pb2 import (
Identify,
)
from libp2p.identity.identify.protocol import (
AGENT_VERSION,
ID,
PROTOCOL_VERSION,
_mk_identify_protobuf,
_multiaddr_to_bytes,
)
from tests.factories import (
host_pair_factory,
@ -24,4 +27,26 @@ async def test_identify_protocol(security_protocol):
identify_response = Identify()
identify_response.ParseFromString(response)
# sanity check
assert identify_response == _mk_identify_protobuf(host_a)
# Check protocol version
assert identify_response.protocol_version == PROTOCOL_VERSION
# Check agent version
assert identify_response.agent_version == AGENT_VERSION
# Check public key
assert identify_response.public_key == host_a.get_public_key().serialize()
# Check listen addresses
assert identify_response.listen_addrs == list(
map(_multiaddr_to_bytes, host_a.get_addrs())
)
# TODO: Check observed address
# assert identify_response.observed_addr == host_b.get_addrs()[0]
# Check protocols
assert set(identify_response.protocols) == set(host_a.get_mux().get_protocols())