mirror of
https://github.com/logos-storage/nim-datastore.git
synced 2026-01-03 14:13:09 +00:00
105 lines
2.2 KiB
Nim
105 lines
2.2 KiB
Nim
import std/options
|
|
import std/os
|
|
import std/sequtils
|
|
from std/algorithm import sort, reversed
|
|
|
|
import pkg/unittest2
|
|
import pkg/chronos
|
|
import pkg/stew/results
|
|
import pkg/stew/byteutils
|
|
|
|
import pkg/datastore/sql/sqliteds
|
|
import pkg/datastore/key
|
|
|
|
import ../dscommontests
|
|
import ../querycommontests
|
|
|
|
proc testBasic[K, V, B](
|
|
ds: SQLiteDatastore,
|
|
key: K,
|
|
bytes: V,
|
|
otherBytes: V,
|
|
batches: B,
|
|
) =
|
|
|
|
test "put":
|
|
ds.put(key, bytes).tryGet()
|
|
|
|
test "get":
|
|
check:
|
|
ds.get(key).tryGet() == bytes
|
|
|
|
test "put update":
|
|
ds.put(key, otherBytes).tryGet()
|
|
|
|
test "get updated":
|
|
check:
|
|
ds.get(key).tryGet() == otherBytes
|
|
|
|
test "delete":
|
|
ds.delete(key).tryGet()
|
|
|
|
test "contains":
|
|
check key notin ds
|
|
|
|
test "put batch":
|
|
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]))
|
|
|
|
ds.put(batch).tryGet
|
|
|
|
for k in batch:
|
|
check: ds.has(k.key).tryGet
|
|
|
|
test "delete batch":
|
|
var
|
|
batch: seq[string]
|
|
|
|
for k in 0..<100:
|
|
batch.add(Key.init(key, $k).tryGet().id())
|
|
|
|
ds.delete(batch).tryGet
|
|
|
|
for k in batch:
|
|
check: not ds.has(k).tryGet
|
|
|
|
test "handle missing key":
|
|
let key = Key.init("/missing/key").tryGet().id()
|
|
|
|
expect(DatastoreKeyNotFound):
|
|
discard ds.get(key).tryGet() # non existing key
|
|
|
|
suite "Test Basic SQLiteDatastore":
|
|
let
|
|
ds = SQLiteDatastore.new(Memory).tryGet()
|
|
keyFull = Key.init("a:b/c/d:e").tryGet()
|
|
key = keyFull.id()
|
|
bytes = "some bytes".toBytes
|
|
otherBytes = "some other bytes".toBytes
|
|
|
|
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]) )
|
|
|
|
suiteTeardown:
|
|
ds.close().tryGet()
|
|
|
|
testBasic(ds, key, bytes, otherBytes, batch)
|
|
|
|
# suite "Test DataBuffer SQLiteDatastore":
|
|
# let
|
|
# ds = SQLiteDatastore.new(Memory).tryGet()
|
|
# key = KeyId.new Key.init("a:b/c/d:e").tryGet().id()
|
|
# bytes = DataBuffer.new "some bytes"
|
|
# otherBytes = DataBuffer.new "some other bytes"
|
|
|
|
# suiteTeardown:
|
|
# ds.close().tryGet()
|
|
|
|
# testBasic(ds, key, bytes, otherBytes)
|