92 lines
2.6 KiB
Nim
92 lines
2.6 KiB
Nim
|
from std/times import getTime, toUnix
|
||
|
|
import pkg/asynctest
|
||
|
|
import pkg/chronos
|
||
|
|
import pkg/dagger/proving
|
||
|
|
import ./helpers/mockprooftiming
|
||
|
|
import ./examples
|
||
|
|
|
||
|
|
suite "Proving":
|
||
|
|
|
||
|
|
var proving: Proving
|
||
|
|
var timing: MockProofTiming
|
||
|
|
|
||
|
|
setup:
|
||
|
|
timing = MockProofTiming.new()
|
||
|
|
proving = Proving.new(timing)
|
||
|
|
proving.start()
|
||
|
|
|
||
|
|
teardown:
|
||
|
|
proving.stop()
|
||
|
|
|
||
|
|
proc advanceToNextPeriod(timing: MockProofTiming) {.async.} =
|
||
|
|
let current = await timing.getCurrentPeriod()
|
||
|
|
timing.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
|
||
|
|
timing.setProofRequired(id, true)
|
||
|
|
await timing.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
|
||
|
|
timing.setProofRequired(id1, true)
|
||
|
|
await timing.advanceToNextPeriod()
|
||
|
|
check callbackIds == @[id1]
|
||
|
|
timing.setProofRequired(id1, false)
|
||
|
|
timing.setProofRequired(id2, true)
|
||
|
|
await timing.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
|
||
|
|
timing.setProofRequired(id, false)
|
||
|
|
timing.setProofToBeRequired(id, true)
|
||
|
|
await timing.advanceToNextPeriod()
|
||
|
|
check called
|
||
|
|
|
||
|
|
test "stops watching when contract has ended":
|
||
|
|
let id = ContractId.example
|
||
|
|
proving.add(id)
|
||
|
|
timing.setProofEnd(id, getTime().toUnix().u256)
|
||
|
|
await timing.advanceToNextPeriod()
|
||
|
|
var called: bool
|
||
|
|
proc onProofRequired(id: ContractId) =
|
||
|
|
called = true
|
||
|
|
proving.onProofRequired = onProofRequired
|
||
|
|
timing.setProofRequired(id, true)
|
||
|
|
await timing.advanceToNextPeriod()
|
||
|
|
check not called
|