133 lines
4.3 KiB
Nim
133 lines
4.3 KiB
Nim
import std/times
|
|
import pkg/asynctest
|
|
import pkg/chronos
|
|
import pkg/codex/sales
|
|
import ./helpers/mockmarket
|
|
import ./helpers/mockclock
|
|
import ./examples
|
|
|
|
suite "Sales":
|
|
|
|
let availability = Availability.init(
|
|
size=100.u256,
|
|
duration=60.u256,
|
|
minPrice=42.u256
|
|
)
|
|
let request = StorageRequest(
|
|
ask: StorageAsk(
|
|
duration: 60.u256,
|
|
size: 100.u256,
|
|
maxPrice:42.u256
|
|
),
|
|
content: StorageContent(
|
|
cid: "some cid"
|
|
)
|
|
)
|
|
|
|
var sales: Sales
|
|
var market: MockMarket
|
|
var clock: MockClock
|
|
|
|
setup:
|
|
market = MockMarket.new()
|
|
clock = MockClock.new()
|
|
sales = Sales.new(market, clock)
|
|
sales.retrieve = proc(_: string) {.async.} = discard
|
|
sales.prove = proc(_: string): Future[seq[byte]] {.async.} = discard
|
|
await sales.start()
|
|
|
|
teardown:
|
|
await sales.stop()
|
|
|
|
test "has no availability initially":
|
|
check sales.available.len == 0
|
|
|
|
test "can add available storage":
|
|
let availability1 = Availability.example
|
|
let availability2 = Availability.example
|
|
sales.add(availability1)
|
|
check sales.available.contains(availability1)
|
|
sales.add(availability2)
|
|
check sales.available.contains(availability1)
|
|
check sales.available.contains(availability2)
|
|
|
|
test "can remove available storage":
|
|
sales.add(availability)
|
|
sales.remove(availability)
|
|
check sales.available.len == 0
|
|
|
|
test "generates unique ids for storage availability":
|
|
let availability1 = Availability.init(1.u256, 2.u256, 3.u256)
|
|
let availability2 = Availability.init(1.u256, 2.u256, 3.u256)
|
|
check availability1.id != availability2.id
|
|
|
|
test "makes storage unavailable when matching request comes in":
|
|
sales.add(availability)
|
|
discard await market.requestStorage(request)
|
|
check sales.available.len == 0
|
|
|
|
test "ignores request when no matching storage is available":
|
|
sales.add(availability)
|
|
var tooBig = request
|
|
tooBig.ask.size = request.ask.size + 1
|
|
discard await market.requestStorage(tooBig)
|
|
check sales.available == @[availability]
|
|
|
|
test "retrieves data":
|
|
var retrievingCid: string
|
|
sales.retrieve = proc(cid: string) {.async.} = retrievingCid = cid
|
|
sales.add(availability)
|
|
discard await market.requestStorage(request)
|
|
check retrievingCid == request.content.cid
|
|
|
|
test "makes storage available again when data retrieval fails":
|
|
let error = newException(IOError, "data retrieval failed")
|
|
sales.retrieve = proc(cid: string) {.async.} = raise error
|
|
sales.add(availability)
|
|
discard await market.requestStorage(request)
|
|
check sales.available == @[availability]
|
|
|
|
test "generates proof of storage":
|
|
var provingCid: string
|
|
sales.prove = proc(cid: string): Future[seq[byte]] {.async.} = provingCid = cid
|
|
sales.add(availability)
|
|
discard await market.requestStorage(request)
|
|
check provingCid == request.content.cid
|
|
|
|
# test "calls onSale when offer is selected":
|
|
# var sold: StorageOffer
|
|
# sales.onSale = proc(offer: StorageOffer) =
|
|
# sold = offer
|
|
# sales.add(availability)
|
|
# discard await market.requestStorage(request)
|
|
# let offer = market.offered[0]
|
|
# await market.selectOffer(offer.id)
|
|
# check sold == offer
|
|
|
|
# test "does not call onSale when a different offer is selected":
|
|
# var didSell: bool
|
|
# sales.onSale = proc(offer: StorageOffer) =
|
|
# didSell = true
|
|
# sales.add(availability)
|
|
# let request = await market.requestStorage(request)
|
|
# var otherOffer = StorageOffer(requestId: request.id, price: 1.u256)
|
|
# otherOffer = await market.offerStorage(otherOffer)
|
|
# await market.selectOffer(otherOffer.id)
|
|
# check not didSell
|
|
|
|
# test "makes storage available again when different offer is selected":
|
|
# sales.add(availability)
|
|
# let request = await market.requestStorage(request)
|
|
# var otherOffer = StorageOffer(requestId: request.id, price: 1.u256)
|
|
# otherOffer = await market.offerStorage(otherOffer)
|
|
# await market.selectOffer(otherOffer.id)
|
|
# check sales.available.contains(availability)
|
|
|
|
# test "makes storage available again when offer expires":
|
|
# sales.add(availability)
|
|
# discard await market.requestStorage(request)
|
|
# let offer = market.offered[0]
|
|
# clock.set(offer.expiry.truncate(int64))
|
|
# await sleepAsync(chronos.seconds(2))
|
|
# check sales.available.contains(availability)
|