117 lines
3.3 KiB
Nim
117 lines
3.3 KiB
Nim
from std/times import getTime, toUnix
|
|
import pkg/asynctest
|
|
import pkg/chronos
|
|
import pkg/dagger/proving
|
|
import ./helpers/mockproofs
|
|
import ./examples
|
|
|
|
suite "Proving":
|
|
|
|
var proving: Proving
|
|
var proofs: MockProofs
|
|
|
|
setup:
|
|
proofs = MockProofs.new()
|
|
proving = Proving.new(proofs)
|
|
await proving.start()
|
|
|
|
teardown:
|
|
await proving.stop()
|
|
|
|
proc advanceToNextPeriod(proofs: MockProofs) {.async.} =
|
|
let current = await proofs.getCurrentPeriod()
|
|
proofs.advanceToPeriod(current + 1)
|
|
await sleepAsync(1.milliseconds)
|
|
|
|
test "maintains a list of contract ids to watch":
|
|
let id1, id2 = ContractId.example
|
|
check proving.contracts.len == 0
|
|
proving.add(id1)
|
|
check proving.contracts.contains(id1)
|
|
proving.add(id2)
|
|
check proving.contracts.contains(id1)
|
|
check proving.contracts.contains(id2)
|
|
|
|
test "removes duplicate contract ids":
|
|
let id = ContractId.example
|
|
proving.add(id)
|
|
proving.add(id)
|
|
check proving.contracts.len == 1
|
|
|
|
test "invokes callback when proof is required":
|
|
let id = ContractId.example
|
|
proving.add(id)
|
|
var called: bool
|
|
proc onProofRequired(id: ContractId) =
|
|
called = true
|
|
proving.onProofRequired = onProofRequired
|
|
proofs.setProofRequired(id, true)
|
|
await proofs.advanceToNextPeriod()
|
|
check called
|
|
|
|
test "callback receives id of contract for which proof is required":
|
|
let id1, id2 = ContractId.example
|
|
proving.add(id1)
|
|
proving.add(id2)
|
|
var callbackIds: seq[ContractId]
|
|
proc onProofRequired(id: ContractId) =
|
|
callbackIds.add(id)
|
|
proving.onProofRequired = onProofRequired
|
|
proofs.setProofRequired(id1, true)
|
|
await proofs.advanceToNextPeriod()
|
|
check callbackIds == @[id1]
|
|
proofs.setProofRequired(id1, false)
|
|
proofs.setProofRequired(id2, true)
|
|
await proofs.advanceToNextPeriod()
|
|
check callbackIds == @[id1, id2]
|
|
|
|
test "invokes callback when proof is about to be required":
|
|
let id = ContractId.example
|
|
proving.add(id)
|
|
var called: bool
|
|
proc onProofRequired(id: ContractId) =
|
|
called = true
|
|
proving.onProofRequired = onProofRequired
|
|
proofs.setProofRequired(id, false)
|
|
proofs.setProofToBeRequired(id, true)
|
|
await proofs.advanceToNextPeriod()
|
|
check called
|
|
|
|
test "stops watching when contract has ended":
|
|
let id = ContractId.example
|
|
proving.add(id)
|
|
proofs.setProofEnd(id, getTime().toUnix().u256)
|
|
await proofs.advanceToNextPeriod()
|
|
var called: bool
|
|
proc onProofRequired(id: ContractId) =
|
|
called = true
|
|
proving.onProofRequired = onProofRequired
|
|
proofs.setProofRequired(id, true)
|
|
await proofs.advanceToNextPeriod()
|
|
check not called
|
|
|
|
test "submits proofs":
|
|
let id = ContractId.example
|
|
let proof = seq[byte].example
|
|
await proving.submitProof(id, proof)
|
|
|
|
test "supports proof submission subscriptions":
|
|
let id = ContractId.example
|
|
let proof = seq[byte].example
|
|
|
|
var receivedIds: seq[ContractId]
|
|
var receivedProofs: seq[seq[byte]]
|
|
|
|
proc onProofSubmission(id: ContractId, proof: seq[byte]) =
|
|
receivedIds.add(id)
|
|
receivedProofs.add(proof)
|
|
|
|
let subscription = await proving.subscribeProofSubmission(onProofSubmission)
|
|
|
|
await proving.submitProof(id, proof)
|
|
|
|
check receivedIds == @[id]
|
|
check receivedProofs == @[proof]
|
|
|
|
await subscription.unsubscribe()
|