fix tests

This commit is contained in:
zixuanzh
2019-04-28 13:49:10 -04:00
parent fdaa4f56b9
commit 3b1a3fb198
2 changed files with 45 additions and 24 deletions

View File

@ -19,15 +19,14 @@ class KadmeliaPeerRouter(IPeerRouting):
# switching peer_id to xor_id used by kademlia as node_id
xor_id = peer_id.get_xor_id()
value = await self.server.get(xor_id)
return decode_peerinfo(value)
return value
def decode_peerinfo(encoded):
if isinstance(encoded, bytes):
encoded = encoded.decode()
lines = encoded.splitlines()
peer_id = lines[0]
addrs = lines[1:]
peer_data = PeerData()
peer_data.add_addrs(addrs)
return PeerInfo(peer_id, addrs)
# def decode_peerinfo(encoded):
# if isinstance(encoded, bytes):
# encoded = encoded.decode()
# lines = encoded.splitlines()
# peer_id = lines[0]
# addrs = lines[1:]
# peer_data = PeerData()
# peer_data.add_addrs(addrs)
# return PeerInfo(peer_id, addrs)