This repository has been archived by the owner on Mar 1, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 506
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
146 additions
and
143 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,143 +1,146 @@ | ||
| /* jshint esversion: 9 */ | ||
|
|
||
| import chai, { assert } from 'chai' | ||
| import chaiAsPromised from 'chai-as-promised' | ||
|
|
||
| import { DrainStakeManager } from '../../helpers/artifacts.js' | ||
|
|
||
| import deployer from '../../helpers/deployer.js' | ||
| import * as utils from '../../helpers/utils.js' | ||
|
|
||
| import { generateFirstWallets, mnemonics } from '../../helpers/wallets.js' | ||
|
|
||
| chai.use(chaiAsPromised).should() | ||
|
|
||
| describe('DrainStakeManager', async function (accounts) { | ||
| accounts = await ethers.getSigners() | ||
| accounts = accounts.map((account) => { | ||
| return account.address | ||
| }) | ||
|
|
||
| const owner = accounts[0] | ||
| describe('Upgrade and drain staking contract', async function () { | ||
| before(async function () { | ||
| this.wallets = generateFirstWallets(mnemonics, 10) | ||
|
|
||
| let contracts = await deployer.deployStakeManager(this.wallets) | ||
|
|
||
| this.governance = contracts.governance | ||
| this.stakeToken = contracts.stakeToken | ||
| this.stakeManager = contracts.stakeManager | ||
| this.proxy = contracts.stakeManagerProxy | ||
| this.stakeManagerImpl = contracts.stakeManagerImpl | ||
|
|
||
| this.gSafe = await deployer.deployGnosisMultisig(accounts.slice(0, 3)) | ||
| await this.stakeManager.transferOwnership(this.gSafe.address) | ||
| }) | ||
|
|
||
| it('must have some tokens', async function () { | ||
| const amount = web3.utils.toWei('90000') | ||
| await this.stakeToken.mint(this.stakeManager.address, amount) | ||
| ;(await this.stakeToken.balanceOf(this.stakeManager.address)).toString().should.be.equal(amount.toString()) | ||
| }) | ||
|
|
||
| it('must lock stake manager', async function () { | ||
| await this.governance.update(this.stakeManager.address, this.stakeManager.interface.encodeFunctionData('lock')) | ||
| ;(await this.stakeManager.locked()).should.be.equal(true) | ||
| }) | ||
|
|
||
| it('must swap to drainable implementaion', async function () { | ||
| this.stakeManagerDrainable = await DrainStakeManager.deploy() | ||
| const result = await ( | ||
| await execSafe( | ||
| this.gSafe, | ||
| this.stakeManager.address, | ||
| this.proxy.interface.encodeFunctionData('updateImplementation', [this.stakeManagerDrainable.address]), | ||
| [accounts[0], accounts[1]] | ||
| ) | ||
| ).wait() | ||
| assert.equal(await this.proxy.implementation(), this.stakeManagerDrainable.address) | ||
| }) | ||
|
|
||
| it('must fail draining when not drained owner', async function () { | ||
| const balance = await this.stakeToken.balanceOf(this.stakeManager.address) | ||
| try { | ||
| await this.stakeManagerDrainable.drain(owner, balance) | ||
| assert.fail('Funds should not be drained') | ||
| } catch (error) { | ||
| assert(error.message.search('revert') >= 0, "Expected revert, got '" + error + "' instead") | ||
| } | ||
| }) | ||
|
|
||
| it('must drain all funds when drained by owner (Gnosis safe)', async function () { | ||
| const balance = await this.stakeToken.balanceOf(this.stakeManager.address) | ||
| const data = this.stakeManagerDrainable.interface.encodeFunctionData('drain', [owner, balance.toString()]) | ||
| await execSafe(this.gSafe, this.stakeManager.address, data, [accounts[1], accounts[2]]) | ||
| assert.equal((await this.stakeToken.balanceOf(this.stakeManager.address)).toString(), '0') | ||
| }) | ||
|
|
||
| it('must swap back to normal implementaion', async function () { | ||
| await execSafe( | ||
| this.gSafe, | ||
| this.stakeManager.address, | ||
| this.proxy.interface.encodeFunctionData('updateImplementation', [this.stakeManagerImpl.address]), | ||
| [accounts[2], accounts[0]] | ||
| ) | ||
| assert.equal(await this.proxy.implementation(), this.stakeManagerImpl.address) | ||
| }) | ||
|
|
||
| it('must unlock stake manager', async function () { | ||
| await this.governance.update(this.stakeManager.address, this.stakeManager.interface.encodeFunctionData('unlock')) | ||
| ;(await this.stakeManager.locked()).should.be.equal(false) | ||
| }) | ||
| }) | ||
| }) | ||
|
|
||
| async function execSafe(gSafe, address, data, confirmingAccounts) { | ||
| const params = safeParams(address, data, await gSafe.nonce()) | ||
| const txHash = await gSafe.getTransactionHash(...params) | ||
| let signatureBytes = '0x' | ||
| confirmingAccounts.sort() | ||
| for (var i = 0; i < confirmingAccounts.length; i++) { | ||
| // Adjust v (it is + 27 => EIP-155 and + 4 to differentiate them from typed data signatures in the Safe) | ||
| let signature = (await ethSign(confirmingAccounts[i], txHash)) | ||
| .replace('0x', '') | ||
| signatureBytes += signature | ||
| } | ||
| params[9] = signatureBytes | ||
| return gSafe.execTransaction(...params) | ||
| } | ||
|
|
||
| function safeParams(address, data, nonce) { | ||
| return [ | ||
| address, | ||
| 0, // value | ||
| data, | ||
| 0, // Operation.Call | ||
| 0, // safeTxGas | ||
| 4000000, // baseGas | ||
| 0, // gasPrice | ||
| utils.ZeroAddress, // gasToken | ||
| utils.ZeroAddress, // refundReceiver | ||
| nonce.toString() | ||
| ] | ||
| } | ||
|
|
||
| function ethSign(account, hash) { | ||
| return new Promise(function (resolve, reject) { | ||
| web3.currentProvider.send( | ||
| { | ||
| jsonrpc: '2.0', | ||
| method: 'eth_sign', | ||
| params: [account, hash], | ||
| id: new Date().getTime() | ||
| }, | ||
| function (err, response) { | ||
| if (err) { | ||
| return reject(err) | ||
| } | ||
| resolve(response.result) | ||
| } | ||
| ) | ||
| }) | ||
| } | ||
| /* jshint esversion: 9 */ | ||
|
|
||
| import chai, { assert } from 'chai' | ||
| import chaiAsPromised from 'chai-as-promised' | ||
|
|
||
| import { DrainStakeManager } from '../../helpers/artifacts.js' | ||
|
|
||
| import deployer from '../../helpers/deployer.js' | ||
| import * as utils from '../../helpers/utils.js' | ||
|
|
||
| import { generateFirstWallets, mnemonics } from '../../helpers/wallets.js' | ||
|
|
||
| chai.use(chaiAsPromised).should() | ||
|
|
||
| describe('DrainStakeManager', async function (accounts) { | ||
| accounts = await ethers.getSigners() | ||
| accounts = accounts.map((account) => { | ||
| return account.address | ||
| }) | ||
|
|
||
| const owner = accounts[0] | ||
| describe('Upgrade and drain staking contract', async function () { | ||
| before(async function () { | ||
| this.wallets = generateFirstWallets(mnemonics, 10) | ||
|
|
||
| let contracts = await deployer.deployStakeManager(this.wallets) | ||
|
|
||
| this.governance = contracts.governance | ||
| this.stakeToken = contracts.stakeToken | ||
| this.stakeManager = contracts.stakeManager | ||
| this.proxy = contracts.stakeManagerProxy | ||
| this.stakeManagerImpl = contracts.stakeManagerImpl | ||
|
|
||
| this.gSafe = await deployer.deployGnosisMultisig(accounts.slice(0, 3)) | ||
| await this.stakeManager.transferOwnership(this.gSafe.address) | ||
| }) | ||
|
|
||
| it('must have some tokens', async function () { | ||
| const amount = web3.utils.toWei('90000') | ||
| await this.stakeToken.mint(this.stakeManager.address, amount) | ||
| ;(await this.stakeToken.balanceOf(this.stakeManager.address)).toString().should.be.equal(amount.toString()) | ||
| }) | ||
|
|
||
| it('must lock stake manager', async function () { | ||
| await this.governance.update(this.stakeManager.address, this.stakeManager.interface.encodeFunctionData('lock')) | ||
| ;(await this.stakeManager.locked()).should.be.equal(true) | ||
| }) | ||
|
|
||
| it('must swap to drainable implementaion', async function () { | ||
| this.stakeManagerDrainable = await DrainStakeManager.deploy() | ||
| const result = await ( | ||
| await execSafe( | ||
| this.gSafe, | ||
| this.stakeManager.address, | ||
| this.proxy.interface.encodeFunctionData('updateImplementation', [this.stakeManagerDrainable.address]), | ||
| [accounts[0], accounts[1]] | ||
| ) | ||
| ).wait() | ||
| assert.equal(await this.proxy.implementation(), this.stakeManagerDrainable.address) | ||
| }) | ||
|
|
||
| it('must fail draining when not drained owner', async function () { | ||
| const balance = await this.stakeToken.balanceOf(this.stakeManager.address) | ||
| try { | ||
| await this.stakeManagerDrainable.drain(owner, balance) | ||
| assert.fail('Funds should not be drained') | ||
| } catch (error) { | ||
| assert(error.message.search('revert') >= 0, "Expected revert, got '" + error + "' instead") | ||
| } | ||
| }) | ||
|
|
||
| it('must drain all funds when drained by owner (Gnosis safe)', async function () { | ||
| const balance = await this.stakeToken.balanceOf(this.stakeManager.address) | ||
| const data = this.stakeManagerDrainable.interface.encodeFunctionData('drain', [owner, balance.toString()]) | ||
| await execSafe(this.gSafe, this.stakeManager.address, data, [accounts[1], accounts[2]]) | ||
| assert.equal((await this.stakeToken.balanceOf(this.stakeManager.address)).toString(), '0') | ||
| }) | ||
|
|
||
| it('must swap back to normal implementaion', async function () { | ||
| await execSafe( | ||
| this.gSafe, | ||
| this.stakeManager.address, | ||
| this.proxy.interface.encodeFunctionData('updateImplementation', [this.stakeManagerImpl.address]), | ||
| [accounts[2], accounts[0]] | ||
| ) | ||
| assert.equal(await this.proxy.implementation(), this.stakeManagerImpl.address) | ||
| }) | ||
|
|
||
| it('must unlock stake manager', async function () { | ||
| await this.governance.update(this.stakeManager.address, this.stakeManager.interface.encodeFunctionData('unlock')) | ||
| ;(await this.stakeManager.locked()).should.be.equal(false) | ||
| }) | ||
| }) | ||
| }) | ||
|
|
||
| async function execSafe(gSafe, address, data, confirmingAccounts) { | ||
| const params = safeParams(address, data, await gSafe.nonce()) | ||
| const txHash = await gSafe.getTransactionHash(...params) | ||
| let signatureBytes = '0x' | ||
| confirmingAccounts.sort() | ||
| for (var i = 0; i < confirmingAccounts.length; i++) { | ||
| // Adjust v (it is + 4 to differentiate them from typed data signatures in the Safe) | ||
| // ethSign already returns 27 or 28 | ||
| let signature = (await ethSign(confirmingAccounts[i], txHash)) | ||
| .replace('0x', '') | ||
| .replace(/1b$/, '1f') | ||
| .replace(/1c$/, '20') | ||
| signatureBytes += signature | ||
| } | ||
| params[9] = signatureBytes | ||
| return gSafe.execTransaction(...params) | ||
| } | ||
|
|
||
| function safeParams(address, data, nonce) { | ||
| return [ | ||
| address, | ||
| 0, // value | ||
| data, | ||
| 0, // Operation.Call | ||
| 0, // safeTxGas | ||
| 4000000, // baseGas | ||
| 0, // gasPrice | ||
| utils.ZeroAddress, // gasToken | ||
| utils.ZeroAddress, // refundReceiver | ||
| nonce.toString() | ||
| ] | ||
| } | ||
|
|
||
| function ethSign(account, hash) { | ||
| return new Promise(function (resolve, reject) { | ||
| web3.currentProvider.send( | ||
| { | ||
| jsonrpc: '2.0', | ||
| method: 'eth_sign', | ||
| params: [account, hash], | ||
| id: new Date().getTime() | ||
| }, | ||
| function (err, response) { | ||
| if (err) { | ||
| return reject(err) | ||
| } | ||
| resolve(response.result) | ||
| } | ||
| ) | ||
| }) | ||
| } |