From e8e6df767b90334e7382dcd27f849f3329a78d14 Mon Sep 17 00:00:00 2001
From: Franck Royer <franck@status.im>
Date: Wed, 1 Jun 2022 13:16:06 +1000
Subject: [PATCH] Use waitForRemotePeer directly

---
 src/lib/waku_store/index.node.spec.ts | 37 +++------------------------
 1 file changed, 4 insertions(+), 33 deletions(-)

diff --git a/src/lib/waku_store/index.node.spec.ts b/src/lib/waku_store/index.node.spec.ts
index 187ebdde7d..76669dd68f 100644
--- a/src/lib/waku_store/index.node.spec.ts
+++ b/src/lib/waku_store/index.node.spec.ts
@@ -7,7 +7,6 @@ import {
   NOISE_KEY_2,
   Nwaku,
 } from "../../test_utils";
-import { delay } from "../../test_utils/delay";
 import {
   generatePrivateKey,
   generateSymmetricKey,
@@ -278,14 +277,7 @@ describe("Waku Store", () => {
 
     dbg("Waku nodes connected to nwaku");
 
-    let lightPushPeerFound = false;
-    while (!lightPushPeerFound) {
-      await delay(100);
-      for await (const _peer of waku1.lightPush.peers) {
-        lightPushPeerFound = true;
-        break;
-      }
-    }
+    await waku1.waitForRemotePeer([Protocols.LightPush]);
 
     dbg("Sending messages using light push");
     await Promise.all([
@@ -295,14 +287,7 @@ describe("Waku Store", () => {
       waku1.lightPush.push(clearMessage),
     ]);
 
-    let storePeerFound = false;
-    while (!storePeerFound) {
-      await delay(100);
-      for await (const _peer of waku2.store.peers) {
-        storePeerFound = true;
-        break;
-      }
-    }
+    await waku2.waitForRemotePeer([Protocols.Store]);
 
     waku2.store.addDecryptionKey(symKey);
 
@@ -396,14 +381,7 @@ describe("Waku Store", () => {
 
     dbg("Waku nodes connected to nwaku");
 
-    let lightPushPeerFound = false;
-    while (!lightPushPeerFound) {
-      await delay(100);
-      for await (const _peer of waku1.lightPush.peers) {
-        lightPushPeerFound = true;
-        break;
-      }
-    }
+    await waku1.waitForRemotePeer([Protocols.LightPush]);
 
     dbg("Sending messages using light push");
     await Promise.all([
@@ -413,14 +391,7 @@ describe("Waku Store", () => {
       waku1.lightPush.push(clearMessage),
     ]);
 
-    let storePeerFound = false;
-    while (!storePeerFound) {
-      await delay(100);
-      for await (const _peer of waku2.store.peers) {
-        storePeerFound = true;
-        break;
-      }
-    }
+    await waku2.waitForRemotePeer([Protocols.Store]);
 
     waku2.addDecryptionKey(symKey, {
       contentTopics: [encryptedSymmetricContentTopic],