From 079ccb6923ac160c98523bf1c188eb3a0558a46b Mon Sep 17 00:00:00 2001
From: danisharora099 <danisharora099@gmail.com>
Date: Thu, 11 Jan 2024 15:43:31 +0530
Subject: [PATCH] fix: test refactoring

---
 packages/tests/tests/peer_exchange.node.spec.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/packages/tests/tests/peer_exchange.node.spec.ts b/packages/tests/tests/peer_exchange.node.spec.ts
index b0eee79ea9..996e1e1adc 100644
--- 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");