mirror of
https://github.com/codex-storage/nim-codex.git
synced 2025-02-13 21:36:34 +00:00
1a57341b7d
* fix: createReservation lock * fix: additional locking places * fix: acquire lock * chore: feedback Co-authored-by: markspanbroek <mark@spanbroek.net> Signed-off-by: Adam Uhlíř <adam@uhlir.dev> * feat: withLock template and fixed tests * fix: use proc for MockReservations constructor * chore: feedback Co-authored-by: Eric <5089238+emizzle@users.noreply.github.com> Signed-off-by: Adam Uhlíř <adam@uhlir.dev> * chore: feedback implementation --------- Signed-off-by: Adam Uhlíř <adam@uhlir.dev> Co-authored-by: markspanbroek <mark@spanbroek.net> Co-authored-by: Eric <5089238+emizzle@users.noreply.github.com>
34 lines
1.1 KiB
Nim
34 lines
1.1 KiB
Nim
import pkg/chronos
|
|
import pkg/codex/sales
|
|
import pkg/codex/stores
|
|
import pkg/questionable/results
|
|
|
|
type
|
|
MockReservations* = ref object of Reservations
|
|
createReservationThrowBytesOutOfBoundsError: bool
|
|
|
|
proc new*(
|
|
T: type MockReservations,
|
|
repo: RepoStore
|
|
): MockReservations =
|
|
## Create a mock clock instance
|
|
MockReservations(availabilityLock: newAsyncLock(), repo: repo)
|
|
|
|
proc setCreateReservationThrowBytesOutOfBoundsError*(self: MockReservations, flag: bool) =
|
|
self.createReservationThrowBytesOutOfBoundsError = flag
|
|
|
|
method createReservation*(
|
|
self: MockReservations,
|
|
availabilityId: AvailabilityId,
|
|
slotSize: UInt256,
|
|
requestId: RequestId,
|
|
slotIndex: UInt256): Future[?!Reservation] {.async.} =
|
|
if self.createReservationThrowBytesOutOfBoundsError:
|
|
let error = newException(
|
|
BytesOutOfBoundsError,
|
|
"trying to reserve an amount of bytes that is greater than the total size of the Availability")
|
|
return failure(error)
|
|
|
|
return await procCall createReservation(Reservations(self), availabilityId, slotSize, requestId, slotIndex)
|
|
|