From 1f23fe57d3a561f24089c33521c8cd92f5c57a5f Mon Sep 17 00:00:00 2001
From: Jaremy Creechley <creechley@gmail.com>
Date: Mon, 25 Sep 2023 19:03:59 -0700
Subject: [PATCH] query fixes

---
 tests/datastore/sql/testsqliteds.nim | 31 +++++++++++++++-------------
 1 file changed, 17 insertions(+), 14 deletions(-)

diff --git a/tests/datastore/sql/testsqliteds.nim b/tests/datastore/sql/testsqliteds.nim
index 66c9209..0ea2364 100644
--- a/tests/datastore/sql/testsqliteds.nim
+++ b/tests/datastore/sql/testsqliteds.nim
@@ -20,6 +20,7 @@ proc testBasic[K, V](
   bytes: V,
   otherBytes: V,
   batch: seq[DbBatchEntry[K, V]],
+  extended = true
 ) =
 
   test "put":
@@ -250,24 +251,26 @@ suite "queryTests":
       res[2].key.get == key3
       res[2].data == val3
 
-  # test "Should apply limit":
-  #   let
-  #     key = Key.init("/a").tryGet
-  #     q = DbQuery(key: key1, value: false)
+  test "Should apply limit":
+    let
+      key = Key.init("/a").tryGet
+      q = DbQuery[KeyId](key: key1, limit: 10, value: false)
 
-  #   for i in 0..<100:
-  #     let
-  #       key = KeyId.new $Key.init(key, Key.init("/" & $i).tryGet).tryGet
-  #       val = DataBuffer.new("val " & $i)
+    for i in 0..<100:
+      let
+        key = KeyId.new $Key.init(key, Key.init("/" & $i).tryGet).tryGet
+        val = DataBuffer.new("val " & $i)
 
-  #     ds.put(key, val).tryGet
+      ds.put(key, val).tryGet
 
-  #   let
-  #     (handle, iter) = ds.query(q).tryGet
-  #     res = iter.mapIt(it.tryGet())
+    var
+      handle  = ds.query(q).tryGet
+    let
+      res = handle.iter().toSeq().mapIt(it.tryGet()).reversed()
 
-  #   check:
-  #     res.len == 10
+    echo "RES: ", res.mapIt(it.key)
+    check:
+      res.len == 10
 
   # test "Should not apply offset":
   #   let