2021-02-04 10:32:58 +00:00
|
|
|
import
|
|
|
|
std/[unittest, options, sets, tables, sequtils],
|
|
|
|
stew/shims/net as stewNet,
|
|
|
|
json_rpc/[rpcserver, rpcclient],
|
|
|
|
eth/[keys, rlp], eth/common/eth_types,
|
|
|
|
libp2p/[standard_setup, switch, multiaddress],
|
|
|
|
libp2p/protobuf/minprotobuf,
|
|
|
|
libp2p/stream/[bufferstream, connection],
|
|
|
|
libp2p/crypto/crypto,
|
|
|
|
libp2p/protocols/pubsub/pubsub,
|
|
|
|
libp2p/protocols/pubsub/rpc/message,
|
|
|
|
../../waku/v2/node/wakunode2,
|
|
|
|
../../waku/v2/node/peer_manager,
|
|
|
|
../../waku/v2/protocol/waku_relay,
|
|
|
|
../test_helpers
|
|
|
|
|
|
|
|
procSuite "Peer Manager":
|
2021-02-05 10:49:11 +00:00
|
|
|
asyncTest "Peer dialing works":
|
2021-02-04 10:32:58 +00:00
|
|
|
let
|
|
|
|
nodeKey1 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
|
|
node1 = WakuNode.init(nodeKey1, ValidIpAddress.init("0.0.0.0"),
|
|
|
|
Port(60000))
|
|
|
|
nodeKey2 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
|
|
node2 = WakuNode.init(nodeKey2, ValidIpAddress.init("0.0.0.0"),
|
|
|
|
Port(60002))
|
|
|
|
peerInfo2 = node2.peerInfo
|
|
|
|
|
|
|
|
await allFutures([node1.start(), node2.start()])
|
|
|
|
|
|
|
|
node1.mountRelay()
|
|
|
|
node2.mountRelay()
|
|
|
|
|
|
|
|
# Dial node2 from node1
|
2021-02-05 10:49:11 +00:00
|
|
|
let conn = (await node1.peerManager.dialPeer(peerInfo2, WakuRelayCodec)).get()
|
2021-02-04 10:32:58 +00:00
|
|
|
|
|
|
|
# Check connection
|
|
|
|
check:
|
|
|
|
conn.activity
|
|
|
|
conn.peerInfo.peerId == peerInfo2.peerId
|
|
|
|
|
|
|
|
# Check that node2 is being managed in node1
|
|
|
|
check:
|
|
|
|
node1.peerManager.peers().anyIt(it.peerId == peerInfo2.peerId)
|
|
|
|
|
|
|
|
# Check connectedness
|
|
|
|
check:
|
|
|
|
node1.peerManager.connectedness(peerInfo2.peerId)
|
2021-02-05 10:49:11 +00:00
|
|
|
|
|
|
|
await allFutures([node1.stop(), node2.stop()])
|
|
|
|
|
|
|
|
asyncTest "Dialing fails gracefully":
|
|
|
|
let
|
|
|
|
nodeKey1 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
|
|
node1 = WakuNode.init(nodeKey1, ValidIpAddress.init("0.0.0.0"),
|
|
|
|
Port(60000))
|
|
|
|
nodeKey2 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
|
|
node2 = WakuNode.init(nodeKey2, ValidIpAddress.init("0.0.0.0"),
|
|
|
|
Port(60002))
|
|
|
|
peerInfo2 = node2.peerInfo
|
|
|
|
|
|
|
|
await node1.start()
|
|
|
|
# Purposefully don't start node2
|
|
|
|
|
|
|
|
node1.mountRelay()
|
|
|
|
node2.mountRelay()
|
2021-02-04 10:32:58 +00:00
|
|
|
|
2021-02-05 10:49:11 +00:00
|
|
|
# Dial node2 from node1
|
|
|
|
let connOpt = await node1.peerManager.dialPeer(peerInfo2, WakuRelayCodec, 2.seconds)
|
|
|
|
|
|
|
|
# Check connection failed gracefully
|
|
|
|
check:
|
|
|
|
connOpt.isNone()
|
|
|
|
|
|
|
|
await node1.stop()
|