2022-09-16 21:15:08 -06:00
|
|
|
import std/options
|
|
|
|
|
import std/os
|
2022-09-20 16:41:54 -04:00
|
|
|
import std/sequtils
|
|
|
|
|
from std/algorithm import sort, reversed
|
2022-09-16 21:15:08 -06:00
|
|
|
|
2023-09-20 22:12:53 -07:00
|
|
|
import pkg/unittest2
|
2022-09-16 21:15:08 -06:00
|
|
|
import pkg/chronos
|
|
|
|
|
import pkg/stew/results
|
|
|
|
|
import pkg/stew/byteutils
|
|
|
|
|
|
|
|
|
|
import pkg/datastore/sql/sqliteds
|
2023-09-20 22:12:53 -07:00
|
|
|
import pkg/datastore/key
|
2022-09-16 21:15:08 -06:00
|
|
|
|
2023-09-27 18:06:23 -07:00
|
|
|
import ../backendCommonTests
|
2022-09-19 15:53:38 -06:00
|
|
|
|
2023-09-20 22:49:11 -07:00
|
|
|
|
|
|
|
|
suite "Test Basic SQLiteDatastore":
|
|
|
|
|
let
|
2023-09-25 17:49:17 -07:00
|
|
|
ds = newSQLiteBackend[string, seq[byte]](path=Memory).tryGet()
|
2023-09-20 22:49:11 -07:00
|
|
|
keyFull = Key.init("a:b/c/d:e").tryGet()
|
|
|
|
|
key = keyFull.id()
|
|
|
|
|
bytes = "some bytes".toBytes
|
|
|
|
|
otherBytes = "some other bytes".toBytes
|
|
|
|
|
|
2023-09-20 23:16:24 -07:00
|
|
|
var batch: seq[tuple[key: string, data: seq[byte]]]
|
|
|
|
|
for k in 0..<100:
|
|
|
|
|
let kk = Key.init(key, $k).tryGet().id()
|
|
|
|
|
batch.add( (kk, @[k.byte]) )
|
|
|
|
|
|
2023-09-20 22:49:11 -07:00
|
|
|
suiteTeardown:
|
|
|
|
|
ds.close().tryGet()
|
2023-09-20 23:07:52 -07:00
|
|
|
|
2023-09-27 18:06:23 -07:00
|
|
|
testBasicBackend(ds, key, bytes, otherBytes, batch)
|
2023-09-20 23:07:52 -07:00
|
|
|
|
2023-09-20 23:28:50 -07:00
|
|
|
suite "Test DataBuffer SQLiteDatastore":
|
|
|
|
|
let
|
2023-09-25 17:35:37 -07:00
|
|
|
ds = newSQLiteBackend[KeyId, DataBuffer](Memory).tryGet()
|
2023-09-20 23:28:50 -07:00
|
|
|
keyFull = Key.init("a:b/c/d:e").tryGet()
|
|
|
|
|
key = KeyId.new keyFull.id()
|
|
|
|
|
bytes = DataBuffer.new "some bytes"
|
|
|
|
|
otherBytes = DataBuffer.new "some other bytes"
|
|
|
|
|
|
|
|
|
|
var batch: seq[tuple[key: KeyId, data: DataBuffer]]
|
|
|
|
|
for k in 0..<100:
|
|
|
|
|
let kk = Key.init(keyFull.id(), $k).tryGet().id()
|
|
|
|
|
batch.add( (KeyId.new kk, DataBuffer.new @[k.byte]) )
|
2023-09-20 23:16:24 -07:00
|
|
|
|
2023-09-20 23:28:50 -07:00
|
|
|
suiteTeardown:
|
|
|
|
|
ds.close().tryGet()
|
2023-09-20 23:44:09 -07:00
|
|
|
|
2023-09-27 18:06:23 -07:00
|
|
|
testBasicBackend(ds, key, bytes, otherBytes, batch)
|
2023-09-21 18:49:08 -07:00
|
|
|
|
|
|
|
|
suite "queryTests":
|
|
|
|
|
|
2023-09-27 18:11:28 -07:00
|
|
|
let
|
|
|
|
|
dsNew = proc(): SQLiteBackend[KeyId, DataBuffer] =
|
|
|
|
|
newSQLiteBackend[KeyId, DataBuffer](Memory).tryGet()
|
|
|
|
|
key1 = KeyId.new "/a"
|
|
|
|
|
key2 = KeyId.new "/a/b"
|
|
|
|
|
key3 = KeyId.new "/a/b/c"
|
|
|
|
|
val1 = DataBuffer.new "value for 1"
|
|
|
|
|
val2 = DataBuffer.new "value for 2"
|
|
|
|
|
val3 = DataBuffer.new "value for 3"
|
|
|
|
|
|
2023-09-27 18:57:17 -07:00
|
|
|
queryTests(dsNew, key1, key2, key3, val1, val2, val3, extended=true)
|