50 lines
1.4 KiB
Nim
50 lines
1.4 KiB
Nim
|
import pkg/questionable
|
||
|
|
import pkg/chronos
|
||
|
|
import pkg/codex/contracts/requests
|
||
|
|
import pkg/codex/sales/states/errored
|
||
|
|
import pkg/codex/sales/salesagent
|
||
|
|
import pkg/codex/sales/salescontext
|
||
|
|
import pkg/codex/market
|
||
|
|
|
||
|
|
import ../../../asynctest
|
||
|
|
import ../../examples
|
||
|
|
import ../../helpers
|
||
|
|
import ../../helpers/mockmarket
|
||
|
|
import ../../helpers/mockclock
|
||
|
|
|
||
|
|
asyncchecksuite "sales state 'errored'":
|
||
|
|
let request = StorageRequest.example
|
||
|
|
let slotIndex = (request.ask.slots div 2).u256
|
||
|
|
let market = MockMarket.new()
|
||
|
|
let clock = MockClock.new()
|
||
|
|
|
||
|
|
var state: SaleErrored
|
||
|
|
var agent: SalesAgent
|
||
|
|
var returnBytesWas = false
|
||
|
|
var reprocessSlotWas = false
|
||
|
|
|
||
|
|
setup:
|
||
|
|
let onCleanUp = proc (returnBytes = false, reprocessSlot = false) {.async.} =
|
||
|
|
returnBytesWas = returnBytes
|
||
|
|
reprocessSlotWas = reprocessSlot
|
||
|
|
|
||
|
|
let context = SalesContext(
|
||
|
|
market: market,
|
||
|
|
clock: clock
|
||
|
|
)
|
||
|
|
agent = newSalesAgent(context,
|
||
|
|
request.id,
|
||
|
|
slotIndex,
|
||
|
|
request.some)
|
||
|
|
agent.onCleanUp = onCleanUp
|
||
|
|
state = SaleErrored(error: newException(ValueError, "oh no!"))
|
||
|
|
|
||
|
|
test "calls onCleanUp with returnBytes = false and reprocessSlot = true":
|
||
|
|
state = SaleErrored(
|
||
|
|
error: newException(ValueError, "oh no!"),
|
||
|
|
reprocessSlot: true
|
||
|
|
)
|
||
|
|
discard await state.run(agent)
|
||
|
|
check eventually returnBytesWas == true
|
||
|
|
check eventually reprocessSlotWas == true
|