mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-02-18 21:03:36 +00:00
84f791100f
* Fix peer selection for cases where ENR is not yet advertiesed but metadata exchange already adjusted supported shards. Fix initialization rendezvous protocol with configured and autoshards to let connect to relay nodes without having a valid subscribed shard already. This solves issue for autoshard nodes to connect ahead of subscribing. * Extend peer selection, rendezvous and metadata tests * Fix rendezvous test, fix metadata test failing due wrong setup, added it into all_tests
128 lines
3.5 KiB
Nim
128 lines
3.5 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/[options, sequtils, tables],
|
|
testutils/unittests,
|
|
chronos,
|
|
chronicles,
|
|
libp2p/switch,
|
|
libp2p/peerId,
|
|
libp2p/crypto/crypto,
|
|
libp2p/multistream,
|
|
libp2p/muxers/muxer,
|
|
eth/keys,
|
|
eth/p2p/discoveryv5/enr
|
|
import
|
|
waku/[
|
|
waku_node,
|
|
waku_core/topics,
|
|
waku_core,
|
|
node/peer_manager,
|
|
discovery/waku_discv5,
|
|
waku_metadata,
|
|
waku_relay/protocol,
|
|
],
|
|
./testlib/wakucore,
|
|
./testlib/wakunode
|
|
|
|
procSuite "Waku Metadata Protocol":
|
|
asyncTest "request() returns the supported metadata of the peer":
|
|
let clusterId = 10.uint16
|
|
let
|
|
node1 = newTestWakuNode(
|
|
generateSecp256k1Key(),
|
|
parseIpAddress("0.0.0.0"),
|
|
Port(0),
|
|
clusterId = clusterId,
|
|
)
|
|
node2 = newTestWakuNode(
|
|
generateSecp256k1Key(),
|
|
parseIpAddress("0.0.0.0"),
|
|
Port(0),
|
|
clusterId = clusterId,
|
|
)
|
|
|
|
# Mount metadata protocol on both nodes before starting
|
|
discard node1.mountMetadata(clusterId, @[])
|
|
discard node2.mountMetadata(clusterId, @[])
|
|
|
|
# Mount relay so metadata can track subscriptions
|
|
discard await node1.mountRelay()
|
|
discard await node2.mountRelay()
|
|
|
|
# Start nodes
|
|
await allFutures([node1.start(), node2.start()])
|
|
|
|
# Subscribe to topics on node1 - relay will track these and metadata will report them
|
|
let noOpHandler: WakuRelayHandler = proc(
|
|
pubsubTopic: PubsubTopic, message: WakuMessage
|
|
): Future[void] {.async.} =
|
|
discard
|
|
|
|
node1.wakuRelay.subscribe("/waku/2/rs/10/7", noOpHandler)
|
|
node1.wakuRelay.subscribe("/waku/2/rs/10/6", noOpHandler)
|
|
|
|
# Create connection
|
|
let connOpt = await node2.peerManager.dialPeer(
|
|
node1.switch.peerInfo.toRemotePeerInfo(), WakuMetadataCodec
|
|
)
|
|
require:
|
|
connOpt.isSome()
|
|
|
|
# Request metadata
|
|
let response1 = await node2.wakuMetadata.request(connOpt.get())
|
|
|
|
# Check the response or dont even continue
|
|
require:
|
|
response1.isOk()
|
|
|
|
check:
|
|
response1.get().clusterId.get() == clusterId
|
|
response1.get().shards == @[uint32(6), uint32(7)]
|
|
|
|
await allFutures([node1.stop(), node2.stop()])
|
|
|
|
asyncTest "Metadata reports configured shards before relay subscription":
|
|
## Given: Node with configured shards but no relay subscriptions yet
|
|
let
|
|
clusterId = 10.uint16
|
|
configuredShards = @[uint16(0), uint16(1)]
|
|
|
|
let node1 = newTestWakuNode(
|
|
generateSecp256k1Key(),
|
|
parseIpAddress("0.0.0.0"),
|
|
Port(0),
|
|
clusterId = clusterId,
|
|
subscribeShards = configuredShards,
|
|
)
|
|
let node2 = newTestWakuNode(
|
|
generateSecp256k1Key(), parseIpAddress("0.0.0.0"), Port(0), clusterId = clusterId
|
|
)
|
|
|
|
# Mount metadata with configured shards on node1
|
|
discard node1.mountMetadata(clusterId, configuredShards)
|
|
# Mount metadata on node2 so it can make requests
|
|
discard node2.mountMetadata(clusterId, @[])
|
|
|
|
# Start nodes (relay is NOT mounted yet on node1)
|
|
await allFutures([node1.start(), node2.start()])
|
|
|
|
## When: Node2 requests metadata from Node1 before relay is active
|
|
let connOpt = await node2.peerManager.dialPeer(
|
|
node1.switch.peerInfo.toRemotePeerInfo(), WakuMetadataCodec
|
|
)
|
|
require:
|
|
connOpt.isSome
|
|
|
|
let response = await node2.wakuMetadata.request(connOpt.get())
|
|
|
|
## Then: Response contains configured shards even without relay subscriptions
|
|
require:
|
|
response.isOk()
|
|
|
|
check:
|
|
response.get().clusterId.get() == clusterId
|
|
response.get().shards == @[uint32(0), uint32(1)]
|
|
|
|
await allFutures([node1.stop(), node2.stop()])
|