NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Migrate finance tests to ethers.js (#4723)

Browse Source 
Co-authored-by: Hadrien Croubois <hadrien.croubois@gmail.com>
Co-authored-by: ernestognw <ernestognw@gmail.com>
This commit is contained in:
Renan Souza
2023-11-08 18:14:06 +00:00
committed by GitHub
parent 248be2fab0
commit f1f427ddaf
2 changed files with 60 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
test/finance/VestingWallet.behavior.js
View File

@ -1,54 +1,36 @@
const { time } = require('@nomicfoundation/hardhat-network-helpers');
const { expectEvent } = require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const { bigint: time } = require('../helpers/time');
function releasedEvent(token, amount) {
return token ? ['ERC20Released', { token: token.address, amount }] : ['EtherReleased', { amount }];
}
function shouldBehaveLikeVesting(beneficiary) {
function shouldBehaveLikeVesting() {
it('check vesting schedule', async function () {
const [vestedAmount, releasable, ...args] = this.token
? ['vestedAmount(address,uint64)', 'releasable(address)', this.token.address]
: ['vestedAmount(uint64)', 'releasable()'];
for (const timestamp of this.schedule) {
await time.increaseTo(timestamp);
await time.forward.timestamp(timestamp);
const vesting = this.vestingFn(timestamp);
expect(await this.mock.methods[vestedAmount](...args, timestamp)).to.be.bignumber.equal(vesting);
expect(await this.mock.methods[releasable](...args)).to.be.bignumber.equal(vesting);
expect(await this.mock.vestedAmount(...this.args, timestamp)).to.be.equal(vesting);
expect(await this.mock.releasable(...this.args)).to.be.equal(vesting);
}
});
it('execute vesting schedule', async function () {
const [release, ...args] = this.token ? ['release(address)', this.token.address] : ['release()'];
let released = web3.utils.toBN(0);
const before = await this.getBalance(beneficiary);
let released = 0n;
{
const receipt = await this.mock.methods[release](...args);
const tx = await this.mock.release(...this.args);
await expect(tx)
.to.emit(this.mock, this.releasedEvent)
.withArgs(...this.argsVerify, 0);
await expectEvent.inTransaction(receipt.tx, this.mock, ...releasedEvent(this.token, '0'));
await this.checkRelease(receipt, beneficiary, '0');
expect(await this.getBalance(beneficiary)).to.be.bignumber.equal(before);
await this.checkRelease(tx, 0n);
}
for (const timestamp of this.schedule) {
await time.setNextBlockTimestamp(timestamp);
await time.forward.timestamp(timestamp, false);
const vested = this.vestingFn(timestamp);
const receipt = await this.mock.methods[release](...args);
await expectEvent.inTransaction(receipt.tx, this.mock, ...releasedEvent(this.token, vested.sub(released)));
await this.checkRelease(receipt, beneficiary, vested.sub(released));
expect(await this.getBalance(beneficiary)).to.be.bignumber.equal(before.add(vested));
const tx = await this.mock.release(...this.args);
await expect(tx).to.emit(this.mock, this.releasedEvent);
await this.checkRelease(tx, vested - released);
released = vested;
}
});