From 4d7e773c429ba94563f88136375935dcd37420ea Mon Sep 17 00:00:00 2001
From: Csaba Kiraly <csaba.kiraly@gmail.com>
Date: Wed, 16 Mar 2022 21:32:21 +0100
Subject: [PATCH] fix tests: make bootstrap nodes async

---
 tests/dht/test_providers.nim | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tests/dht/test_providers.nim b/tests/dht/test_providers.nim
index e952923..748930a 100644
--- a/tests/dht/test_providers.nim
+++ b/tests/dht/test_providers.nim
@@ -28,9 +28,9 @@ import
 
 proc bootstrapNodes(
     nodecount: int,
-    bootnodes: openArray[SignedPeerRecord],
+    bootnodes: seq[SignedPeerRecord],
     rng = keys.newRng()
-  ) : seq[(discv5_protocol.Protocol, keys.PrivateKey)] =
+  ) : Future[seq[(discv5_protocol.Protocol, keys.PrivateKey)]] {.async.} =
 
   for i in 0..<nodecount:
     let privKey = keys.PrivateKey.random(rng[])
@@ -44,7 +44,7 @@ proc bootstrapNodes(
 proc bootstrapNetwork(
     nodecount: int,
     rng = keys.newRng()
-  ) : seq[(discv5_protocol.Protocol, keys.PrivateKey)] =
+  ) : Future[seq[(discv5_protocol.Protocol, keys.PrivateKey)]] {.async.} =
 
   let
     bootNodeKey = keys.PrivateKey.fromHex(
@@ -54,7 +54,7 @@ proc bootstrapNetwork(
 
   #waitFor bootNode.bootstrap()  # immediate, since no bootnodes are defined above
 
-  var res = bootstrapNodes(nodecount - 1,
+  var res = await bootstrapNodes(nodecount - 1,
                            @[bootnode.localNode.record],
                            rng)
   res.insert((bootNode, bootNodeKey), 0)
@@ -80,7 +80,7 @@ suite "Providers Tests: node alone":
 
   setupAll:
     rng = keys.newRng()
-    nodes = bootstrapNetwork(nodecount=1)
+    nodes = await bootstrapNetwork(nodecount=1)
     targetId = toNodeId(keys.PrivateKey.random(rng[]).toPublicKey)
     (node0, privKey_keys0) = nodes[0]
     privKey0 = privKey_keys0.pkToPk.get
@@ -142,7 +142,7 @@ suite "Providers Tests: two nodes":
 
   setupAll:
     rng = keys.newRng()
-    nodes = bootstrapNetwork(nodecount=2)
+    nodes = await bootstrapNetwork(nodecount=3)
     targetId = toNodeId(keys.PrivateKey.random(rng[]).toPublicKey)
     (node0, privKey_keys0) = nodes[0]
     privKey0 = privKey_keys0.pkToPk.get
@@ -195,7 +195,7 @@ suite "Providers Tests: 20 nodes":
 
   setupAll:
     rng = keys.newRng()
-    nodes = bootstrapNetwork(nodecount=20)
+    nodes = await bootstrapNetwork(nodecount=20)
     targetId = toNodeId(keys.PrivateKey.random(rng[]).toPublicKey)
     (node0, privKey_keys0) = nodes[0]
     privKey0 = privKey_keys0.pkToPk.get