chore: fix test

waku-org · Jul 17, 2024 · c0822ae · c0822ae
1 parent f565e10
commit c0822ae
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/packages/tests/tests/filter/peer_management.spec.ts b/packages/tests/tests/filter/peer_management.spec.ts
@@ -187,6 +187,11 @@ describe("Waku Filter: Peer Management: E2E", function () {
       undefined,
       2
     );
+    const serviceNodesPeerIdStr = await Promise.all(
+      serviceNodes.nodes.map(async (node) =>
+        (await node.getPeerId()).toString()
+      )
+    );
     const nodeWithoutDiscovery = new ServiceNode("WithoutDiscovery");
     await nodeWithoutDiscovery.start({ lightpush: true, filter: true });
     const nodeWithouDiscoveryPeerIdStr = (
@@ -216,17 +221,12 @@ describe("Waku Filter: Peer Management: E2E", function () {
     await sendMessage();
 
     successes
-      .map(async (peerId) =>
-        [
-          nodeWithouDiscoveryPeerIdStr,
-          serviceNodes.nodes.map(async (node) =>
-            (await node.getPeerId()).toString()
-          )
-        ]
-          .flat()
-          .includes(peerId.toString())
+      .map((peerId) =>
+        [nodeWithouDiscoveryPeerIdStr, ...serviceNodesPeerIdStr].includes(
+          peerId.toString()
+        )
       )
-      .forEach((isConnected) => expect(isConnected).to.be.true);
+      .forEach((isConnected) => expect(isConnected).to.eq(true));
 
     // send 2 more messages
     await sendMessage();