mirror of
https://github.com/codex-storage/nim-codex.git
synced 2025-02-08 11:06:44 +00:00
72da534856
* Setting up testfixture for proof datasampler * Sets up calculating number of cells in a slot * Sets up tests for bitwise modulo * Implements cell index collection * setting up slot blocks module * Implements getting treeCID from slot * implements getting slot blocks by index * Implements out-of-range check for slot index * cleanup * Sets up getting sample from block * Implements selecting a cell sample from a block * Implements building a minitree for block cells * Adds method to get dataset block index from slot block index * It's running * splits up indexing * almost there * Fixes test. Implementation is now functional * Refactoring to object-oriented * Cleanup * Lining up output type with updated reference code. * setting up * Updates expected samples * Updates proof checking test to match new format * move builder to own dir * move sampler to own dir * fix paths * various changes to add support for the sampler * wip sampler implementation * don't use upraises * wip sampler integration * misc * move tests around * Various fixes to select correct slot and block index * removing old tests * cleanup * misc fix tests that work with correct cell indices * remove unused file * fixup logging * add logscope * truncate entropy to 31 bytes, otherwise it might be > than mod * forwar getCidAndProof to local store * misc * Adds missing test for initial-proving state * reverting back to correct slot/block indexing * fix tests for revert * misc * misc --------- Co-authored-by: benbierens <thatbenbierens@gmail.com>
76 lines
2.4 KiB
Nim
76 lines
2.4 KiB
Nim
import pkg/asynctest
|
|
import pkg/questionable
|
|
import pkg/chronos
|
|
import pkg/codex/contracts/requests
|
|
import pkg/codex/sales/states/initialproving
|
|
import pkg/codex/sales/states/cancelled
|
|
import pkg/codex/sales/states/failed
|
|
import pkg/codex/sales/states/filling
|
|
import pkg/codex/sales/states/errored
|
|
import pkg/codex/sales/salesagent
|
|
import pkg/codex/sales/salescontext
|
|
import pkg/codex/market
|
|
import ../../examples
|
|
import ../../helpers
|
|
import ../../helpers/mockmarket
|
|
|
|
asyncchecksuite "sales state 'initialproving'":
|
|
let proof = exampleProof()
|
|
let request = StorageRequest.example
|
|
let slotIndex = (request.ask.slots div 2).u256
|
|
let market = MockMarket.new()
|
|
|
|
var state: SaleInitialProving
|
|
var agent: SalesAgent
|
|
var receivedChallenge: ProofChallenge
|
|
|
|
setup:
|
|
let onProve = proc (slot: Slot, challenge: ProofChallenge): Future[?!seq[byte]] {.async.} =
|
|
receivedChallenge = challenge
|
|
return success(proof)
|
|
let context = SalesContext(
|
|
onProve: onProve.some,
|
|
market: market
|
|
)
|
|
agent = newSalesAgent(context,
|
|
request.id,
|
|
slotIndex,
|
|
request.some)
|
|
state = SaleInitialProving.new()
|
|
|
|
test "switches to cancelled state when request expires":
|
|
let next = state.onCancelled(request)
|
|
check !next of SaleCancelled
|
|
|
|
test "switches to failed state when request fails":
|
|
let next = state.onFailed(request)
|
|
check !next of SaleFailed
|
|
|
|
test "switches to filling state when initial proving is complete":
|
|
let next = await state.run(agent)
|
|
check !next of SaleFilling
|
|
check SaleFilling(!next).proof == proof
|
|
|
|
test "onProve callback provides proof challenge":
|
|
market.proofChallenge = ProofChallenge.example
|
|
|
|
let future = state.run(agent)
|
|
|
|
check receivedChallenge == market.proofChallenge
|
|
|
|
test "switches to errored state when onProve callback fails":
|
|
let onProveFailed: OnProve = proc(slot: Slot, challenge: ProofChallenge): Future[?!seq[byte]] {.async.} =
|
|
return failure("oh no!")
|
|
|
|
let proofFailedContext = SalesContext(
|
|
onProve: onProveFailed.some,
|
|
market: market
|
|
)
|
|
agent = newSalesAgent(proofFailedContext,
|
|
request.id,
|
|
slotIndex,
|
|
request.some)
|
|
|
|
let next = await state.run(agent)
|
|
check !next of SaleErrored
|