From 907b4d6fdbf3ab9e25e43e45d91029ec625ab040 Mon Sep 17 00:00:00 2001
From: Gabriel mermelstein <gabrielmermelstein@gmail.com>
Date: Thu, 9 May 2024 17:36:40 +0200
Subject: [PATCH] fixing test_rest_store.nim

---
 tests/wakunode_rest/test_rest_store.nim | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/wakunode_rest/test_rest_store.nim b/tests/wakunode_rest/test_rest_store.nim
index 345e35f497..3d699c38a8 100644
--- a/tests/wakunode_rest/test_rest_store.nim
+++ b/tests/wakunode_rest/test_rest_store.nim
@@ -123,7 +123,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("c2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -195,7 +195,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 09], ts = ts(90, timeOrigin)),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -282,7 +282,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -354,7 +354,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("ct2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -442,7 +442,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("ct2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -513,7 +513,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("ct2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     # Sending no peer-store node address
     response = await client.getStoreMessagesV3(
@@ -560,7 +560,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 9], contentTopic = ContentTopic("ct2"), ts = 9),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -616,7 +616,7 @@ procSuite "Waku Rest API - Store v3":
     let msg = fakeWakuMessage(
       @[byte 0], contentTopic = ContentTopic("ct1"), ts = 0, meta = (@[byte 8])
     )
-    require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+    require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))
 
@@ -682,7 +682,7 @@ procSuite "Waku Rest API - Store v3":
         fakeWakuMessage(@[byte 09], ts = ts(90, timeOrigin)),
       ]
     for msg in msgList:
-      require (waitFor driver.put(DefaultPubsubTopic, msg)).isOk()
+      require (await driver.put(DefaultPubsubTopic, msg)).isOk()
 
     let client = newRestHttpClient(initTAddress(restAddress, restPort))