varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong peer id type used in KadPeerInfo

Browse Source
This commit is contained in:
NIC619
2019-07-31 21:35:50 +08:00
parent b928bdb356
commit f00e80bc25
4 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/routing/test_kad_peer_router.py
View File

@ -1,5 +1,6 @@
import pytest
from libp2p.peer.id import ID
from libp2p.kademlia.network import KademliaServer
from libp2p.routing.kademlia.kademlia_peer_router import KadmeliaPeerRouter
@ -17,7 +18,7 @@ async def test_simple_two_nodes():
await node_a.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
router = KadmeliaPeerRouter(node_b)
returned_info = await router.find_peer(node_a_kad_peerinfo.peer_id_obj)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id))
print(repr(returned_info))
print(repr(node_a_kad_peerinfo))
assert repr(returned_info) == repr(node_a_kad_peerinfo)
@ -41,7 +42,7 @@ async def test_simple_three_nodes():
await node_a.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
router = KadmeliaPeerRouter(node_c)
returned_info = await router.find_peer(node_a_kad_peerinfo.peer_id_obj)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id))
assert str(returned_info) == str(node_a_kad_peerinfo)
@ -69,5 +70,5 @@ async def test_simple_four_nodes():
await node_b.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
router = KadmeliaPeerRouter(node_d)
returned_info = await router.find_peer(node_a_kad_peerinfo.peer_id_obj)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id))
assert str(returned_info) == str(node_a_kad_peerinfo)