dagger-contracts/test/SlotReservations.test.js

119 lines
4.1 KiB
JavaScript

const { expect } = require("chai")
const { ethers } = require("hardhat")
const { exampleRequest, exampleConfiguration } = require("./examples")
const { requestId, slotId } = require("./ids")
describe("SlotReservations", function () {
let reservations
let provider, address1, address2, address3
let request
let reqId
let slot
let slotIndex
let id // can't use slotId because it'll shadow the function slotId
const config = exampleConfiguration()
beforeEach(async function () {
let SlotReservations = await ethers.getContractFactory(
"TestSlotReservations"
)
reservations = await SlotReservations.deploy(config.reservations)
;[provider, address1, address2, address3] = await ethers.getSigners()
request = await exampleRequest()
reqId = requestId(request)
slotIndex = request.ask.slots / 2
slot = {
request: reqId,
index: slotIndex,
}
id = slotId(slot)
})
function switchAccount(account) {
reservations = reservations.connect(account)
}
it("allows a slot to be reserved", async function () {
expect(reservations.reserveSlot(reqId, slotIndex)).to.not.be.reverted
})
it("contains the correct addresses after reservation", async function () {
await reservations.reserveSlot(reqId, slotIndex)
expect(await reservations.contains(id, provider.address)).to.be.true
switchAccount(address1)
await reservations.reserveSlot(reqId, slotIndex)
expect(await reservations.contains(id, address1.address)).to.be.true
})
it("has the correct number of addresses after reservation", async function () {
await reservations.reserveSlot(reqId, slotIndex)
expect(await reservations.length(id)).to.equal(1)
switchAccount(address1)
await reservations.reserveSlot(reqId, slotIndex)
expect(await reservations.length(id)).to.equal(2)
})
it("reports a slot can be reserved", async function () {
expect(await reservations.canReserveSlot(reqId, slotIndex)).to.be.true
})
it("cannot reserve a slot more than once", async function () {
await reservations.reserveSlot(reqId, slotIndex)
await expect(reservations.reserveSlot(reqId, slotIndex)).to.be.revertedWith(
"Reservation not allowed"
)
expect(await reservations.length(id)).to.equal(1)
})
it("reports a slot cannot be reserved if already reserved", async function () {
await reservations.reserveSlot(reqId, slotIndex)
expect(await reservations.canReserveSlot(reqId, slotIndex)).to.be.false
})
it("cannot reserve a slot if reservations are at capacity", async function () {
switchAccount(address1)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address2)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address3)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(provider)
await expect(reservations.reserveSlot(reqId, slotIndex)).to.be.revertedWith(
"Reservation not allowed"
)
expect(await reservations.length(id)).to.equal(3)
expect(await reservations.contains(id, provider.address)).to.be.false
})
it("reports a slot cannot be reserved if reservations are at capacity", async function () {
switchAccount(address1)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address2)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address3)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(provider)
expect(await reservations.canReserveSlot(reqId, slotIndex)).to.be.false
})
it("should emit an event when slot reservations are full", async function () {
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address1)
await reservations.reserveSlot(reqId, slotIndex)
switchAccount(address2)
await expect(reservations.reserveSlot(reqId, slotIndex))
.to.emit(reservations, "SlotReservationsFull")
.withArgs(reqId, slotIndex)
})
it("should not emit an event when reservations are not full", async function () {
await expect(reservations.reserveSlot(reqId, slotIndex)).to.not.emit(
reservations,
"SlotReservationsFull"
)
})
})