laconicd-deprecated/tests/solidity/suites/staking/test/locking/locking.js

const { assertRevert } = require('@aragon/contract-helpers-test/assertThrow')
const { bn, bigExp, assertBn, MAX_UINT64 } = require('@aragon/contract-helpers-test/numbers')

const { deploy } = require('../helpers/deploy')(artifacts)
const { approveAndStake, approveStakeAndLock } = require('../helpers/helpers')(artifacts)
const { DEFAULT_STAKE_AMOUNT, DEFAULT_LOCK_AMOUNT, EMPTY_DATA, ZERO_ADDRESS } = require('../helpers/constants')
const { STAKING_ERRORS } = require('../helpers/errors')

contract('Staking app, Locking', ([owner, user1, user2]) => {
  let staking, lockManager

  beforeEach(async () => {
    const deployment = await deploy(owner)
    staking = deployment.staking
    lockManager = deployment.lockManager
  })

  // it('allows new manager and locks amount', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   // check lock values
  //   const { _amount, _allowance } = await staking.getLock(owner, user1)
  //   assertBn(_amount, DEFAULT_LOCK_AMOUNT, "locked amount should match")
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT, "locked allowance should match")

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT.sub(DEFAULT_LOCK_AMOUNT), "Unlocked balance should match")
  //   const { staked, locked } = await staking.getBalancesOf(owner)
  //   assertBn(staked, DEFAULT_STAKE_AMOUNT, "Staked balance should match")
  //   assertBn(locked, DEFAULT_LOCK_AMOUNT, "Locked balance should match")
  // })

  it('fails locking 0 tokens', async () => {
    await approveAndStake({ staking, from: owner })
    await assertRevert(staking.allowManagerAndLock(0, user1, 1, EMPTY_DATA), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  })

  // it('fails locking without enough allowance', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.allowManagerAndLock(2, user1, 1, EMPTY_DATA), STAKING_ERRORS.ERROR_NOT_ENOUGH_ALLOWANCE)
  // })

  // it('fails locking more tokens than staked', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.allowManagerAndLock(DEFAULT_STAKE_AMOUNT.add(bn(1)), user1, DEFAULT_STAKE_AMOUNT.add(bn(1)), EMPTY_DATA), STAKING_ERRORS.ERROR_NOT_ENOUGH_BALANCE)
  // })

  // it('fails locking if already locked', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.allowManagerAndLock(DEFAULT_STAKE_AMOUNT, user1, DEFAULT_STAKE_AMOUNT, "0x02"), STAKING_ERRORS.ERROR_LOCK_ALREADY_EXISTS)
  // })

  // it('fails unstaking locked tokens', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await assertRevert(staking.unstake(DEFAULT_STAKE_AMOUNT, EMPTY_DATA), STAKING_ERRORS.ERROR_NOT_ENOUGH_BALANCE)
  // })

  // it('creates a new allowance', async () => {
  //   await staking.allowManager(user1, DEFAULT_LOCK_AMOUNT, EMPTY_DATA)

  //   const { _allowance  } = await staking.getLock(owner, user1)
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT, "allowed amount should match")
  // })

  // it('creates a new allowance and then lock manager locks', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await staking.allowManager(user1, DEFAULT_LOCK_AMOUNT, EMPTY_DATA)
  //   await staking.lock(owner, user1, DEFAULT_LOCK_AMOUNT, { from: user1 })

  //   // check lock values
  //   const { _amount, _allowance } = await staking.getLock(owner, user1)
  //   assertBn(_amount, DEFAULT_LOCK_AMOUNT, "locked amount should match")
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT, "locked allowance should match")

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT.sub(DEFAULT_LOCK_AMOUNT), "Unlocked balance should match")
  // })

  // it('fails creating allowance of 0 tokens', async () => {
  //   await assertRevert(staking.allowManager(user1, 0, EMPTY_DATA), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails creating allowance if lock exists', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })
  //   await assertRevert(staking.allowManager(user1, 1, EMPTY_DATA), STAKING_ERRORS.ERROR_LOCK_ALREADY_EXISTS)
  // })

  // it('increases allowance of existing lock', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await staking.increaseLockAllowance(user1, DEFAULT_LOCK_AMOUNT)

  //   const { _allowance } = await staking.getLock(owner, user1)
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT.mul(bn(2)), "allowed amount should match")
  // })

  // it('fails increasing allowance of non-existing', async () => {
  //   await assertRevert(staking.increaseLockAllowance(user1, 1), STAKING_ERRORS.ERROR_LOCK_DOES_NOT_EXIST)
  // })

  // it('fails increasing allowance of existing lock by 0', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await assertRevert(staking.increaseLockAllowance(user1, 0), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails increasing allowance of existing lock if not owner or manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await assertRevert(staking.increaseLockAllowance(user1, 1, { from: user2 }), STAKING_ERRORS.ERROR_LOCK_DOES_NOT_EXIST)
  // })

  // it('decreases allowance of existing lock by the owner', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await staking.allowManagerAndLock(DEFAULT_LOCK_AMOUNT, user1, DEFAULT_LOCK_AMOUNT.add(bn(1)), EMPTY_DATA)

  //   await staking.decreaseLockAllowance(owner, user1, 1, { from: owner })

  //   const { _allowance } = await staking.getLock(owner, user1)
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT, "allowed amount should match")
  // })

  // it('decreases allowance of existing lock by manager', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await staking.allowManagerAndLock(DEFAULT_LOCK_AMOUNT, user1, DEFAULT_LOCK_AMOUNT.add(bn(1)), EMPTY_DATA)

  //   await staking.decreaseLockAllowance(owner, user1, 1, { from: user1 })

  //   const { _allowance } = await staking.getLock(owner, user1)
  //   assertBn(_allowance, DEFAULT_LOCK_AMOUNT, "allowed amount should match")
  // })

  // it('fails decreasing allowance of existing lock by 0', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await assertRevert(staking.decreaseLockAllowance(owner, user1, 0), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails decreasing allowance of existing lock to 0', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await staking.unlock(owner, user1, DEFAULT_LOCK_AMOUNT, { from: user1 })

  //   await assertRevert(staking.decreaseLockAllowance(owner, user1, DEFAULT_LOCK_AMOUNT), STAKING_ERRORS.ERROR_ALLOWANCE_ZERO)
  // })

  // it('fails decreasing allowance to less than lock', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await staking.allowManagerAndLock(DEFAULT_LOCK_AMOUNT, user1, DEFAULT_LOCK_AMOUNT.add(bn(1)), EMPTY_DATA)

  //   await assertRevert(staking.decreaseLockAllowance(owner, user1, 2), STAKING_ERRORS.ERROR_NOT_ENOUGH_ALLOWANCE)
  // })

  // it('fails decreasing allowance by 3rd party', async () => {
  //   await approveAndStake({ staking, from: owner })
  //   await staking.allowManagerAndLock(DEFAULT_LOCK_AMOUNT, user1, DEFAULT_LOCK_AMOUNT.add(bn(1)), EMPTY_DATA)

  //   await assertRevert(staking.decreaseLockAllowance(owner, user1, 1, { from: user2 }), STAKING_ERRORS.ERROR_CANNOT_CHANGE_ALLOWANCE)
  // })

  // it('increases amount of existing lock', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await approveAndStake({ staking, from: owner })
  //   await staking.increaseLockAllowance(user1, DEFAULT_LOCK_AMOUNT)
  //   await staking.lock(owner, user1, DEFAULT_LOCK_AMOUNT)

  //   const { _amount } = await staking.getLock(owner, user1)
  //   assertBn(_amount, DEFAULT_LOCK_AMOUNT.mul(bn(2)), "locked amount should match")
  // })

  // it('fails increasing lock with 0 tokens', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.lock(owner, user1, 0), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails increasing lock with more tokens than staked', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.lock(owner, user1, DEFAULT_STAKE_AMOUNT.mul(bn(2)).add(bn(1))), STAKING_ERRORS.ERROR_NOT_ENOUGH_BALANCE)
  // })

  // it('fails increasing lock if not owner or manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   await approveAndStake({ staking, from: owner })
  //   await assertRevert(staking.lock(owner, user1, 1, { from: user2 }), STAKING_ERRORS.ERROR_SENDER_NOT_ALLOWED)
  // })

  // it('unlocks with only 1 lock, EOA manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, lockAmount: DEFAULT_LOCK_AMOUNT, from: owner })

  //   // unlock
  //   await staking.unlockAndRemoveManager(owner, user1, { from: user1 })

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), bn(0), "total locked doesn’t match")
  // })

  // it('unlocks with more than 1 lock, EOA manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })
  //   // lock again
  //   await staking.allowManagerAndLock(DEFAULT_LOCK_AMOUNT, user2, DEFAULT_LOCK_AMOUNT, EMPTY_DATA)

  //   const previousTotalLocked = await staking.lockedBalanceOf(owner)

  //   // unlock
  //   await staking.unlockAndRemoveManager(owner, user1, { from: user1 })

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT.sub(DEFAULT_LOCK_AMOUNT), "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), previousTotalLocked.sub(bn(DEFAULT_LOCK_AMOUNT)), "total locked doesn’t match")
  // })

  // it('unlocks completely, contract manager, called by owner', async () => {
  //   await lockManager.setResult(true)
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // unlock
  //   await staking.unlockAndRemoveManager(owner, lockManager.address, { from: owner })

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), bn(0), "total locked doesn’t match")
  // })

  // it('unlocks completely, contract manager, called by manager', async () => {
  //   await lockManager.setResult(true)
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // unlock
  //   await lockManager.unlockAndRemoveManager(staking.address, owner, lockManager.address)

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), bn(0), "total locked doesn’t match")
  // })

  // it('unlocks completely, contract manager, called by manager, even if condition is not satisfied', async () => {
  //   // not needed, is false by default
  //   //await lockManager.setResult(false)
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // unlock
  //   await lockManager.unlockAndRemoveManager(staking.address, owner, lockManager.address)

  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), bn(0), "total locked doesn’t match")
  // })

  // it('fails calling canUnlock, EOA manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   // call canUnlock
  //   await assertRevert(staking.canUnlock(owner, owner, user1, 0)) // no reason: it’s trying to call an EOA
  // })

  // it('can unlock if amount is zero', async () => {
  //   await staking.allowManager(user1, DEFAULT_LOCK_AMOUNT, EMPTY_DATA, { from: owner })
  //   assert.isTrue(await staking.canUnlock(owner, owner, user1, 0))
  // })

  // it('fails to unlock if it cannot unlock, EOA manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })

  //   // tries to unlock
  //   await assertRevert(staking.unlockAndRemoveManager(owner, user1)) // no reason: it’s trying to call an EOA
  // })

  // it('fails to unlock if can not unlock, contract manager, called by owner', async () => {
  //   // not needed, is false by default
  //   // await lockManager.setResult(false)
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // tries to unlock
  //   await assertRevert(staking.unlockAndRemoveManager(owner, lockManager.address, { from: owner }), STAKING_ERRORS.ERROR_CANNOT_UNLOCK)
  // })

  // it('fails to unlock if, contract manager, called by 3rd party (even if condition is true)', async () => {
  //   await lockManager.setResult(true)
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // tries to unlock
  //   await assertRevert(staking.unlockAndRemoveManager(owner, lockManager.address, { from: user1 }), STAKING_ERRORS.ERROR_CANNOT_UNLOCK)
  // })

  // it('transfers (slash) and unlocks (everything else) in one transaction', async () => {
  //   const totalLock = bigExp(120, 18)
  //   const transferAmount = bigExp(40, 18)

  //   await approveStakeAndLock({ staking, manager: user1, allowanceAmount: totalLock, lockAmount: totalLock, stakeAmount: totalLock, from: owner })

  //   // unlock and transfer
  //   await staking.slashAndUnlock(owner, user2, totalLock.sub(transferAmount), transferAmount, { from: user1 })

  //   assertBn(await staking.unlockedBalanceOf(owner), totalLock.sub(transferAmount), "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), bn(0), "total locked doesn’t match")
  //   // lock manager
  //   assertBn(await staking.unlockedBalanceOf(user1), bn(0), "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(user1), bn(0), "total locked doesn’t match")
  //   // recipient
  //   assertBn(await staking.unlockedBalanceOf(user2), transferAmount, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(user2), bn(0), "total locked doesn’t match")
  // })

  // it('transfers (slash) and unlocks in one transaction', async () => {
  //   const totalLock = bigExp(120, 18)
  //   const transferAmount = bigExp(40, 18)
  //   const decreaseAmount = bigExp(60, 18)

  //   await approveStakeAndLock({ staking, manager: user1, allowanceAmount: totalLock, lockAmount: totalLock, stakeAmount: totalLock, from: owner })

  //   // unlock and transfer
  //   await staking.slashAndUnlock(owner, user2, decreaseAmount, transferAmount, { from: user1 })

  //   assertBn(await staking.unlockedBalanceOf(owner), decreaseAmount, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(owner), totalLock.sub(decreaseAmount).sub(transferAmount), "total locked doesn’t match")
  //   // lock manager
  //   assertBn(await staking.unlockedBalanceOf(user1), bn(0), "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(user1), bn(0), "total locked doesn’t match")
  //   // recipient
  //   assertBn(await staking.unlockedBalanceOf(user2), transferAmount, "Unlocked balance should match")
  //   assertBn(await staking.lockedBalanceOf(user2), bn(0), "total locked doesn’t match")
  // })

  // it('fails to transfer (slash) and unlocks in one transaction if unlock amount is zero', async () => {
  //   const totalLock = bigExp(120, 18)
  //   const transferAmount = bigExp(40, 18)
  //   const decreaseAmount = bigExp(0, 18)

  //   await approveStakeAndLock({ staking, manager: user1, allowanceAmount: totalLock, lockAmount: totalLock, stakeAmount: totalLock, from: owner })

  //   // unlock and transfer
  //   await assertRevert(staking.slashAndUnlock(owner, user2, decreaseAmount, transferAmount, { from: user1 }), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails to transfer (slash) and unlock in one transaction if not owner nor manager', async () => {
  //   const totalLock = bigExp(120, 18)
  //   const transferAmount = bigExp(40, 18)
  //   const decreaseAmount = bigExp(60, 18)

  //   await approveStakeAndLock({ staking, manager: user1, allowanceAmount: totalLock, lockAmount: totalLock, stakeAmount: totalLock, from: owner })

  //   // unlock and transfer
  //   await assertRevert(staking.slashAndUnlock(owner, user2, decreaseAmount, transferAmount, { from: user2 }), STAKING_ERRORS.ERROR_NOT_ENOUGH_LOCK)
  // })

  // it('change lock amount', async () => {
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })
  //   const { _amount: amount1 } = await staking.getLock(owner, lockManager.address)
  //   assertBn(amount1, bn(DEFAULT_LOCK_AMOUNT), "Amount should match")
  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT.sub(DEFAULT_LOCK_AMOUNT), "Unlocked balance should match")

  //   // change amount
  //   const unlockAmount = DEFAULT_LOCK_AMOUNT.div(bn(2))
  //   await lockManager.unlock(staking.address, owner, unlockAmount)

  //   const { _amount: amount2 } = await staking.getLock(owner, lockManager.address)
  //   assertBn(amount2, unlockAmount, "Amount should match")
  //   assertBn(await staking.unlockedBalanceOf(owner), DEFAULT_STAKE_AMOUNT.sub(unlockAmount), "Unlocked balance should match")
  // })

  // it('fails to change lock amount to zero', async () => {
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // try to change amount
  //   await assertRevert(lockManager.unlock(staking.address, owner, 0), STAKING_ERRORS.ERROR_AMOUNT_ZERO)
  // })

  // it('fails to change lock amount to greater than before', async () => {
  //   await approveStakeAndLock({ staking, manager: lockManager.address, from: owner })

  //   // try to change amount
  //   await assertRevert(lockManager.unlock(staking.address, owner, DEFAULT_LOCK_AMOUNT.add(bn(1))), STAKING_ERRORS.ERROR_NOT_ENOUGH_LOCK)
  // })

  // it('change lock manager', async () => {
  //   await approveStakeAndLock({ staking, manager: user1, from: owner })
  //   assert.equal(await staking.canUnlock(user1, owner, user1, 0), true, "User 1 can unlock")
  //   assert.equal(await staking.canUnlock(user2, owner, user1, 0), false, "User 2 can not unlock")
  //   await assertRevert(staking.canUnlock(user2, owner, user2, 0), STAKING_ERRORS.ERROR_LOCK_DOES_NOT_EXIST) // it doesn’t exist

  //   // change manager
  //   await staking.setLockManager(owner, user2, { from: user1 })

  //   await assertRevert(staking.canUnlock(user1, owner, user1, 0), STAKING_ERRORS.ERROR_LOCK_DOES_NOT_EXIST) // it doesn’t exist
  //   assert.equal(await staking.canUnlock(user1, owner, user2, 0), false, "User 1 can not unlock")
  //   assert.equal(await staking.canUnlock(user2, owner, user2, 0), true, "User 2 can unlock")
  // })

  // it('fails to change lock manager if it doesn’t exist', async () => {
  //   await assertRevert(staking.setLockManager(owner, user2, { from: user1 }), STAKING_ERRORS.ERROR_LOCK_DOES_NOT_EXIST)
  // })
})