uncomment engine tests
This commit is contained in:
gmega
parent
a274d24583
commit
102ecdf571
|
|
@ -20,110 +20,111 @@ import ../../../asynctest
|
|||
import ../../helpers
|
||||
import ../../examples
|
||||
|
||||
# asyncchecksuite "NetworkStore engine basic":
|
||||
# var
|
||||
# rng: Rng
|
||||
# seckey: PrivateKey
|
||||
# peerId: PeerId
|
||||
# chunker: Chunker
|
||||
# wallet: WalletRef
|
||||
# blockDiscovery: Discovery
|
||||
# peerStore: PeerCtxStore
|
||||
# pendingBlocks: PendingBlocksManager
|
||||
# blocks: seq[Block]
|
||||
# done: Future[void]
|
||||
asyncchecksuite "NetworkStore engine basic":
|
||||
var
|
||||
rng: Rng
|
||||
seckey: PrivateKey
|
||||
peerId: PeerId
|
||||
chunker: Chunker
|
||||
wallet: WalletRef
|
||||
blockDiscovery: Discovery
|
||||
peerStore: PeerCtxStore
|
||||
pendingBlocks: PendingBlocksManager
|
||||
blocks: seq[Block]
|
||||
done: Future[void]
|
||||
|
||||
# setup:
|
||||
# rng = Rng.instance()
|
||||
# seckey = PrivateKey.random(rng[]).tryGet()
|
||||
# peerId = PeerId.init(seckey.getPublicKey().tryGet()).tryGet()
|
||||
# chunker = RandomChunker.new(Rng.instance(), size = 1024'nb, chunkSize = 256'nb)
|
||||
# wallet = WalletRef.example
|
||||
# blockDiscovery = Discovery.new()
|
||||
# peerStore = PeerCtxStore.new()
|
||||
# pendingBlocks = PendingBlocksManager.new()
|
||||
setup:
|
||||
rng = Rng.instance()
|
||||
seckey = PrivateKey.random(rng[]).tryGet()
|
||||
peerId = PeerId.init(seckey.getPublicKey().tryGet()).tryGet()
|
||||
chunker = RandomChunker.new(Rng.instance(), size = 1024'nb, chunkSize = 256'nb)
|
||||
wallet = WalletRef.example
|
||||
blockDiscovery = Discovery.new()
|
||||
peerStore = PeerCtxStore.new()
|
||||
pendingBlocks = PendingBlocksManager.new()
|
||||
|
||||
# while true:
|
||||
# let chunk = await chunker.getBytes()
|
||||
# if chunk.len <= 0:
|
||||
# break
|
||||
while true:
|
||||
let chunk = await chunker.getBytes()
|
||||
if chunk.len <= 0:
|
||||
break
|
||||
|
||||
# blocks.add(Block.new(chunk).tryGet())
|
||||
blocks.add(Block.new(chunk).tryGet())
|
||||
|
||||
# done = newFuture[void]()
|
||||
done = newFuture[void]()
|
||||
|
||||
# test "Should send want list to new peers":
|
||||
# proc sendWantList(
|
||||
# id: PeerId,
|
||||
# addresses: seq[BlockAddress],
|
||||
# priority: int32 = 0,
|
||||
# cancel: bool = false,
|
||||
# wantType: WantType = WantType.WantHave,
|
||||
# full: bool = false,
|
||||
# sendDontHave: bool = false) {.gcsafe, async: (handleException: true).} =
|
||||
# check addresses.mapIt($it.cidOrTreeCid).sorted == blocks.mapIt($it.cid).sorted
|
||||
# done.complete()
|
||||
test "Should send want list to new peers":
|
||||
proc sendWantList(
|
||||
id: PeerId,
|
||||
addresses: seq[BlockAddress],
|
||||
priority: int32 = 0,
|
||||
cancel: bool = false,
|
||||
wantType: WantType = WantType.WantHave,
|
||||
full: bool = false,
|
||||
sendDontHave: bool = false) {.gcsafe, async: (handleException: true).} =
|
||||
check addresses.mapIt($it.cidOrTreeCid).sorted == blocks.mapIt( $it.cid ).sorted
|
||||
done.complete()
|
||||
|
||||
# let
|
||||
# network = BlockExcNetwork(request: BlockExcRequest(
|
||||
# sendWantList: sendWantList,
|
||||
# ))
|
||||
let
|
||||
network = BlockExcNetwork(request: BlockExcRequest(
|
||||
sendWantList: sendWantList,
|
||||
))
|
||||
|
||||
# localStore = CacheStore.new(blocks.mapIt( it ))
|
||||
# discovery = DiscoveryEngine.new(
|
||||
# localStore,
|
||||
# peerStore,
|
||||
# network,
|
||||
# blockDiscovery,
|
||||
# pendingBlocks)
|
||||
localStore = CacheStore.new(blocks.mapIt( it ))
|
||||
discovery = DiscoveryEngine.new(
|
||||
localStore,
|
||||
peerStore,
|
||||
network,
|
||||
blockDiscovery,
|
||||
pendingBlocks)
|
||||
|
||||
# engine = BlockExcEngine.new(
|
||||
# localStore,
|
||||
# wallet,
|
||||
# network,
|
||||
# discovery,
|
||||
# peerStore,
|
||||
# pendingBlocks)
|
||||
engine = BlockExcEngine.new(
|
||||
localStore,
|
||||
wallet,
|
||||
network,
|
||||
discovery,
|
||||
peerStore,
|
||||
pendingBlocks)
|
||||
|
||||
# for b in blocks:
|
||||
# discard engine.pendingBlocks.getWantHandle(b.cid)
|
||||
# await engine.setupPeer(peerId)
|
||||
for b in blocks:
|
||||
discard engine.pendingBlocks.getWantHandle(b.cid)
|
||||
await engine.setupPeer(peerId)
|
||||
|
||||
# await done.wait(100.millis)
|
||||
await done.wait(100.millis)
|
||||
|
||||
# test "Should send account to new peers":
|
||||
# let pricing = Pricing.example
|
||||
test "Should send account to new peers":
|
||||
let pricing = Pricing.example
|
||||
|
||||
# proc sendAccount(peer: PeerId, account: Account) {.gcsafe, async: (handleException: true).} =
|
||||
# check account.address == pricing.address
|
||||
# done.complete()
|
||||
proc sendAccount(peer: PeerId,
|
||||
account: Account) {.gcsafe, async: (handleException: true).} =
|
||||
check account.address == pricing.address
|
||||
done.complete()
|
||||
|
||||
# let
|
||||
# network = BlockExcNetwork(
|
||||
# request: BlockExcRequest(
|
||||
# sendAccount: sendAccount
|
||||
# ))
|
||||
let
|
||||
network = BlockExcNetwork(
|
||||
request: BlockExcRequest(
|
||||
sendAccount: sendAccount
|
||||
))
|
||||
|
||||
# localStore = CacheStore.new()
|
||||
# discovery = DiscoveryEngine.new(
|
||||
# localStore,
|
||||
# peerStore,
|
||||
# network,
|
||||
# blockDiscovery,
|
||||
# pendingBlocks)
|
||||
localStore = CacheStore.new()
|
||||
discovery = DiscoveryEngine.new(
|
||||
localStore,
|
||||
peerStore,
|
||||
network,
|
||||
blockDiscovery,
|
||||
pendingBlocks)
|
||||
|
||||
# engine = BlockExcEngine.new(
|
||||
# localStore,
|
||||
# wallet,
|
||||
# network,
|
||||
# discovery,
|
||||
# peerStore,
|
||||
# pendingBlocks)
|
||||
engine = BlockExcEngine.new(
|
||||
localStore,
|
||||
wallet,
|
||||
network,
|
||||
discovery,
|
||||
peerStore,
|
||||
pendingBlocks)
|
||||
|
||||
# engine.pricing = pricing.some
|
||||
# await engine.setupPeer(peerId)
|
||||
engine.pricing = pricing.some
|
||||
await engine.setupPeer(peerId)
|
||||
|
||||
# await done.wait(100.millis)
|
||||
await done.wait(100.millis)
|
||||
|
||||
asyncchecksuite "NetworkStore engine handlers":
|
||||
var
|
||||
|
|
@ -183,90 +184,99 @@ asyncchecksuite "NetworkStore engine handlers":
|
|||
)
|
||||
engine.peers.add(peerCtx)
|
||||
|
||||
# test "Should schedule block requests":
|
||||
# let
|
||||
# wantList = makeWantList(
|
||||
# blocks.mapIt( it.cid ),
|
||||
# wantType = WantType.WantBlock) # only `wantBlock` are stored in `peerWants`
|
||||
test "Should schedule block requests":
|
||||
let
|
||||
wantList = makeWantList(
|
||||
blocks.mapIt( it.cid ),
|
||||
wantType = WantType.WantBlock) # only `wantBlock` are stored in `peerWants`
|
||||
|
||||
# proc handler() {.async: (handleException: true).} =
|
||||
# let ctx = await engine.taskQueue.pop()
|
||||
# check ctx.id == peerId
|
||||
# # only `wantBlock` scheduled
|
||||
# check ctx.peerWants.mapIt( it.address.cidOrTreeCid ) == blocks.mapIt( it.cid )
|
||||
proc handler() {.async: (handleException: true).} =
|
||||
let ctx = await engine.taskQueue.pop()
|
||||
check ctx.id == peerId
|
||||
# only `wantBlock` scheduled
|
||||
check ctx.peerWants.mapIt( it.address.cidOrTreeCid ) == blocks.mapIt( it.cid )
|
||||
|
||||
# let done = handler()
|
||||
# await engine.wantListHandler(peerId, wantList)
|
||||
# await done
|
||||
let done = handler()
|
||||
await engine.wantListHandler(peerId, wantList)
|
||||
await done
|
||||
|
||||
# test "Should handle want list":
|
||||
# let
|
||||
# done = newFuture[void]()
|
||||
# wantList = makeWantList(blocks.mapIt( it.cid ))
|
||||
test "Should handle want list":
|
||||
let
|
||||
done = newFuture[void]()
|
||||
wantList = makeWantList(blocks.mapIt( it.cid ))
|
||||
|
||||
# proc sendPresence(peerId: PeerId, presence: seq[BlockPresence]) {.gcsafe, async: (handleException: true).} =
|
||||
# check presence.mapIt( it.address ) == wantList.entries.mapIt( it.address )
|
||||
# done.complete()
|
||||
proc sendPresence(
|
||||
peerId: PeerId,
|
||||
presence: seq[BlockPresence]
|
||||
) {.gcsafe, async: (handleException: true).} =
|
||||
check presence.mapIt( it.address ) == wantList.entries.mapIt( it.address )
|
||||
done.complete()
|
||||
|
||||
# engine.network = BlockExcNetwork(
|
||||
# request: BlockExcRequest(
|
||||
# sendPresence: sendPresence
|
||||
# ))
|
||||
engine.network = BlockExcNetwork(
|
||||
request: BlockExcRequest(
|
||||
sendPresence: sendPresence
|
||||
))
|
||||
|
||||
# await allFuturesThrowing(
|
||||
# allFinished(blocks.mapIt( localStore.putBlock(it) )))
|
||||
await allFuturesThrowing(
|
||||
allFinished(blocks.mapIt( localStore.putBlock(it) )))
|
||||
|
||||
# await engine.wantListHandler(peerId, wantList)
|
||||
# await done
|
||||
await engine.wantListHandler(peerId, wantList)
|
||||
await done
|
||||
|
||||
# test "Should handle want list - `dont-have`":
|
||||
# let
|
||||
# done = newFuture[void]()
|
||||
# wantList = makeWantList(
|
||||
# blocks.mapIt( it.cid ),
|
||||
# sendDontHave = true)
|
||||
test "Should handle want list - `dont-have`":
|
||||
let
|
||||
done = newFuture[void]()
|
||||
wantList = makeWantList(
|
||||
blocks.mapIt( it.cid ),
|
||||
sendDontHave = true)
|
||||
|
||||
# proc sendPresence(peerId: PeerId, presence: seq[BlockPresence]) {.gcsafe, async: (handleException: true).} =
|
||||
# check presence.mapIt( it.address ) == wantList.entries.mapIt( it.address )
|
||||
# for p in presence:
|
||||
# check:
|
||||
# p.`type` == BlockPresenceType.DontHave
|
||||
proc sendPresence(
|
||||
peerId: PeerId,
|
||||
presence: seq[BlockPresence]
|
||||
) {.gcsafe, async: (handleException: true).} =
|
||||
check presence.mapIt( it.address ) == wantList.entries.mapIt( it.address )
|
||||
for p in presence:
|
||||
check:
|
||||
p.`type` == BlockPresenceType.DontHave
|
||||
|
||||
# done.complete()
|
||||
done.complete()
|
||||
|
||||
# engine.network = BlockExcNetwork(request: BlockExcRequest(
|
||||
# sendPresence: sendPresence
|
||||
# ))
|
||||
engine.network = BlockExcNetwork(request: BlockExcRequest(
|
||||
sendPresence: sendPresence
|
||||
))
|
||||
|
||||
# await engine.wantListHandler(peerId, wantList)
|
||||
# await done
|
||||
await engine.wantListHandler(peerId, wantList)
|
||||
await done
|
||||
|
||||
# test "Should handle want list - `dont-have` some blocks":
|
||||
# let
|
||||
# done = newFuture[void]()
|
||||
# wantList = makeWantList(
|
||||
# blocks.mapIt( it.cid ),
|
||||
# sendDontHave = true)
|
||||
test "Should handle want list - `dont-have` some blocks":
|
||||
let
|
||||
done = newFuture[void]()
|
||||
wantList = makeWantList(
|
||||
blocks.mapIt( it.cid ),
|
||||
sendDontHave = true)
|
||||
|
||||
# proc sendPresence(peerId: PeerId, presence: seq[BlockPresence]) {.gcsafe, async: (handleException: true).} =
|
||||
# for p in presence:
|
||||
# if p.address.cidOrTreeCid != blocks[0].cid and p.address.cidOrTreeCid != blocks[1].cid:
|
||||
# check p.`type` == BlockPresenceType.DontHave
|
||||
# else:
|
||||
# check p.`type` == BlockPresenceType.Have
|
||||
proc sendPresence(
|
||||
peerId: PeerId,
|
||||
presence: seq[BlockPresence]
|
||||
) {.gcsafe, async: (handleException: true).} =
|
||||
for p in presence:
|
||||
if p.address.cidOrTreeCid != blocks[0].cid and p.address.cidOrTreeCid != blocks[1].cid:
|
||||
check p.`type` == BlockPresenceType.DontHave
|
||||
else:
|
||||
check p.`type` == BlockPresenceType.Have
|
||||
|
||||
# done.complete()
|
||||
done.complete()
|
||||
|
||||
# engine.network = BlockExcNetwork(
|
||||
# request: BlockExcRequest(
|
||||
# sendPresence: sendPresence
|
||||
# ))
|
||||
engine.network = BlockExcNetwork(
|
||||
request: BlockExcRequest(
|
||||
sendPresence: sendPresence
|
||||
))
|
||||
|
||||
# (await engine.localStore.putBlock(blocks[0])).tryGet()
|
||||
# (await engine.localStore.putBlock(blocks[1])).tryGet()
|
||||
# await engine.wantListHandler(peerId, wantList)
|
||||
(await engine.localStore.putBlock(blocks[0])).tryGet()
|
||||
(await engine.localStore.putBlock(blocks[1])).tryGet()
|
||||
await engine.wantListHandler(peerId, wantList)
|
||||
|
||||
# await done
|
||||
await done
|
||||
|
||||
test "Should store blocks in local store":
|
||||
let pending = blocks.mapIt(
|
||||
|
|
@ -275,8 +285,6 @@ asyncchecksuite "NetworkStore engine handlers":
|
|||
|
||||
let blocksDelivery = blocks.mapIt(BlockDelivery(blk: it, address: it.address))
|
||||
|
||||
echo "BLD: ", $blocksDelivery[0]
|
||||
|
||||
await engine.blocksDeliveryHandler(peerId, blocksDelivery)
|
||||
let resolved = await allFinished(pending)
|
||||
check resolved.mapIt( it.read ) == blocks
|
||||
|
|
@ -284,221 +292,226 @@ asyncchecksuite "NetworkStore engine handlers":
|
|||
let present = await engine.localStore.hasBlock(b.cid)
|
||||
check present.tryGet()
|
||||
|
||||
# test "Should send payments for received blocks":
|
||||
# let
|
||||
# done = newFuture[void]()
|
||||
# account = Account(address: EthAddress.example)
|
||||
# peerContext = peerStore.get(peerId)
|
||||
test "Should send payments for received blocks":
|
||||
let
|
||||
done = newFuture[void]()
|
||||
account = Account(address: EthAddress.example)
|
||||
peerContext = peerStore.get(peerId)
|
||||
|
||||
# peerContext.account = account.some
|
||||
# peerContext.blocks = blocks.mapIt(
|
||||
# (it.address, Presence(address: it.address, price: rand(uint16).u256))
|
||||
# ).toTable
|
||||
peerContext.account = account.some
|
||||
peerContext.blocks = blocks.mapIt(
|
||||
(it.address, Presence(address: it.address, price: rand(uint16).u256))
|
||||
).toTable
|
||||
|
||||
# engine.network = BlockExcNetwork(
|
||||
# request: BlockExcRequest(
|
||||
# sendPayment: proc(receiver: PeerId, payment: SignedState) {.gcsafe, async: (handleException: true).} =
|
||||
# let
|
||||
# amount =
|
||||
# blocks.mapIt(
|
||||
# peerContext.blocks[it.address].price
|
||||
# ).foldl(a + b)
|
||||
engine.network = BlockExcNetwork(
|
||||
request: BlockExcRequest(
|
||||
sendPayment: proc(
|
||||
receiver: PeerId,
|
||||
payment: SignedState
|
||||
) {.gcsafe, async: (handleException: true).} =
|
||||
let
|
||||
amount =
|
||||
blocks.mapIt(
|
||||
peerContext.blocks[it.address].price
|
||||
).foldl(a + b)
|
||||
|
||||
# balances = !payment.state.outcome.balances(Asset)
|
||||
balances = !payment.state.outcome.balances(Asset)
|
||||
|
||||
# check receiver == peerId
|
||||
# check balances[account.address.toDestination] == amount
|
||||
# done.complete()
|
||||
# ))
|
||||
check receiver == peerId
|
||||
check balances[account.address.toDestination] == amount
|
||||
done.complete()
|
||||
))
|
||||
|
||||
# await engine.blocksDeliveryHandler(peerId, blocks.mapIt(BlockDelivery(blk: it, address: it.address)))
|
||||
# await done.wait(100.millis)
|
||||
await engine.blocksDeliveryHandler(peerId, blocks.mapIt(BlockDelivery(blk: it, address: it.address)))
|
||||
await done.wait(100.millis)
|
||||
|
||||
# test "Should handle block presence":
|
||||
# var
|
||||
# handles: Table[Cid, Future[Block]]
|
||||
test "Should handle block presence":
|
||||
var
|
||||
handles: Table[Cid, Future[Block]]
|
||||
|
||||
# proc sendWantList(
|
||||
# id: PeerId,
|
||||
# addresses: seq[BlockAddress],
|
||||
# priority: int32 = 0,
|
||||
# cancel: bool = false,
|
||||
# wantType: WantType = WantType.WantHave,
|
||||
# full: bool = false,
|
||||
# sendDontHave: bool = false) {.gcsafe, async.} =
|
||||
# engine.pendingBlocks.resolve(blocks
|
||||
# .filterIt( it.address in addresses )
|
||||
# .mapIt(BlockDelivery(blk: it, address: it.address)))
|
||||
proc sendWantList(
|
||||
id: PeerId,
|
||||
addresses: seq[BlockAddress],
|
||||
priority: int32 = 0,
|
||||
cancel: bool = false,
|
||||
wantType: WantType = WantType.WantHave,
|
||||
full: bool = false,
|
||||
sendDontHave: bool = false) {.gcsafe, async: (handleException: true).} =
|
||||
engine.pendingBlocks.resolve(blocks
|
||||
.filterIt( it.address in addresses )
|
||||
.mapIt(BlockDelivery(blk: it, address: it.address)))
|
||||
|
||||
# engine.network = BlockExcNetwork(
|
||||
# request: BlockExcRequest(
|
||||
# sendWantList: sendWantList
|
||||
# ))
|
||||
engine.network = BlockExcNetwork(
|
||||
request: BlockExcRequest(
|
||||
sendWantList: sendWantList
|
||||
))
|
||||
|
||||
# # only Cids in peer want lists are requested
|
||||
# handles = blocks.mapIt(
|
||||
# (it.cid, engine.pendingBlocks.getWantHandle( it.cid ))).toTable
|
||||
# only Cids in peer want lists are requested
|
||||
handles = blocks.mapIt(
|
||||
(it.cid, engine.pendingBlocks.getWantHandle( it.cid ))).toTable
|
||||
|
||||
# let price = UInt256.example
|
||||
# await engine.blockPresenceHandler(
|
||||
# peerId,
|
||||
# blocks.mapIt(
|
||||
# PresenceMessage.init(
|
||||
# Presence(
|
||||
# address: it.address,
|
||||
# have: true,
|
||||
# price: price
|
||||
# ))))
|
||||
let price = UInt256.example
|
||||
await engine.blockPresenceHandler(
|
||||
peerId,
|
||||
blocks.mapIt(
|
||||
PresenceMessage.init(
|
||||
Presence(
|
||||
address: it.address,
|
||||
have: true,
|
||||
price: price
|
||||
))))
|
||||
|
||||
# for a in blocks.mapIt(it.address):
|
||||
# check a in peerCtx.peerHave
|
||||
# check peerCtx.blocks[a].price == price
|
||||
for a in blocks.mapIt(it.address):
|
||||
check a in peerCtx.peerHave
|
||||
check peerCtx.blocks[a].price == price
|
||||
|
||||
# asyncchecksuite "Task Handler":
|
||||
# var
|
||||
# rng: Rng
|
||||
# seckey: PrivateKey
|
||||
# peerId: PeerId
|
||||
# chunker: Chunker
|
||||
# wallet: WalletRef
|
||||
# blockDiscovery: Discovery
|
||||
# peerStore: PeerCtxStore
|
||||
# pendingBlocks: PendingBlocksManager
|
||||
# network: BlockExcNetwork
|
||||
# engine: BlockExcEngine
|
||||
# discovery: DiscoveryEngine
|
||||
# localStore: BlockStore
|
||||
asyncchecksuite "Task Handler":
|
||||
var
|
||||
rng: Rng
|
||||
seckey: PrivateKey
|
||||
peerId: PeerId
|
||||
chunker: Chunker
|
||||
wallet: WalletRef
|
||||
blockDiscovery: Discovery
|
||||
peerStore: PeerCtxStore
|
||||
pendingBlocks: PendingBlocksManager
|
||||
network: BlockExcNetwork
|
||||
engine: BlockExcEngine
|
||||
discovery: DiscoveryEngine
|
||||
localStore: BlockStore
|
||||
|
||||
# peersCtx: seq[BlockExcPeerCtx]
|
||||
# peers: seq[PeerId]
|
||||
# blocks: seq[Block]
|
||||
peersCtx: seq[BlockExcPeerCtx]
|
||||
peers: seq[PeerId]
|
||||
blocks: seq[Block]
|
||||
|
||||
# setup:
|
||||
# rng = Rng.instance()
|
||||
# chunker = RandomChunker.new(rng, size = 1024, chunkSize = 256'nb)
|
||||
# while true:
|
||||
# let chunk = await chunker.getBytes()
|
||||
# if chunk.len <= 0:
|
||||
# break
|
||||
setup:
|
||||
rng = Rng.instance()
|
||||
chunker = RandomChunker.new(rng, size = 1024, chunkSize = 256'nb)
|
||||
while true:
|
||||
let chunk = await chunker.getBytes()
|
||||
if chunk.len <= 0:
|
||||
break
|
||||
|
||||
# blocks.add(Block.new(chunk).tryGet())
|
||||
blocks.add(Block.new(chunk).tryGet())
|
||||
|
||||
# seckey = PrivateKey.random(rng[]).tryGet()
|
||||
# peerId = PeerId.init(seckey.getPublicKey().tryGet()).tryGet()
|
||||
# wallet = WalletRef.example
|
||||
# blockDiscovery = Discovery.new()
|
||||
# peerStore = PeerCtxStore.new()
|
||||
# pendingBlocks = PendingBlocksManager.new()
|
||||
seckey = PrivateKey.random(rng[]).tryGet()
|
||||
peerId = PeerId.init(seckey.getPublicKey().tryGet()).tryGet()
|
||||
wallet = WalletRef.example
|
||||
blockDiscovery = Discovery.new()
|
||||
peerStore = PeerCtxStore.new()
|
||||
pendingBlocks = PendingBlocksManager.new()
|
||||
|
||||
# localStore = CacheStore.new()
|
||||
# network = BlockExcNetwork()
|
||||
localStore = CacheStore.new()
|
||||
network = BlockExcNetwork()
|
||||
|
||||
# discovery = DiscoveryEngine.new(
|
||||
# localStore,
|
||||
# peerStore,
|
||||
# network,
|
||||
# blockDiscovery,
|
||||
# pendingBlocks)
|
||||
discovery = DiscoveryEngine.new(
|
||||
localStore,
|
||||
peerStore,
|
||||
network,
|
||||
blockDiscovery,
|
||||
pendingBlocks)
|
||||
|
||||
# engine = BlockExcEngine.new(
|
||||
# localStore,
|
||||
# wallet,
|
||||
# network,
|
||||
# discovery,
|
||||
# peerStore,
|
||||
# pendingBlocks)
|
||||
# peersCtx = @[]
|
||||
engine = BlockExcEngine.new(
|
||||
localStore,
|
||||
wallet,
|
||||
network,
|
||||
discovery,
|
||||
peerStore,
|
||||
pendingBlocks)
|
||||
peersCtx = @[]
|
||||
|
||||
# for i in 0..3:
|
||||
# let seckey = PrivateKey.random(rng[]).tryGet()
|
||||
# peers.add(PeerId.init(seckey.getPublicKey().tryGet()).tryGet())
|
||||
for i in 0..3:
|
||||
let seckey = PrivateKey.random(rng[]).tryGet()
|
||||
peers.add(PeerId.init(seckey.getPublicKey().tryGet()).tryGet())
|
||||
|
||||
# peersCtx.add(BlockExcPeerCtx(
|
||||
# id: peers[i]
|
||||
# ))
|
||||
# peerStore.add(peersCtx[i])
|
||||
peersCtx.add(BlockExcPeerCtx(
|
||||
id: peers[i]
|
||||
))
|
||||
peerStore.add(peersCtx[i])
|
||||
|
||||
# engine.pricing = Pricing.example.some
|
||||
engine.pricing = Pricing.example.some
|
||||
|
||||
# test "Should send want-blocks in priority order":
|
||||
# proc sendBlocksDelivery(
|
||||
# id: PeerId,
|
||||
# blocksDelivery: seq[BlockDelivery]) {.gcsafe, async: (handleException: true).} =
|
||||
# check blocksDelivery.len == 2
|
||||
# check:
|
||||
# blocksDelivery[1].address == blocks[0].address
|
||||
# blocksDelivery[0].address == blocks[1].address
|
||||
test "Should send want-blocks in priority order":
|
||||
proc sendBlocksDelivery(
|
||||
id: PeerId,
|
||||
blocksDelivery: seq[BlockDelivery]
|
||||
) {.gcsafe, async: (handleException: true).} =
|
||||
check blocksDelivery.len == 2
|
||||
check:
|
||||
blocksDelivery[1].address == blocks[0].address
|
||||
blocksDelivery[0].address == blocks[1].address
|
||||
|
||||
# for blk in blocks:
|
||||
# (await engine.localStore.putBlock(blk)).tryGet()
|
||||
# engine.network.request.sendBlocksDelivery = sendBlocksDelivery
|
||||
for blk in blocks:
|
||||
(await engine.localStore.putBlock(blk)).tryGet()
|
||||
engine.network.request.sendBlocksDelivery = sendBlocksDelivery
|
||||
|
||||
# # second block to send by priority
|
||||
# peersCtx[0].peerWants.add(
|
||||
# WantListEntry(
|
||||
# address: blocks[0].address,
|
||||
# priority: 49,
|
||||
# cancel: false,
|
||||
# wantType: WantType.WantBlock,
|
||||
# sendDontHave: false)
|
||||
# )
|
||||
# second block to send by priority
|
||||
peersCtx[0].peerWants.add(
|
||||
WantListEntry(
|
||||
address: blocks[0].address,
|
||||
priority: 49,
|
||||
cancel: false,
|
||||
wantType: WantType.WantBlock,
|
||||
sendDontHave: false)
|
||||
)
|
||||
|
||||
# # first block to send by priority
|
||||
# peersCtx[0].peerWants.add(
|
||||
# WantListEntry(
|
||||
# address: blocks[1].address,
|
||||
# priority: 50,
|
||||
# cancel: false,
|
||||
# wantType: WantType.WantBlock,
|
||||
# sendDontHave: false)
|
||||
# )
|
||||
# first block to send by priority
|
||||
peersCtx[0].peerWants.add(
|
||||
WantListEntry(
|
||||
address: blocks[1].address,
|
||||
priority: 50,
|
||||
cancel: false,
|
||||
wantType: WantType.WantBlock,
|
||||
sendDontHave: false)
|
||||
)
|
||||
|
||||
# await engine.taskHandler(peersCtx[0])
|
||||
await engine.taskHandler(peersCtx[0])
|
||||
|
||||
# test "Should send presence":
|
||||
# let present = blocks
|
||||
# let missing = @[Block.new("missing".toBytes).tryGet()]
|
||||
# let price = (!engine.pricing).price
|
||||
test "Should send presence":
|
||||
let present = blocks
|
||||
let missing = @[Block.new("missing".toBytes).tryGet()]
|
||||
let price = (!engine.pricing).price
|
||||
|
||||
# proc sendPresence(id: PeerId, presence: seq[BlockPresence]) {.gcsafe, async: (handleException: true).} =
|
||||
# check presence.mapIt(!Presence.init(it)) == @[
|
||||
# Presence(address: present[0].address, have: true, price: price),
|
||||
# Presence(address: present[1].address, have: true, price: price),
|
||||
# Presence(address: missing[0].address, have: false)
|
||||
# ]
|
||||
proc sendPresence(id: PeerId,
|
||||
presence: seq[BlockPresence]) {.gcsafe, async: (handleException: true).} =
|
||||
check presence.mapIt(!Presence.init(it)) == @[
|
||||
Presence(address: present[0].address, have: true, price: price),
|
||||
Presence(address: present[1].address, have: true, price: price),
|
||||
Presence(address: missing[0].address, have: false)
|
||||
]
|
||||
|
||||
# for blk in blocks:
|
||||
# (await engine.localStore.putBlock(blk)).tryGet()
|
||||
# engine.network.request.sendPresence = sendPresence
|
||||
for blk in blocks:
|
||||
(await engine.localStore.putBlock(blk)).tryGet()
|
||||
engine.network.request.sendPresence = sendPresence
|
||||
|
||||
# # have block
|
||||
# peersCtx[0].peerWants.add(
|
||||
# WantListEntry(
|
||||
# address: present[0].address,
|
||||
# priority: 1,
|
||||
# cancel: false,
|
||||
# wantType: WantType.WantHave,
|
||||
# sendDontHave: false)
|
||||
# )
|
||||
# have block
|
||||
peersCtx[0].peerWants.add(
|
||||
WantListEntry(
|
||||
address: present[0].address,
|
||||
priority: 1,
|
||||
cancel: false,
|
||||
wantType: WantType.WantHave,
|
||||
sendDontHave: false)
|
||||
)
|
||||
|
||||
# # have block
|
||||
# peersCtx[0].peerWants.add(
|
||||
# WantListEntry(
|
||||
# address: present[1].address,
|
||||
# priority: 1,
|
||||
# cancel: false,
|
||||
# wantType: WantType.WantHave,
|
||||
# sendDontHave: false)
|
||||
# )
|
||||
# have block
|
||||
peersCtx[0].peerWants.add(
|
||||
WantListEntry(
|
||||
address: present[1].address,
|
||||
priority: 1,
|
||||
cancel: false,
|
||||
wantType: WantType.WantHave,
|
||||
sendDontHave: false)
|
||||
)
|
||||
|
||||
# # don't have block
|
||||
# peersCtx[0].peerWants.add(
|
||||
# WantListEntry(
|
||||
# address: missing[0].address,
|
||||
# priority: 1,
|
||||
# cancel: false,
|
||||
# wantType: WantType.WantHave,
|
||||
# sendDontHave: false)
|
||||
# )
|
||||
# don't have block
|
||||
peersCtx[0].peerWants.add(
|
||||
WantListEntry(
|
||||
address: missing[0].address,
|
||||
priority: 1,
|
||||
cancel: false,
|
||||
wantType: WantType.WantHave,
|
||||
sendDontHave: false)
|
||||
)
|
||||
|
||||
# await engine.taskHandler(peersCtx[0])
|
||||
await engine.taskHandler(peersCtx[0])
|
||||
|
|
|
|||
Loading…
Reference in New Issue