114 lines
3.7 KiB
Nim
114 lines
3.7 KiB
Nim
import std/times
|
|
import pkg/asynctest
|
|
import pkg/chronos
|
|
import pkg/upraises
|
|
import pkg/stint
|
|
import pkg/codex/purchasing
|
|
import ./helpers/mockmarket
|
|
import ./helpers/mockclock
|
|
import ./examples
|
|
|
|
suite "Purchasing":
|
|
|
|
var purchasing: Purchasing
|
|
var market: MockMarket
|
|
var clock: MockClock
|
|
var request: StorageRequest
|
|
|
|
setup:
|
|
market = MockMarket.new()
|
|
clock = MockClock.new()
|
|
purchasing = Purchasing.new(market, clock)
|
|
request = StorageRequest(
|
|
ask: StorageAsk(
|
|
slots: uint8.example.uint64,
|
|
slotSize: uint32.example.u256,
|
|
duration: uint16.example.u256,
|
|
reward: uint8.example.u256
|
|
)
|
|
)
|
|
|
|
test "submits a storage request when asked":
|
|
discard purchasing.purchase(request)
|
|
let submitted = market.requested[0]
|
|
check submitted.ask.slots == request.ask.slots
|
|
check submitted.ask.slotSize == request.ask.slotSize
|
|
check submitted.ask.duration == request.ask.duration
|
|
check submitted.ask.reward == request.ask.reward
|
|
|
|
test "remembers purchases":
|
|
let purchase1 = purchasing.purchase(request)
|
|
let purchase2 = purchasing.purchase(request)
|
|
check purchasing.getPurchase(purchase1.id) == some purchase1
|
|
check purchasing.getPurchase(purchase2.id) == some purchase2
|
|
|
|
test "has a default value for proof probability":
|
|
check purchasing.proofProbability != 0.u256
|
|
|
|
test "can change default value for proof probability":
|
|
purchasing.proofProbability = 42.u256
|
|
discard purchasing.purchase(request)
|
|
check market.requested[0].ask.proofProbability == 42.u256
|
|
|
|
test "can override proof probability per request":
|
|
request.ask.proofProbability = 42.u256
|
|
discard purchasing.purchase(request)
|
|
check market.requested[0].ask.proofProbability == 42.u256
|
|
|
|
test "has a default value for request expiration interval":
|
|
check purchasing.requestExpiryInterval != 0.u256
|
|
|
|
test "can change default value for request expiration interval":
|
|
purchasing.requestExpiryInterval = 42.u256
|
|
let start = getTime().toUnix()
|
|
discard purchasing.purchase(request)
|
|
check market.requested[0].expiry == (start + 42).u256
|
|
|
|
test "can override expiry time per request":
|
|
let expiry = (getTime().toUnix() + 42).u256
|
|
request.expiry = expiry
|
|
discard purchasing.purchase(request)
|
|
check market.requested[0].expiry == expiry
|
|
|
|
test "includes a random nonce in every storage request":
|
|
discard purchasing.purchase(request)
|
|
discard purchasing.purchase(request)
|
|
check market.requested[0].nonce != market.requested[1].nonce
|
|
|
|
test "succeeds when request is fulfilled":
|
|
let purchase = purchasing.purchase(request)
|
|
let request = market.requested[0]
|
|
market.emitRequestFulfilled(request.id)
|
|
await purchase.wait()
|
|
check purchase.error.isNone
|
|
|
|
test "fails when request times out":
|
|
let purchase = purchasing.purchase(request)
|
|
let request = market.requested[0]
|
|
clock.set(request.expiry.truncate(int64))
|
|
expect PurchaseTimeout:
|
|
await purchase.wait()
|
|
|
|
test "checks that funds were withdrawn when purchase times out":
|
|
let purchase = purchasing.purchase(request)
|
|
let request = market.requested[0]
|
|
var receivedIds: seq[RequestId]
|
|
clock.set(request.expiry.truncate(int64))
|
|
|
|
proc onRequestCancelled(id: RequestId) {.gcsafe, upraises:[].} =
|
|
receivedIds.add(id)
|
|
|
|
# will only be fired when `withdrawFunds` is called on purchase timeout
|
|
let subscription = await market.subscribeRequestCancelled(
|
|
request.id,
|
|
onRequestCancelled)
|
|
var purchaseTimedOut = false
|
|
try:
|
|
await purchase.wait()
|
|
except PurchaseTimeout:
|
|
purchaseTimedOut = true
|
|
|
|
await subscription.unsubscribe()
|
|
check purchaseTimedOut
|
|
check receivedIds == @[request.id]
|