83 lines
2.0 KiB
Nim
83 lines
2.0 KiB
Nim
|
import std/sequtils
|
||
|
|
import pkg/chronos
|
||
|
|
import pkg/asynctest
|
||
|
|
import pkg/libp2p
|
||
|
|
import pkg/stew/byteutils
|
||
|
|
|
||
|
|
import pkg/dagger/p2p/rng
|
||
|
|
import pkg/dagger/stores/memorystore
|
||
|
|
import pkg/dagger/chunker
|
||
|
|
|
||
|
|
import ./helpers
|
||
|
|
|
||
|
|
suite "Memory Store":
|
||
|
|
|
||
|
|
var store: MemoryStore
|
||
|
|
var chunker = newRandomChunker(Rng.instance(), size = 1024, chunkSize = 256)
|
||
|
|
var blocks = chunker.mapIt( Block.new(it) )
|
||
|
|
|
||
|
|
setup:
|
||
|
|
store = MemoryStore.new(blocks)
|
||
|
|
|
||
|
|
test "getBlocks single":
|
||
|
|
let blk = await store.getBlocks(@[blocks[0].cid])
|
||
|
|
check blk[0] == blocks[0]
|
||
|
|
|
||
|
|
test "getBlocks multiple":
|
||
|
|
let blk = await store.getBlocks(blocks[0..2].mapIt( it.cid ))
|
||
|
|
check blk == blocks[0..2]
|
||
|
|
|
||
|
|
test "hasBlock":
|
||
|
|
check store.hasBlock(blocks[0].cid)
|
||
|
|
|
||
|
|
test "delBlocks single":
|
||
|
|
let blks = blocks[1..3].mapIt( it.cid )
|
||
|
|
store.delBlocks(blks)
|
||
|
|
|
||
|
|
check not store.hasBlock(blks[0])
|
||
|
|
check not store.hasBlock(blks[1])
|
||
|
|
check not store.hasBlock(blks[2])
|
||
|
|
|
||
|
|
test "add blocks change handler":
|
||
|
|
let blocks = @[
|
||
|
|
Block.new("Block 1".toBytes),
|
||
|
|
Block.new("Block 2".toBytes),
|
||
|
|
Block.new("Block 3".toBytes),
|
||
|
|
]
|
||
|
|
|
||
|
|
var triggered = false
|
||
|
|
store.addChangeHandler(
|
||
|
|
proc(evt: BlockStoreChangeEvt) =
|
||
|
|
check evt.kind == ChangeType.Added
|
||
|
|
check evt.cids == blocks.mapIt( it.cid )
|
||
|
|
triggered = true
|
||
|
|
, ChangeType.Added
|
||
|
|
)
|
||
|
|
|
||
|
|
store.putBlocks(blocks)
|
||
|
|
check triggered
|
||
|
|
|
||
|
|
test "add blocks change handler":
|
||
|
|
let blocks = @[
|
||
|
|
Block.new("Block 1".toBytes),
|
||
|
|
Block.new("Block 2".toBytes),
|
||
|
|
Block.new("Block 3".toBytes),
|
||
|
|
]
|
||
|
|
|
||
|
|
var triggered = false
|
||
|
|
store.addChangeHandler(
|
||
|
|
proc(evt: BlockStoreChangeEvt) =
|
||
|
|
check evt.kind == ChangeType.Removed
|
||
|
|
check evt.cids == blocks.mapIt( it.cid )
|
||
|
|
triggered = true
|
||
|
|
, ChangeType.Removed
|
||
|
|
)
|
||
|
|
|
||
|
|
store.putBlocks(blocks)
|
||
|
|
check store.hasBlock(blocks[0].cid)
|
||
|
|
check store.hasBlock(blocks[1].cid)
|
||
|
|
check store.hasBlock(blocks[2].cid)
|
||
|
|
|
||
|
|
store.delBlocks(blocks.mapIt( it.cid ))
|
||
|
|
check triggered
|