mirror of https://github.com/waku-org/nwaku.git
300 lines
9.1 KiB
Nim
300 lines
9.1 KiB
Nim
{.used.}
|
||
|
||
import
|
||
std/[options, sequtils],
|
||
testutils/unittests,
|
||
chronos,
|
||
chronicles,
|
||
stew/shims/net,
|
||
libp2p/switch,
|
||
libp2p/peerId,
|
||
libp2p/crypto/crypto,
|
||
eth/keys,
|
||
eth/p2p/discoveryv5/enr
|
||
|
||
import
|
||
waku/[
|
||
waku_node,
|
||
discovery/waku_discv5,
|
||
waku_peer_exchange,
|
||
node/peer_manager,
|
||
waku_relay/protocol,
|
||
waku_core,
|
||
],
|
||
../waku_peer_exchange/utils,
|
||
../testlib/[wakucore, wakunode, testasync]
|
||
|
||
suite "Waku Peer Exchange":
|
||
let
|
||
bindIp: IPAddress = parseIpAddress("0.0.0.0")
|
||
bindPort: Port = Port(0)
|
||
|
||
var node {.threadvar.}: WakuNode
|
||
|
||
suite "mountPeerExchange":
|
||
asyncSetup:
|
||
node = newTestWakuNode(generateSecp256k1Key(), bindIp, bindPort)
|
||
|
||
asyncTest "Started node mounts peer exchange":
|
||
# Given a started node without peer exchange mounted
|
||
await node.start()
|
||
check:
|
||
node.wakuPeerExchange == nil
|
||
|
||
# When mounting peer exchange
|
||
await node.mountPeerExchange()
|
||
|
||
# Then peer exchange is mounted
|
||
check:
|
||
node.wakuPeerExchange != nil
|
||
node.wakuPeerExchange.started == true
|
||
|
||
# Cleanup
|
||
await node.stop()
|
||
|
||
asyncTest "Stopped node mounts peer exchange":
|
||
# Given a stopped node without peer exchange mounted
|
||
check:
|
||
node.wakuPeerExchange == nil
|
||
|
||
# When mounting peer exchange
|
||
await node.mountPeerExchange()
|
||
|
||
# Then peer exchange is mounted
|
||
check:
|
||
node.wakuPeerExchange != nil
|
||
node.wakuPeerExchange.started == false
|
||
|
||
suite "fetchPeerExchangePeers":
|
||
var node2 {.threadvar.}: WakuNode
|
||
|
||
asyncSetup:
|
||
node = newTestWakuNode(generateSecp256k1Key(), bindIp, bindPort)
|
||
node2 = newTestWakuNode(generateSecp256k1Key(), bindIp, bindPort)
|
||
|
||
await allFutures(node.start(), node2.start())
|
||
|
||
asyncTeardown:
|
||
await allFutures(node.stop(), node2.stop())
|
||
|
||
asyncTest "Node fetches without mounting peer exchange":
|
||
# When a node, without peer exchange mounted, fetches peers
|
||
let res = await node.fetchPeerExchangePeers(1)
|
||
|
||
# Then no peers are fetched
|
||
check:
|
||
node.peerManager.peerStore.peers.len == 0
|
||
res.error.status_code == SERVICE_UNAVAILABLE
|
||
res.error.status_desc == some("PeerExchange is not mounted")
|
||
|
||
asyncTest "Node fetches with mounted peer exchange, but no peers":
|
||
# Given a node with peer exchange mounted
|
||
await node.mountPeerExchange()
|
||
|
||
# When a node fetches peers
|
||
let res = await node.fetchPeerExchangePeers(1)
|
||
check:
|
||
res.error.status_code == SERVICE_UNAVAILABLE
|
||
res.error.status_desc == some("peer_not_found_failure")
|
||
|
||
# Then no peers are fetched
|
||
check node.peerManager.peerStore.peers.len == 0
|
||
|
||
asyncTest "Node succesfully exchanges px peers with faked discv5":
|
||
# Given both nodes mount peer exchange
|
||
await allFutures([node.mountPeerExchange(), node2.mountPeerExchange()])
|
||
check node.peerManager.peerStore.peers.len == 0
|
||
|
||
# Mock that we discovered a node (to avoid running discv5)
|
||
var enr = enr.Record()
|
||
assert enr.fromUri(
|
||
"enr:-Iu4QGNuTvNRulF3A4Kb9YHiIXLr0z_CpvWkWjWKU-o95zUPR_In02AWek4nsSk7G_-YDcaT4bDRPzt5JIWvFqkXSNcBgmlkgnY0gmlwhE0WsGeJc2VjcDI1NmsxoQKp9VzU2FAh7fwOwSpg1M_Ekz4zzl0Fpbg6po2ZwgVwQYN0Y3CC6mCFd2FrdTIB"
|
||
), "Failed to parse ENR"
|
||
node2.wakuPeerExchange.enrCache.add(enr)
|
||
|
||
# Set node2 as service peer (default one) for px protocol
|
||
node.peerManager.addServicePeer(
|
||
node2.peerInfo.toRemotePeerInfo(), WakuPeerExchangeCodec
|
||
)
|
||
|
||
# Request 1 peer from peer exchange protocol
|
||
let res = await node.fetchPeerExchangePeers(1)
|
||
check res.tryGet() == 1
|
||
|
||
# Check that the peer ended up in the peerstore
|
||
let rpInfo = enr.toRemotePeerInfo.get()
|
||
check:
|
||
node.peerManager.peerStore.peers.anyIt(it.peerId == rpInfo.peerId)
|
||
node.peerManager.peerStore.peers.anyIt(it.addrs == rpInfo.addrs)
|
||
|
||
suite "setPeerExchangePeer":
|
||
var node2 {.threadvar.}: WakuNode
|
||
|
||
asyncSetup:
|
||
node = newTestWakuNode(generateSecp256k1Key(), bindIp, bindPort)
|
||
node2 = newTestWakuNode(generateSecp256k1Key(), bindIp, bindPort)
|
||
|
||
await allFutures(node.start(), node2.start())
|
||
|
||
asyncTeardown:
|
||
await allFutures(node.stop(), node2.stop())
|
||
|
||
asyncTest "peer set successfully":
|
||
# Given a node with peer exchange mounted
|
||
await node.mountPeerExchange()
|
||
let initialPeers = node.peerManager.peerStore.peers.len
|
||
|
||
# And a valid peer info
|
||
let remotePeerInfo2 = node2.peerInfo.toRemotePeerInfo()
|
||
|
||
# When making a request with a valid peer info
|
||
node.setPeerExchangePeer(remotePeerInfo2)
|
||
|
||
# Then the peer is added to the peer store
|
||
check:
|
||
node.peerManager.peerStore.peers.len == (initialPeers + 1)
|
||
|
||
asyncTest "peer exchange not mounted":
|
||
# Given a node without peer exchange mounted
|
||
check node.wakuPeerExchange == nil
|
||
let initialPeers = node.peerManager.peerStore.peers.len
|
||
|
||
# And a valid peer info
|
||
let invalidMultiAddress = MultiAddress.init("/ip4/0.0.0.0/tcp/0").tryGet()
|
||
|
||
# When making any request with an invalid peer info
|
||
node.setPeerExchangePeer(invalidMultiAddress)
|
||
|
||
# Then no peer is added to the peer store
|
||
check:
|
||
node.peerManager.peerStore.peers.len == initialPeers
|
||
|
||
asyncTest "peer info parse error":
|
||
# Given a node with peer exchange mounted
|
||
await node.mountPeerExchange()
|
||
let initialPeers = node.peerManager.peerStore.peers.len
|
||
|
||
# And given a peer info with an invalid peer id
|
||
var remotePeerInfo2 = node2.peerInfo.toRemotePeerInfo()
|
||
remotePeerInfo2.peerId.data.add(255.byte)
|
||
|
||
# When making any request with an invalid peer info
|
||
node.setPeerExchangePeer("invalidpeerinfo")
|
||
|
||
# Then no peer is added to the peer store
|
||
check:
|
||
node.peerManager.peerStore.peers.len == initialPeers
|
||
|
||
suite "Waku Peer Exchange with discv5":
|
||
asyncTest "Node successfully exchanges px peers with real discv5":
|
||
## Given (copied from test_waku_discv5.nim)
|
||
let
|
||
# todo: px flag
|
||
flags = CapabilitiesBitfield.init(
|
||
lightpush = false, filter = false, store = false, relay = true
|
||
)
|
||
bindIp = parseIpAddress("0.0.0.0")
|
||
extIp = parseIpAddress("127.0.0.1")
|
||
|
||
nodeKey1 = generateSecp256k1Key()
|
||
nodeTcpPort1 = Port(64010)
|
||
nodeUdpPort1 = Port(9000)
|
||
node1 = newTestWakuNode(
|
||
nodeKey1,
|
||
bindIp,
|
||
nodeTcpPort1,
|
||
some(extIp),
|
||
wakuFlags = some(flags),
|
||
discv5UdpPort = some(nodeUdpPort1),
|
||
)
|
||
|
||
nodeKey2 = generateSecp256k1Key()
|
||
nodeTcpPort2 = Port(64012)
|
||
nodeUdpPort2 = Port(9002)
|
||
node2 = newTestWakuNode(
|
||
nodeKey2,
|
||
bindIp,
|
||
nodeTcpPort2,
|
||
some(extIp),
|
||
wakuFlags = some(flags),
|
||
discv5UdpPort = some(nodeUdpPort2),
|
||
)
|
||
|
||
nodeKey3 = generateSecp256k1Key()
|
||
nodeTcpPort3 = Port(64014)
|
||
nodeUdpPort3 = Port(9004)
|
||
node3 = newTestWakuNode(
|
||
nodeKey3,
|
||
bindIp,
|
||
nodeTcpPort3,
|
||
some(extIp),
|
||
wakuFlags = some(flags),
|
||
discv5UdpPort = some(nodeUdpPort3),
|
||
)
|
||
|
||
# discv5
|
||
let conf1 = WakuDiscoveryV5Config(
|
||
discv5Config: none(DiscoveryConfig),
|
||
address: bindIp,
|
||
port: nodeUdpPort1,
|
||
privateKey: keys.PrivateKey(nodeKey1.skkey),
|
||
bootstrapRecords: @[],
|
||
autoupdateRecord: true,
|
||
)
|
||
|
||
let disc1 =
|
||
WakuDiscoveryV5.new(node1.rng, conf1, some(node1.enr), some(node1.peerManager))
|
||
|
||
let conf2 = WakuDiscoveryV5Config(
|
||
discv5Config: none(DiscoveryConfig),
|
||
address: bindIp,
|
||
port: nodeUdpPort2,
|
||
privateKey: keys.PrivateKey(nodeKey2.skkey),
|
||
bootstrapRecords: @[disc1.protocol.getRecord()],
|
||
autoupdateRecord: true,
|
||
)
|
||
|
||
let disc2 =
|
||
WakuDiscoveryV5.new(node2.rng, conf2, some(node2.enr), some(node2.peerManager))
|
||
|
||
await allFutures(node1.start(), node2.start(), node3.start())
|
||
let resultDisc1StartRes = await disc1.start()
|
||
assert resultDisc1StartRes.isOk(), resultDisc1StartRes.error
|
||
let resultDisc2StartRes = await disc2.start()
|
||
assert resultDisc2StartRes.isOk(), resultDisc2StartRes.error
|
||
|
||
## When
|
||
var attempts = 10
|
||
while (disc1.protocol.nodesDiscovered < 1 or disc2.protocol.nodesDiscovered < 1) and
|
||
attempts > 0:
|
||
await sleepAsync(1.seconds)
|
||
attempts -= 1
|
||
|
||
# node2 can be connected, so will be returned by peer exchange
|
||
require (
|
||
await node1.peerManager.connectRelay(node2.switch.peerInfo.toRemotePeerInfo())
|
||
)
|
||
|
||
# Mount peer exchange
|
||
await node1.mountPeerExchange()
|
||
await node3.mountPeerExchange()
|
||
|
||
let dialResponse =
|
||
await node3.dialForPeerExchange(node1.switch.peerInfo.toRemotePeerInfo())
|
||
|
||
check dialResponse.isOk
|
||
|
||
let
|
||
requestPeers = 1
|
||
currentPeers = node3.peerManager.peerStore.peers.len
|
||
let res = await node3.fetchPeerExchangePeers(1)
|
||
check res.tryGet() == 1
|
||
|
||
# Then node3 has received 1 peer from node1
|
||
check:
|
||
node3.peerManager.peerStore.peers.len == currentPeers + requestPeers
|
||
|
||
await allFutures(
|
||
[node1.stop(), node2.stop(), node3.stop(), disc1.stop(), disc2.stop()]
|
||
)
|