You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.6 KiB
JavaScript
69 lines
2.6 KiB
JavaScript
const { BN, expectRevert } = require('@openzeppelin/test-helpers');
|
|
|
|
const ERC20MulticallMock = artifacts.require('$ERC20MulticallMock');
|
|
|
|
contract('Multicall', function (accounts) {
|
|
const [deployer, alice, bob] = accounts;
|
|
const amount = 12000;
|
|
|
|
beforeEach(async function () {
|
|
this.multicallToken = await ERC20MulticallMock.new('name', 'symbol');
|
|
await this.multicallToken.$_mint(deployer, amount);
|
|
});
|
|
|
|
it('batches function calls', async function () {
|
|
expect(await this.multicallToken.balanceOf(alice)).to.be.bignumber.equal(new BN('0'));
|
|
expect(await this.multicallToken.balanceOf(bob)).to.be.bignumber.equal(new BN('0'));
|
|
|
|
await this.multicallToken.multicall(
|
|
[
|
|
this.multicallToken.contract.methods.transfer(alice, amount / 2).encodeABI(),
|
|
this.multicallToken.contract.methods.transfer(bob, amount / 3).encodeABI(),
|
|
],
|
|
{ from: deployer },
|
|
);
|
|
|
|
expect(await this.multicallToken.balanceOf(alice)).to.be.bignumber.equal(new BN(amount / 2));
|
|
expect(await this.multicallToken.balanceOf(bob)).to.be.bignumber.equal(new BN(amount / 3));
|
|
});
|
|
|
|
it('returns an array with the result of each call', async function () {
|
|
const MulticallTest = artifacts.require('MulticallTest');
|
|
const multicallTest = await MulticallTest.new({ from: deployer });
|
|
await this.multicallToken.transfer(multicallTest.address, amount, { from: deployer });
|
|
expect(await this.multicallToken.balanceOf(multicallTest.address)).to.be.bignumber.equal(new BN(amount));
|
|
|
|
const recipients = [alice, bob];
|
|
const amounts = [amount / 2, amount / 3].map(n => new BN(n));
|
|
|
|
await multicallTest.checkReturnValues(this.multicallToken.address, recipients, amounts);
|
|
});
|
|
|
|
it('reverts previous calls', async function () {
|
|
expect(await this.multicallToken.balanceOf(alice)).to.be.bignumber.equal(new BN('0'));
|
|
|
|
const call = this.multicallToken.multicall(
|
|
[
|
|
this.multicallToken.contract.methods.transfer(alice, amount).encodeABI(),
|
|
this.multicallToken.contract.methods.transfer(bob, amount).encodeABI(),
|
|
],
|
|
{ from: deployer },
|
|
);
|
|
|
|
await expectRevert(call, 'ERC20: transfer amount exceeds balance');
|
|
expect(await this.multicallToken.balanceOf(alice)).to.be.bignumber.equal(new BN('0'));
|
|
});
|
|
|
|
it('bubbles up revert reasons', async function () {
|
|
const call = this.multicallToken.multicall(
|
|
[
|
|
this.multicallToken.contract.methods.transfer(alice, amount).encodeABI(),
|
|
this.multicallToken.contract.methods.transfer(bob, amount).encodeABI(),
|
|
],
|
|
{ from: deployer },
|
|
);
|
|
|
|
await expectRevert(call, 'ERC20: transfer amount exceeds balance');
|
|
});
|
|
});
|