2019-09-02 17:32:15 +08:00
|
|
|
import pytest
|
2020-01-07 14:14:34 +08:00
|
|
|
import trio
|
2019-09-02 17:32:15 +08:00
|
|
|
|
2020-01-07 14:14:34 +08:00
|
|
|
from libp2p.tools.factories import HostFactory
|
2019-11-21 11:47:54 +08:00
|
|
|
from libp2p.tools.interop.utils import connect
|
2019-09-02 18:40:12 +08:00
|
|
|
|
2019-09-02 17:32:15 +08:00
|
|
|
|
2020-01-07 14:14:34 +08:00
|
|
|
@pytest.mark.trio
|
2020-02-19 23:15:03 +08:00
|
|
|
async def test_connect(security_protocol, p2pds):
|
|
|
|
|
async with HostFactory.create_batch_and_listen(
|
|
|
|
|
1, security_protocol=security_protocol
|
|
|
|
|
) as hosts:
|
2020-01-07 14:14:34 +08:00
|
|
|
p2pd = p2pds[0]
|
|
|
|
|
host = hosts[0]
|
|
|
|
|
assert len(await p2pd.control.list_peers()) == 0
|
|
|
|
|
# Test: connect from Py
|
|
|
|
|
await connect(host, p2pd)
|
|
|
|
|
assert len(await p2pd.control.list_peers()) == 1
|
|
|
|
|
# Test: `disconnect` from Py
|
|
|
|
|
await host.disconnect(p2pd.peer_id)
|
|
|
|
|
assert len(await p2pd.control.list_peers()) == 0
|
|
|
|
|
# Test: connect from Go
|
|
|
|
|
await connect(p2pd, host)
|
|
|
|
|
assert len(host.get_network().connections) == 1
|
|
|
|
|
# Test: `disconnect` from Go
|
|
|
|
|
await p2pd.control.disconnect(host.get_id())
|
|
|
|
|
await trio.sleep(0.01)
|
|
|
|
|
assert len(host.get_network().connections) == 0
|
