def test_unpack_eip8_packets():
# Test our _unpack() function against the sample packets specified in
# https://github.com/ethereum/EIPs/blob/master/EIPS/eip-8.md
for cmd, packets in eip8_packets.items():
for _, packet in packets.items():
pubkey, cmd_id, payload, _ = _unpack_v4(packet)
assert pubkey.to_hex() == '0xca634cae0d49acb401d8a4c6b6fe8c55b70d115bf400769cc1400f3258cd31387574077f301b421bc84df7266c44e9e6d569fc56be00812904767bf5ccd1fc7f' # noqa: E501
assert cmd.id == cmd_id
def test_discover_v4_message_pack():
sender, recipient = AddressFactory.create_batch(2)
version = rlp.sedes.big_endian_int.serialize(PROTO_VERSION)
payload = (version, sender.to_endpoint(), recipient.to_endpoint())
privkey = PrivateKeyFactory()
message = _pack_v4(CMD_PING.id, payload, privkey)
pubkey, cmd_id, payload, _ = _unpack_v4(message)
assert pubkey == privkey.public_key
assert cmd_id == CMD_PING.id
def test_pack():
sender, recipient = random_address(), random_address()
version = rlp.sedes.big_endian_int.serialize(discovery.PROTO_VERSION)
payload = (version, sender.to_endpoint(), recipient.to_endpoint())
privkey = keys.PrivateKey(keccak(b"seed"))
message = discovery._pack_v4(discovery.CMD_PING.id, payload, privkey)
pubkey, cmd_id, payload, _ = discovery._unpack_v4(message)
assert pubkey == privkey.public_key
assert cmd_id == discovery.CMD_PING.id
assert len(payload) == discovery.CMD_PING.elem_count
