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

Refactored for 'lint' testenv

Browse Source
This commit is contained in:
Aratz M. Lasa
2019-10-15 19:02:03 +02:00
parent 65b5e7aeea
commit d1d91e4091
7 changed files with 584 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tests/host/test_routed_host.py
View File

@ -4,7 +4,11 @@ import pytest
from libp2p.host.exceptions import ConnectionFailure
from libp2p.peer.peerinfo import PeerInfo
from tests.utils import set_up_routers, set_up_nodes_by_transport_opt, set_up_nodes_by_transport_and_disc_opt
from tests.utils import (
set_up_nodes_by_transport_and_disc_opt,
set_up_nodes_by_transport_opt,
set_up_routers,
)
@pytest.mark.asyncio
@ -12,11 +16,19 @@ async def test_host_routing_success():
routers = await set_up_routers([5678, 5679])
transports = [["/ip4/127.0.0.1/tcp/0"], ["/ip4/127.0.0.1/tcp/0"]]
transport_disc_opt_list = zip(transports, routers)
(host_a, host_b) = await set_up_nodes_by_transport_and_disc_opt(transport_disc_opt_list)
(host_a, host_b) = await set_up_nodes_by_transport_and_disc_opt(
transport_disc_opt_list
)
# Set routing info
await routers[0].server.set(host_a.get_id().xor_id, PeerInfo(host_a.get_id(), host_a.get_addrs()).to_string())
await routers[1].server.set(host_b.get_id().xor_id, PeerInfo(host_b.get_id(), host_b.get_addrs()).to_string())
await routers[0].server.set(
host_a.get_id().xor_id,
PeerInfo(host_a.get_id(), host_a.get_addrs()).to_string(),
)
await routers[1].server.set(
host_b.get_id().xor_id,
PeerInfo(host_b.get_id(), host_b.get_addrs()).to_string(),
)
# forces to use routing as no addrs are provided
await host_a.connect(PeerInfo(host_b.get_id(), []))
@ -33,13 +45,21 @@ async def test_host_routing_fail():
routers = await set_up_routers([5678, 5679])
transports = [["/ip4/127.0.0.1/tcp/0"], ["/ip4/127.0.0.1/tcp/0"]]
transport_disc_opt_list = zip(transports, routers)
(host_a, host_b) = await set_up_nodes_by_transport_and_disc_opt(transport_disc_opt_list)
(host_a, host_b) = await set_up_nodes_by_transport_and_disc_opt(
transport_disc_opt_list
)
host_c = (await set_up_nodes_by_transport_opt([["/ip4/127.0.0.1/tcp/0"]]))[0]
# Set routing info
await routers[0].server.set(host_a.get_id().xor_id, PeerInfo(host_a.get_id(), host_a.get_addrs()).to_string())
await routers[1].server.set(host_b.get_id().xor_id, PeerInfo(host_b.get_id(), host_b.get_addrs()).to_string())
await routers[0].server.set(
host_a.get_id().xor_id,
PeerInfo(host_a.get_id(), host_a.get_addrs()).to_string(),
)
await routers[1].server.set(
host_b.get_id().xor_id,
PeerInfo(host_b.get_id(), host_b.get_addrs()).to_string(),
)
# routing fails because host_c does not use routing
with pytest.raises(ConnectionFailure):