mirror of
https://github.com/codex-storage/nim-codex.git
synced 2025-02-24 10:38:08 +00:00
aaf1c53e69
- getState / state > requestState - getSlot > getRequestFromSlotId - support MarketplaceConfig - support slotState, remove unneeded Slot type - collateral > config.collateral.initialAmount - remove proofPeriod contract call - Revert reason “Slot empty” > “Slot is free” - getProofEnd > read SlotState Tests for changes
75 lines
2.4 KiB
Nim
75 lines
2.4 KiB
Nim
import std/sets
|
|
import std/tables
|
|
import std/sequtils
|
|
import pkg/upraises
|
|
import pkg/codex/storageproofs
|
|
|
|
type
|
|
MockProofs* = ref object of Proofs
|
|
periodicity: Periodicity
|
|
proofsRequired: HashSet[SlotId]
|
|
proofsToBeRequired: HashSet[SlotId]
|
|
proofEnds: Table[SlotId, UInt256]
|
|
subscriptions: seq[MockSubscription]
|
|
slotStates: Table[SlotId, SlotState]
|
|
MockSubscription* = ref object of Subscription
|
|
proofs: MockProofs
|
|
callback: OnProofSubmitted
|
|
|
|
const DefaultPeriodLength = 10.u256
|
|
|
|
func new*(_: type MockProofs): MockProofs =
|
|
MockProofs(periodicity: Periodicity(seconds: DefaultPeriodLength))
|
|
|
|
func setPeriodicity*(mock: MockProofs, periodicity: Periodicity) =
|
|
mock.periodicity = periodicity
|
|
|
|
method periodicity*(mock: MockProofs): Future[Periodicity] {.async.} =
|
|
return mock.periodicity
|
|
|
|
proc setProofRequired*(mock: MockProofs, id: SlotId, required: bool) =
|
|
if required:
|
|
mock.proofsRequired.incl(id)
|
|
else:
|
|
mock.proofsRequired.excl(id)
|
|
|
|
method isProofRequired*(mock: MockProofs,
|
|
id: SlotId): Future[bool] {.async.} =
|
|
return mock.proofsRequired.contains(id)
|
|
|
|
proc setProofToBeRequired*(mock: MockProofs, id: SlotId, required: bool) =
|
|
if required:
|
|
mock.proofsToBeRequired.incl(id)
|
|
else:
|
|
mock.proofsToBeRequired.excl(id)
|
|
|
|
method willProofBeRequired*(mock: MockProofs,
|
|
id: SlotId): Future[bool] {.async.} =
|
|
return mock.proofsToBeRequired.contains(id)
|
|
|
|
proc setProofEnd*(mock: MockProofs, id: SlotId, proofEnd: UInt256) =
|
|
mock.proofEnds[id] = proofEnd
|
|
|
|
method submitProof*(mock: MockProofs,
|
|
id: SlotId,
|
|
proof: seq[byte]) {.async.} =
|
|
for subscription in mock.subscriptions:
|
|
subscription.callback(id, proof)
|
|
|
|
method subscribeProofSubmission*(mock: MockProofs,
|
|
callback: OnProofSubmitted):
|
|
Future[Subscription] {.async.} =
|
|
let subscription = MockSubscription(proofs: mock, callback: callback)
|
|
mock.subscriptions.add(subscription)
|
|
return subscription
|
|
|
|
method unsubscribe*(subscription: MockSubscription) {.async, upraises:[].} =
|
|
subscription.proofs.subscriptions.keepItIf(it != subscription)
|
|
|
|
method slotState*(mock: MockProofs,
|
|
slotId: SlotId): Future[SlotState] {.async.} =
|
|
return mock.slotStates[slotId]
|
|
|
|
proc setSlotState*(mock: MockProofs, slotId: SlotId, state: SlotState) =
|
|
mock.slotStates[slotId] = state
|