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

Added tests for 'RoutedHost' and modified 'FindPeer'

Browse Source
This commit is contained in:
Aratz M. Lasa
2019-10-15 01:01:16 +02:00
parent 3f24b015ab
commit 8143563831
7 changed files with 105 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/routing/test_kad_peer_router.py
View File

@ -15,11 +15,11 @@ async def test_simple_two_nodes():
node_a_value = await node_b.bootstrap([("127.0.0.1", 5678)])
node_a_kad_peerinfo = node_a_value[0]
await node_a.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
await node_a.set(node_a_kad_peerinfo.xor_id, node_a_kad_peerinfo.to_string())
router = KadmeliaPeerRouter(node_b)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id_bytes))
assert repr(returned_info) == repr(node_a_kad_peerinfo)
assert returned_info == node_a_kad_peerinfo
@pytest.mark.asyncio
@ -37,11 +37,11 @@ async def test_simple_three_nodes():
node_a_kad_peerinfo = node_a_value[0]
await node_c.bootstrap([("127.0.0.1", 5702)])
await node_a.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
await node_a.set(node_a_kad_peerinfo.xor_id, node_a_kad_peerinfo.to_string())
router = KadmeliaPeerRouter(node_c)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id_bytes))
assert str(returned_info) == str(node_a_kad_peerinfo)
assert returned_info == node_a_kad_peerinfo
@pytest.mark.asyncio
@ -65,8 +65,8 @@ async def test_simple_four_nodes():
await node_d.bootstrap([("127.0.0.1", 5803)])
await node_b.set(node_a_kad_peerinfo.xor_id, repr(node_a_kad_peerinfo))
await node_b.set(node_a_kad_peerinfo.xor_id, node_a_kad_peerinfo.to_string())
router = KadmeliaPeerRouter(node_d)
returned_info = await router.find_peer(ID(node_a_kad_peerinfo.peer_id_bytes))
assert str(returned_info) == str(node_a_kad_peerinfo)
assert returned_info == node_a_kad_peerinfo