From b0746d57ec871f16d9e6e67c8bf4bfd3ef3a7d68 Mon Sep 17 00:00:00 2001
From: DarshanBPatel <darshan@status.im>
Date: Tue, 13 Aug 2024 18:50:56 +0530
Subject: [PATCH] chore: update the PR

---
 .../peer_exchange_tester.nim                  | 61 ++++++++++---------
 1 file changed, 33 insertions(+), 28 deletions(-)

diff --git a/apps/peer_exchange_tester/peer_exchange_tester.nim b/apps/peer_exchange_tester/peer_exchange_tester.nim
index 07774f4cc..660b93829 100644
--- a/apps/peer_exchange_tester/peer_exchange_tester.nim
+++ b/apps/peer_exchange_tester/peer_exchange_tester.nim
@@ -1,55 +1,60 @@
 import random, chronos, chronicles, stew/byteutils, os
 import libp2p/peerId
+import
+  waku/[
+    waku_peer_exchange,
+    node/peer_manager,
+    node/peer_manager/peer_store/waku_peer_storage,
+    node/peer_manager/peer_manager,
+  ]
 import waku/[waku_peer_exchange, node/peer_manager]
 import tests/testlib/wakucore
 
-proc connectToPeer(sw: Switch, peerAddr: string) {.async.} =
+proc connectToPeer(sw: Switch, peerAddr: string, peerId: string) {.async.} =
   let ma = MultiAddress.init(peerAddr).tryGet()
-  let str = "16Uiu2HAkykgaECHswi3YKJ5dMLbq2kPVCo89fcyTd38UcQD6ej5W"
-  # let peerIdData = cast[seq[byte]](str)
-  let peerId = PeerId.init(str).tryGet()
+  let peerId = PeerId.init(peerId).tryGet()
   await sw.connect(peerId, @[ma])
-  discard sw.dial(peerId, @[ma], "/vac/waku/peer-exchange/2.0.0-alpha1")
-  echo "Connected to peer with address: ", peerAddr
+  let conn = await sw.dial(peerId, @[ma], "/vac/waku/peer-exchange/2.0.0-alpha1")
+  if conn.isNil:
+    echo "--------> Connection failed: No connection returned."
+  else:
+    echo "--------> Connection successful: "
 
 proc main() {.async.} =
   echo "main started"
   let addrs = "/ip4/178.128.141.171/tcp/30303/"
+  let id = "16Uiu2HAkykgaECHswi3YKJ5dMLbq2kPVCo89fcyTd38UcQD6ej5W"
   let switch = newStandardSwitch()
   discard switch.start()
-  echo "\n ---------------------------------- switch started ----------------------------------------- \n"
-  await connectToPeer(switch, addrs)
+
+  await connectToPeer(switch, addrs, id)
   if len(switch.connectedPeers(Direction.Out)) > 0:
     echo "Successfully out connected with Waku test fleet"
   elif len(switch.connectedPeers(Direction.In)) > 0:
     echo "Successfully in connected with Waku test fleet"
   else:
     echo "Not connected to any peers"
-  
+
   echo "\n ---------------------------------- connection completed -------------------------------- \n"
 
-  let peerManager = PeerManager.new(switch)
+  let database = SqliteDatabase.new(":memory:").tryGet()
+  let storage = WakuPeerStorage.new(database).tryGet()
+  let peerManager = PeerManager.new(switch = switch, storage = storage)
   let wakuPx = WakuPeerExchange(peerManager: peerManager)
-  
-  echo "----------------------------------------------------------------------------------------------"
+
   for i in 1 .. 20:
     echo "Seq No :- " & $i & " ---> "
-    echo $wakuPx.enrCache 
+    echo $wakuPx.enrCache
+
+    let res1 = await wakuPx.request(1)
+    if res1.isOk:
+      echo "response count :- " & $res1.get().peerInfos.len
+    else:
+        echo "request isn't ok"
+    
     sleep(120000)
-  # let res1 = await wakuPx.request(1)
-  # echo "\n -------------------------------- response -------------------------------------\n"
-  # if res1.isOk:
-      # echo "response count :- " & $res1.get().peerInfos.len
-  # echo "\n -------------- requested 4 peer through px protocol ------------------------------------\n"
-  # echo res1
-  # for i in 1 .. 20: 
-  #   echo "Request no :- " & $i & " ------> "
-  #   let res1 = await wakuPx.request(5)
-  #   if res1.get().peerInfos.len == 5:
-  #     echo "It's 5 peers"
-  #   else:
-  #     echo "it isn't 5 peers" 
-  #   sleep(300000)
-  echo "---------------------------- Done ------------------------------- " 
+
+  echo "---------------------------- Done ------------------------------- "
+
 when isMainModule:
   waitFor main()