115 lines
4.3 KiB
JavaScript
115 lines
4.3 KiB
JavaScript
const { expect } = require("chai")
|
|
const { exampleLock } = require("./examples")
|
|
|
|
describe("Collateral", function () {
|
|
let collateral, token
|
|
let account0, account1
|
|
|
|
beforeEach(async function () {
|
|
let Collateral = await ethers.getContractFactory("TestCollateral")
|
|
let TestToken = await ethers.getContractFactory("TestToken")
|
|
token = await TestToken.deploy()
|
|
collateral = await Collateral.deploy(token.address)
|
|
;[account0, account1] = await ethers.getSigners()
|
|
await token.mint(account0.address, 1000)
|
|
await token.mint(account1.address, 1000)
|
|
})
|
|
|
|
it("assigns zero collateral by default", async function () {
|
|
expect(await collateral.balanceOf(account0.address)).to.equal(0)
|
|
expect(await collateral.balanceOf(account1.address)).to.equal(0)
|
|
})
|
|
|
|
describe("depositing", function () {
|
|
beforeEach(async function () {
|
|
await token.connect(account0).approve(collateral.address, 100)
|
|
await token.connect(account1).approve(collateral.address, 100)
|
|
})
|
|
|
|
it("updates the amount of collateral", async function () {
|
|
await collateral.connect(account0).deposit(40)
|
|
await collateral.connect(account1).deposit(2)
|
|
expect(await collateral.balanceOf(account0.address)).to.equal(40)
|
|
expect(await collateral.balanceOf(account1.address)).to.equal(2)
|
|
})
|
|
|
|
it("transfers tokens to the contract", async function () {
|
|
let before = await token.balanceOf(collateral.address)
|
|
await collateral.deposit(42)
|
|
let after = await token.balanceOf(collateral.address)
|
|
expect(after - before).to.equal(42)
|
|
})
|
|
|
|
it("fails when token transfer fails", async function () {
|
|
let allowed = await token.allowance(account0.address, collateral.address)
|
|
let invalidAmount = allowed.toNumber() + 1
|
|
await expect(collateral.deposit(invalidAmount)).to.be.revertedWith(
|
|
"ERC20: insufficient allowance"
|
|
)
|
|
})
|
|
})
|
|
|
|
describe("withdrawing", function () {
|
|
beforeEach(async function () {
|
|
await token.connect(account0).approve(collateral.address, 100)
|
|
await token.connect(account1).approve(collateral.address, 100)
|
|
await collateral.connect(account0).deposit(40)
|
|
await collateral.connect(account1).deposit(2)
|
|
})
|
|
|
|
it("updates the amount of collateral", async function () {
|
|
await collateral.connect(account0).withdraw()
|
|
expect(await collateral.balanceOf(account0.address)).to.equal(0)
|
|
expect(await collateral.balanceOf(account1.address)).to.equal(2)
|
|
await collateral.connect(account1).withdraw()
|
|
expect(await collateral.balanceOf(account0.address)).to.equal(0)
|
|
expect(await collateral.balanceOf(account1.address)).to.equal(0)
|
|
})
|
|
|
|
it("transfers balance to owner", async function () {
|
|
let balance = await collateral.balanceOf(account0.address)
|
|
let before = await token.balanceOf(account0.address)
|
|
await collateral.withdraw()
|
|
let after = await token.balanceOf(account0.address)
|
|
expect(after - before).to.equal(balance)
|
|
})
|
|
})
|
|
|
|
describe("slashing", function () {
|
|
beforeEach(async function () {
|
|
await token.connect(account0).approve(collateral.address, 1000)
|
|
await token.connect(account1).approve(collateral.address, 1000)
|
|
await collateral.connect(account0).deposit(1000)
|
|
await collateral.connect(account1).deposit(1000)
|
|
})
|
|
|
|
it("reduces the amount of collateral by a percentage", async function () {
|
|
await collateral.slash(account0.address, 10)
|
|
await collateral.slash(account1.address, 5)
|
|
expect(await collateral.balanceOf(account0.address)).to.equal(900)
|
|
expect(await collateral.balanceOf(account1.address)).to.equal(950)
|
|
})
|
|
})
|
|
|
|
describe("locking", function () {
|
|
let lock
|
|
|
|
beforeEach(async function () {
|
|
await token.approve(collateral.address, 42)
|
|
await collateral.deposit(42)
|
|
lock = await exampleLock()
|
|
await collateral.createLock(lock.id, lock.expiry)
|
|
await collateral.lock(account0.address, lock.id)
|
|
})
|
|
|
|
it("withdrawal fails when account is locked", async function () {
|
|
await expect(collateral.withdraw()).to.be.revertedWith("Account locked")
|
|
})
|
|
|
|
it("withdrawal succeeds when account is unlocked", async function () {
|
|
await collateral.unlock(lock.id)
|
|
await expect(collateral.withdraw()).not.to.be.reverted
|
|
})
|
|
})
|
|
})
|