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

fixed the linter <> protobuf issues

Browse Source
This commit is contained in:
lla-dane
2025-07-17 23:22:49 +05:30
parent 8625226be8
commit 9d597012cc
11 changed files with 117 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/core/peer/test_envelope.py
View File

@ -19,11 +19,11 @@ DOMAIN = "libp2p-peer-record"
def test_basic_protobuf_serialization_deserialization():
pubkey = crypto_pb.PublicKey() # type: ignore[attr-defined]
pubkey.Type = crypto_pb.KeyType.Ed25519 # type: ignore[attr-defined]
pubkey = crypto_pb.PublicKey()
pubkey.Type = crypto_pb.KeyType.Ed25519
pubkey.Data = b"\x01\x02\x03"
env = env_pb.Envelope() # type: ignore[attr-defined]
env = env_pb.Envelope()
env.public_key.CopyFrom(pubkey)
env.payload_type = b"\x03\x01"
env.payload = b"test-payload"
@ -31,10 +31,10 @@ def test_basic_protobuf_serialization_deserialization():
serialized = env.SerializeToString()
new_env = env_pb.Envelope() # type: ignore[attr-defined]
new_env = env_pb.Envelope()
new_env.ParseFromString(serialized)
assert new_env.public_key.Type == crypto_pb.KeyType.Ed25519 # type: ignore[attr-defined]
assert new_env.public_key.Type == crypto_pb.KeyType.Ed25519
assert new_env.public_key.Data == b"\x01\x02\x03"
assert new_env.payload_type == b"\x03\x01"
assert new_env.payload == b"test-payload"

14
tests/core/peer/test_peer_record.py
View File

@ -15,11 +15,11 @@ from libp2p.peer.peer_record import (
def test_basic_protobuf_serializatrion_deserialization():
record = pb.PeerRecord() # type: ignore[attr-defined]
record = pb.PeerRecord()
record.seq = 1
serialized = record.SerializeToString()
new_record = pb.PeerRecord() # type: ignore[attr-defined]
new_record = pb.PeerRecord()
new_record.ParseFromString(serialized)
assert new_record.seq == 1
@ -39,10 +39,10 @@ def test_addrs_from_protobuf_multiple_addresses():
ma1 = Multiaddr("/ip4/127.0.0.1/tcp/4001")
ma2 = Multiaddr("/ip4/127.0.0.1/tcp/4002")
addr_info1 = pb.PeerRecord.AddressInfo() # type: ignore[attr-defined]
addr_info1 = pb.PeerRecord.AddressInfo()
addr_info1.multiaddr = ma1.to_bytes()
addr_info2 = pb.PeerRecord.AddressInfo() # type: ignore[attr-defined]
addr_info2 = pb.PeerRecord.AddressInfo()
addr_info2.multiaddr = ma2.to_bytes()
result = addrs_from_protobuf([addr_info1, addr_info2])
@ -51,13 +51,13 @@ def test_addrs_from_protobuf_multiple_addresses():
def test_peer_record_from_protobuf():
peer_id = ID.from_base58("QmNM23MiU1Kd7yfiKVdUnaDo8RYca8By4zDmr7uSaVV8Px")
record = pb.PeerRecord() # type: ignore[attr-defined]
record = pb.PeerRecord()
record.peer_id = peer_id.to_bytes()
record.seq = 42
for addr_str in ["/ip4/127.0.0.1/tcp/4001", "/ip4/127.0.0.1/tcp/4002"]:
ma = Multiaddr(addr_str)
addr_info = pb.PeerRecord.AddressInfo() # type: ignore[attr-defined]
addr_info = pb.PeerRecord.AddressInfo()
addr_info.multiaddr = ma.to_bytes()
record.addresses.append(addr_info)
@ -78,7 +78,7 @@ def test_to_protobuf_generates_correct_message():
record = PeerRecord(peer_id, addrs, seq)
proto = record.to_protobuf()
assert isinstance(proto, pb.PeerRecord) # type: ignore[attr-defined]
assert isinstance(proto, pb.PeerRecord)
assert proto.peer_id == peer_id.to_bytes()
assert proto.seq == seq
assert len(proto.addresses) == 1