85 lines
2.1 KiB
Nim
85 lines
2.1 KiB
Nim
import std/os
|
|
import std/sequtils
|
|
|
|
import pkg/chronos
|
|
import pkg/asynctest
|
|
|
|
import pkg/codex/rng
|
|
import pkg/codex/streams
|
|
import pkg/codex/storageproofs as st
|
|
import pkg/codex/blocktype as bt
|
|
|
|
import ../helpers
|
|
|
|
const
|
|
SectorSize = 31
|
|
SectorsPerBlock = BlockSize div SectorSize
|
|
DataSetSize = BlockSize * 100
|
|
|
|
suite "Test PoR store":
|
|
let
|
|
(path, _, _) = instantiationInfo(-2, fullPaths = true) # get this file's name
|
|
blocks = toSeq([1, 5, 10, 14, 20, 12, 22]) # TODO: maybe make them random
|
|
|
|
var
|
|
chunker: RandomChunker
|
|
manifest: Manifest
|
|
store: BlockStore
|
|
ssk: st.SecretKey
|
|
spk: st.PublicKey
|
|
repoDir: string
|
|
stpstore: st.StpStore
|
|
por: PoR
|
|
porMsg: PorMessage
|
|
cid: Cid
|
|
tags: seq[Tag]
|
|
|
|
setupAll:
|
|
chunker = RandomChunker.new(Rng.instance(), size = DataSetSize, chunkSize = BlockSize)
|
|
store = CacheStore.new(cacheSize = DataSetSize, chunkSize = BlockSize)
|
|
manifest = Manifest.new(blockSize = BlockSize).tryGet()
|
|
(spk, ssk) = st.keyGen()
|
|
|
|
while (
|
|
let chunk = await chunker.getBytes();
|
|
chunk.len > 0):
|
|
|
|
let
|
|
blk = bt.Block.new(chunk).tryGet()
|
|
|
|
manifest.add(blk.cid)
|
|
if not (await store.putBlock(blk)):
|
|
raise newException(CatchableError, "Unable to store block " & $blk.cid)
|
|
|
|
cid = manifest.cid.tryGet()
|
|
por = await PoR.init(
|
|
StoreStream.new(store, manifest),
|
|
ssk, spk,
|
|
BlockSize)
|
|
|
|
porMsg = por.toMessage()
|
|
tags = blocks.mapIt(
|
|
Tag(idx: it, tag: porMsg.authenticators[it]) )
|
|
|
|
repoDir = path.parentDir / "stp"
|
|
createDir(repoDir)
|
|
stpstore = st.StpStore.init(repoDir)
|
|
|
|
teardownAll:
|
|
removeDir(repoDir)
|
|
|
|
test "Should store Storage Proofs":
|
|
check (await stpstore.store(por.toMessage(), cid)).isOk
|
|
check fileExists(stpstore.stpPath(cid) / "por")
|
|
|
|
test "Should retrieve Storage Proofs":
|
|
check (await stpstore.retrieve(cid)).tryGet() == porMsg
|
|
|
|
test "Should store tags":
|
|
check (await stpstore.store(tags, cid)).isOk
|
|
for t in tags:
|
|
check fileExists(stpstore.stpPath(cid) / $t.idx )
|
|
|
|
test "Should retrieve tags":
|
|
check (await stpstore.retrieve(cid, blocks)).tryGet() == tags
|