From e20536ba5957ed993adc86bf319650608cf19122 Mon Sep 17 00:00:00 2001
From: Jaremy Creechley <creechley@gmail.com>
Date: Wed, 30 Aug 2023 18:21:59 -0700
Subject: [PATCH] fix test

---
 tests/datastore/testdatastore.nim | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/datastore/testdatastore.nim b/tests/datastore/testdatastore.nim
index 4361eef..cd52915 100644
--- a/tests/datastore/testdatastore.nim
+++ b/tests/datastore/testdatastore.nim
@@ -25,5 +25,6 @@ suite "Datastore (base)":
 
   test "query":
     expect Defect:
-      let iter = (await ds.query(Query.init(key))).tryGet
-      for n in iter: discard
+      let iter = tryGet(await waitForAllQueryResults ds.query(Query.init(key)))
+      for n in iter:
+        discard