chore: deduplicate ecies encrypted payloads

waku-org · Jan 22, 2025 · ac4edd9 · ac4edd9
1 parent c48d1b8
commit ac4edd9
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/packages/tests/src/lib/index.ts b/packages/tests/src/lib/index.ts
@@ -111,7 +111,16 @@ export class ServiceNodesFleet {
     return relayMessages.every((message) => message);
   }
 
-  public async confirmMessageLength(numMessages: number): Promise<void> {
+  public async confirmMessageLength(
+    numMessages: number,
+    { encryptedPayload }: { encryptedPayload?: boolean } = {
+      encryptedPayload: false
+    }
+  ): Promise<void> {
+    if (encryptedPayload) {
+      expect(this.messageCollector.count).to.equal(numMessages);
+    }
+
     if (this.strictChecking) {
       await Promise.all(
         this.nodes.map(async (node) =>
@@ -144,6 +153,12 @@ class MultipleNodesMessageCollector {
   ) {
     this.callback = (msg: DecodedMessage): void => {
       log.info("Got a message");
+      if (
+        this.messageList.find(
+          (m) => m.payload.toString() === msg.payload.toString()
+        )
+      )
+        return;
       this.messageList.push(msg);
     };
   }

diff --git a/packages/tests/tests/filter/subscribe.node.spec.ts b/packages/tests/tests/filter/subscribe.node.spec.ts
@@ -102,8 +102,7 @@ const runTests = (strictCheckNodes: boolean): void => {
         expectedVersion: 1,
         expectedPubsubTopic: TestPubsubTopic
       });
-
-      await serviceNodes.confirmMessageLength(1);
+      await serviceNodes.confirmMessageLength(1, { encryptedPayload: true });
     });
 
     it("Subscribe and receive symmetrically encrypted messages via lightPush", async function () {

diff --git a/packages/tests/tests/sharding/auto_sharding.spec.ts b/packages/tests/tests/sharding/auto_sharding.spec.ts
@@ -61,7 +61,6 @@ describe("Autosharding: Running Nodes", function () {
       });
 
       expect(request.successes.length).to.eq(2); // Expect 2 successes for 2 nodes
-      console.log("good");
       expect(
         await serviceNodes.messageCollector.waitForMessagesAutosharding(1, {
           contentTopic: ContentTopic