test(sharding): Implement sharding tests (#2603)

* Implement sharding tests.
waku-org · May 13, 2024 · 6c3ad50 · 6c3ad50
1 parent 49dd6c1
commit 6c3ad50
Show file tree

Hide file tree

Showing 8 changed files with 1,317 additions and 72 deletions.
diff --git a/tests/node/test_wakunode_relay_rln.nim b/tests/node/test_wakunode_relay_rln.nim
@@ -16,71 +16,10 @@ import
     [node/waku_node, node/peer_manager, waku_core, waku_node, waku_rln_relay],
   ../waku_store/store_utils,
   ../waku_archive/archive_utils,
+  ../waku_relay/utils,
   ../testlib/[wakucore, wakunode, testasync, futures],
   ../resources/payloads
 
-proc setupRln(node: WakuNode, identifier: uint) {.async.} =
-  await node.mountRlnRelay(
-    WakuRlnConfig(
-      rlnRelayDynamic: false,
-      rlnRelayCredIndex: some(identifier),
-      rlnRelayTreePath: genTempPath("rln_tree", "wakunode_" & $identifier),
-      rlnEpochSizeSec: 1,
-    )
-  )
-
-proc setupRelayWithRln(
-    node: WakuNode, identifier: uint, pubsubTopics: seq[string]
-) {.async.} =
-  await node.mountRelay(pubsubTopics)
-  await setupRln(node, identifier)
-
-proc subscribeCompletionHandler(node: WakuNode, pubsubTopic: string): Future[bool] =
-  var completionFut = newFuture[bool]()
-  proc relayHandler(
-      topic: PubsubTopic, msg: WakuMessage
-  ): Future[void] {.async, gcsafe.} =
-    if topic == pubsubTopic:
-      completionFut.complete(true)
-
-  node.subscribe((kind: PubsubSub, topic: pubsubTopic), some(relayHandler))
-  return completionFut
-
-proc sendRlnMessage(
-    client: WakuNode,
-    pubsubTopic: string,
-    contentTopic: string,
-    completionFuture: Future[bool],
-    payload: seq[byte] = "Hello".toBytes(),
-): Future[bool] {.async.} =
-  var message = WakuMessage(payload: payload, contentTopic: contentTopic)
-  doAssert(client.wakuRlnRelay.appendRLNProof(message, epochTime()).isOk())
-  discard await client.publish(some(pubsubTopic), message)
-  let isCompleted = await completionFuture.withTimeout(FUTURE_TIMEOUT)
-  return isCompleted
-
-proc sendRlnMessageWithInvalidProof(
-    client: WakuNode,
-    pubsubTopic: string,
-    contentTopic: string,
-    completionFuture: Future[bool],
-    payload: seq[byte] = "Hello".toBytes(),
-): Future[bool] {.async.} =
-  let
-    extraBytes: seq[byte] = @[byte(1), 2, 3]
-    rateLimitProofRes = client.wakuRlnRelay.groupManager.generateProof(
-      concat(payload, extraBytes),
-        # we add extra bytes to invalidate proof verification against original payload
-      client.wakuRlnRelay.getCurrentEpoch(),
-    )
-    rateLimitProof = rateLimitProofRes.get().encode().buffer
-    message =
-      WakuMessage(payload: @payload, contentTopic: contentTopic, proof: rateLimitProof)
-
-  discard await client.publish(some(pubsubTopic), message)
-  let isCompleted = await completionFuture.withTimeout(FUTURE_TIMEOUT)
-  return isCompleted
-
 suite "Waku RlnRelay - End to End":
   var
     pubsubTopic {.threadvar.}: PubsubTopic
@@ -237,23 +176,29 @@ suite "Waku RlnRelay - End to End":
 
       doAssert(
         client.wakuRlnRelay
-        .appendRLNProof(message1b, epoch + client.wakuRlnRelay.rlnEpochSizeSec * 0)
+        .appendRLNProof(
+          message1b, epoch + float64(client.wakuRlnRelay.rlnEpochSizeSec * 0)
+        )
         .isOk()
       )
       doAssert(
         client.wakuRlnRelay
-        .appendRLNProof(message1kib, epoch + client.wakuRlnRelay.rlnEpochSizeSec * 1)
+        .appendRLNProof(
+          message1kib, epoch + float64(client.wakuRlnRelay.rlnEpochSizeSec * 1)
+        )
         .isOk()
       )
       doAssert(
         client.wakuRlnRelay
-        .appendRLNProof(message150kib, epoch + client.wakuRlnRelay.rlnEpochSizeSec * 2)
+        .appendRLNProof(
+          message150kib, epoch + float64(client.wakuRlnRelay.rlnEpochSizeSec * 2)
+        )
         .isOk()
       )
       doAssert(
         client.wakuRlnRelay
         .appendRLNProof(
-          message151kibPlus, epoch + client.wakuRlnRelay.rlnEpochSizeSec * 3
+          message151kibPlus, epoch + float64(client.wakuRlnRelay.rlnEpochSizeSec * 3)
         )
         .isOk()
       )
@@ -317,9 +262,11 @@ suite "Waku RlnRelay - End to End":
         WakuMessage(payload: @payload150kibPlus, contentTopic: contentTopic)
 
       doAssert(
-        client.wakuRlnRelay.appendRLNProof(
-          message151kibPlus, epoch + client.wakuRlnRelay.rlnEpochSizeSec * 3
+        client.wakuRlnRelay
+        .appendRLNProof(
+          message151kibPlus, epoch + float64(client.wakuRlnRelay.rlnEpochSizeSec * 3)
         )
+        .isOk()
       )
 
       # When sending the 150KiB plus message