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 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) (await store.putBlock(blk)).tryGet() 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 = getAppDir() / "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