mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-01-06 23:43:08 +00:00
* cleanup imports and logs * add BlockHandle type * revert deps * refactor: async error handling and future tracking improvements - Update async procedures to use explicit raises annotation - Modify TrackedFutures to handle futures with no raised exceptions - Replace `asyncSpawn` with explicit future tracking - Update test suites to use `unittest2` - Standardize error handling across network and async components - Remove deprecated error handling patterns This commit introduces a more robust approach to async error handling and future management, improving type safety and reducing potential runtime errors. * bump nim-serde * remove asyncSpawn * rework background downloads and prefetch * imporove logging * refactor: enhance async procedures with error handling and raise annotations * misc cleanup * misc * refactor: implement allFinishedFailed to aggregate future results with success and failure tracking * refactor: update error handling in reader procedures to raise ChunkerError and CancelledError * refactor: improve error handling in wantListHandler and accountHandler procedures * refactor: simplify LPStreamReadError creation by consolidating parameters * refactor: enhance error handling in AsyncStreamWrapper to catch unexpected errors * refactor: enhance error handling in advertiser and discovery loops to improve resilience * misc * refactor: improve code structure and readability * remove cancellation from addSlotToQueue * refactor: add assertion for unexpected errors in local store checks * refactor: prevent tracking of finished futures and improve test assertions * refactor: improve error handling in local store checks * remove usage of msgDetail * feat: add initial implementation of discovery engine and related components * refactor: improve task scheduling logic by removing unnecessary break statement * break after scheduling a task * make taskHandler cancelable * refactor: update async handlers to raise CancelledError * refactor(advertiser): streamline error handling and improve task flow in advertise loops * fix: correct spelling of "divisible" in error messages and comments * refactor(discovery): simplify discovery task loop and improve error handling * refactor(engine): filter peers before processing in cancelBlocks procedure
69 lines
2.0 KiB
Nim
69 lines
2.0 KiB
Nim
import pkg/questionable/results
|
|
|
|
import pkg/codex/clock
|
|
import pkg/codex/contracts/requests
|
|
import pkg/codex/sales
|
|
import pkg/codex/sales/salesagent
|
|
import pkg/codex/sales/salescontext
|
|
import pkg/codex/sales/states/filled
|
|
import pkg/codex/sales/states/errored
|
|
import pkg/codex/sales/states/proving
|
|
|
|
import ../../../asynctest
|
|
import ../../helpers/mockmarket
|
|
import ../../examples
|
|
import ../../helpers
|
|
|
|
suite "sales state 'filled'":
|
|
let request = StorageRequest.example
|
|
let slotIndex = request.ask.slots div 2
|
|
|
|
var market: MockMarket
|
|
var slot: MockSlot
|
|
var agent: SalesAgent
|
|
var state: SaleFilled
|
|
var onExpiryUpdatePassedExpiry: SecondsSince1970
|
|
|
|
setup:
|
|
market = MockMarket.new()
|
|
slot = MockSlot(
|
|
requestId: request.id,
|
|
host: Address.example,
|
|
slotIndex: slotIndex,
|
|
proof: Groth16Proof.default,
|
|
)
|
|
|
|
market.requestEnds[request.id] = 321
|
|
onExpiryUpdatePassedExpiry = -1
|
|
let onExpiryUpdate = proc(
|
|
rootCid: Cid, expiry: SecondsSince1970
|
|
): Future[?!void] {.async.} =
|
|
onExpiryUpdatePassedExpiry = expiry
|
|
return success()
|
|
let context = SalesContext(market: market, onExpiryUpdate: some onExpiryUpdate)
|
|
|
|
agent = newSalesAgent(context, request.id, slotIndex, some request)
|
|
state = SaleFilled.new()
|
|
|
|
test "switches to proving state when slot is filled by me":
|
|
slot.host = await market.getSigner()
|
|
market.filled = @[slot]
|
|
let next = await state.run(agent)
|
|
check !next of SaleProving
|
|
|
|
test "calls onExpiryUpdate with request end":
|
|
slot.host = await market.getSigner()
|
|
market.filled = @[slot]
|
|
|
|
let expectedExpiry = 123
|
|
market.requestEnds[request.id] = expectedExpiry
|
|
let next = await state.run(agent)
|
|
check !next of SaleProving
|
|
check onExpiryUpdatePassedExpiry == expectedExpiry
|
|
|
|
test "switches to error state when slot is filled by another host":
|
|
slot.host = Address.example
|
|
market.filled = @[slot]
|
|
let next = await state.run(agent)
|
|
check !next of SaleErrored
|