mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
bdbb7b2394
And skip the tests for `RoutedHost` for now, since there are too many to be fixed in `Kademlia`, and it's not that necessary now.
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
import pytest
|
|
|
|
from libp2p.host.exceptions import ConnectionFailure
|
|
from libp2p.peer.peerinfo import PeerInfo
|
|
from libp2p.routing.kademlia.kademlia_peer_router import peer_info_to_str
|
|
from libp2p.tools.utils import (
|
|
set_up_nodes_by_transport_and_disc_opt,
|
|
set_up_nodes_by_transport_opt,
|
|
set_up_routers,
|
|
)
|
|
from libp2p.tools.factories import RoutedHostFactory
|
|
|
|
|
|
# FIXME:
|
|
|
|
# TODO: Kademlia is full of asyncio code. Skip it for now
|
|
@pytest.mark.skip
|
|
@pytest.mark.trio
|
|
async def test_host_routing_success(is_host_secure):
|
|
async with RoutedHostFactory.create_batch_and_listen(
|
|
is_host_secure, 2
|
|
) as routed_hosts:
|
|
# Set routing info
|
|
await routed_hosts[0]._router.server.set(
|
|
routed_hosts[0].get_id().xor_id,
|
|
peer_info_to_str(
|
|
PeerInfo(routed_hosts[0].get_id(), routed_hosts[0].get_addrs())
|
|
),
|
|
)
|
|
await routed_hosts[1]._router.server.set(
|
|
routed_hosts[1].get_id().xor_id,
|
|
peer_info_to_str(
|
|
PeerInfo(routed_hosts[1].get_id(), routed_hosts[1].get_addrs())
|
|
),
|
|
)
|
|
|
|
# forces to use routing as no addrs are provided
|
|
await routed_hosts[0].connect(PeerInfo(routed_hosts[1].get_id(), []))
|
|
await routed_hosts[1].connect(PeerInfo(routed_hosts[0].get_id(), []))
|
|
|
|
|
|
# TODO: Kademlia is full of asyncio code. Skip it for now
|
|
@pytest.mark.skip
|
|
@pytest.mark.trio
|
|
async def test_host_routing_fail():
|
|
routers = await set_up_routers()
|
|
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_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,
|
|
peer_info_to_str(PeerInfo(host_a.get_id(), host_a.get_addrs())),
|
|
)
|
|
await routers[1].server.set(
|
|
host_b.get_id().xor_id,
|
|
peer_info_to_str(PeerInfo(host_b.get_id(), host_b.get_addrs())),
|
|
)
|
|
|
|
# routing fails because host_c does not use routing
|
|
with pytest.raises(ConnectionFailure):
|
|
await host_a.connect(PeerInfo(host_c.get_id(), []))
|
|
with pytest.raises(ConnectionFailure):
|
|
await host_b.connect(PeerInfo(host_c.get_id(), []))
|
|
|
|
# Clean up
|
|
routers[0].server.stop()
|
|
routers[1].server.stop()
|