fix: test refactoring

waku-org · Jan 11, 2024 · 079ccb6 · 079ccb6
1 parent 7a6e0a9
commit 079ccb6
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/packages/tests/tests/peer_exchange.node.spec.ts b/packages/tests/tests/peer_exchange.node.spec.ts
@@ -77,11 +77,17 @@ describe("Peer Exchange", () => {
       expect(peerInfos[0].ENR).to.not.be.null;
       expect(peerInfos[0].ENR?.peerInfo?.multiaddrs).to.not.be.null;
 
-      const foundNodeMultiaddrs: Multiaddr[] = [];
-      const foundNodePeerId: PeerId | undefined = undefined;
-      const doesPeerIdExistInResponse = peerInfos.some(
-        ({ ENR }) => ENR?.peerInfo?.id.toString() === nwaku1PeerId.toString()
-      );
+      let foundNodeMultiaddrs: Multiaddr[] = [];
+      let foundNodePeerId: PeerId | undefined = undefined;
+      const doesPeerIdExistInResponse = peerInfos.some(({ ENR }) => {
+        foundNodeMultiaddrs = ENR?.peerInfo?.multiaddrs ?? [];
+        foundNodePeerId = ENR?.peerInfo?.id;
+        return ENR?.peerInfo?.id.toString() === nwaku1PeerId.toString();
+      });
+
+      if (!foundNodePeerId) {
+        throw new Error("Peer ID not found");
+      }
 
       if (!foundNodePeerId) {
         throw new Error("Peer ID not found");