You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.4 KiB
40 lines
1.4 KiB
const { expectRevert } = require('@openzeppelin/test-helpers');
|
|
|
|
const { expect } = require('chai');
|
|
|
|
const ReentrancyMock = artifacts.require('ReentrancyMock');
|
|
const ReentrancyAttack = artifacts.require('ReentrancyAttack');
|
|
|
|
contract('ReentrancyGuard', function (accounts) {
|
|
beforeEach(async function () {
|
|
this.reentrancyMock = await ReentrancyMock.new();
|
|
expect(await this.reentrancyMock.counter()).to.be.bignumber.equal('0');
|
|
});
|
|
|
|
it('nonReentrant function can be called', async function () {
|
|
expect(await this.reentrancyMock.counter()).to.be.bignumber.equal('0');
|
|
await this.reentrancyMock.callback();
|
|
expect(await this.reentrancyMock.counter()).to.be.bignumber.equal('1');
|
|
});
|
|
|
|
it('does not allow remote callback', async function () {
|
|
const attacker = await ReentrancyAttack.new();
|
|
await expectRevert(
|
|
this.reentrancyMock.countAndCall(attacker.address), 'ReentrancyAttack: failed call');
|
|
});
|
|
|
|
// The following are more side-effects than intended behavior:
|
|
// I put them here as documentation, and to monitor any changes
|
|
// in the side-effects.
|
|
it('does not allow local recursion', async function () {
|
|
await expectRevert(
|
|
this.reentrancyMock.countLocalRecursive(10), 'ReentrancyGuard: reentrant call',
|
|
);
|
|
});
|
|
|
|
it('does not allow indirect local recursion', async function () {
|
|
await expectRevert(
|
|
this.reentrancyMock.countThisRecursive(10), 'ReentrancyMock: failed call',
|
|
);
|
|
});
|
|
});
|
|
|