logos-storage/logos-storage-nim
1
0
Fork 0
mirror of https://github.com/logos-storage/logos-storage-nim.git synced 2026-01-15 03:43:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adds tests for pendingblock inFlight logic

Browse Source
This commit is contained in:
thatben 2025-01-31 13:03:03 +01:00
parent 88522d9093
commit 870f6ccd8b
No known key found for this signature in database
GPG Key ID: 62C543548433D43E
1 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
tests/codex/blockexchange/testpendingblocks.nim
View File

@ -79,3 +79,46 @@ checksuite "Pending Blocks":
check:
(await allFinished(wantHandles)).mapIt($it.read.cid).sorted(cmp[string]) ==
(await allFinished(handles)).mapIt($it.read.cid).sorted(cmp[string])
checksuite "Pending Blocks - inFlight":
var
pendingBlocks: PendingBlocksManager
blk: bt.Block
addrs: BlockAddress
setup:
pendingBlocks = PendingBlocksManager.new()
blk = bt.Block.new("Hello".toBytes).tryGet
addrs = BlockAddress.init(blk.cid)
discard pendingBlocks.getWantHandle(addrs)
proc req(): BlockReq =
pendingBlocks.blocks[addrs]
test "Sets inFlight (single)":
pendingBlocks.setInFlight(addrs)
check:
req().inFlight
test "Sets inFlight (multiple)":
pendingBlocks.setInFlight(@[addrs])
check:
req().inFlight
test "IsInFlight":
pendingBlocks.setInFlight(addrs)
check:
pendingBlocks.isInFlight(addrs)
test "getNotInFlight":
let
blk2 = bt.Block.new("Hello2".toBytes).tryGet
addrs2 = BlockAddress.init(blk2.cid)
pendingBlocks.setInFlight(addrs)
check:
pendingBlocks.getNotInFlight(@[addrs, addrs2]) == @[addrs2]