b0da0fded0
* Replaced assertJump, assertRevert and expectThrow with shouldFail. * Fixed linter errors. * Fixed typo. * Made the helpers async.
34 lines
1.1 KiB
JavaScript
34 lines
1.1 KiB
JavaScript
const shouldFail = require('../helpers/shouldFail');
|
|
const ReentrancyMock = artifacts.require('ReentrancyMock');
|
|
const ReentrancyAttack = artifacts.require('ReentrancyAttack');
|
|
|
|
const BigNumber = web3.BigNumber;
|
|
|
|
require('chai')
|
|
.use(require('chai-bignumber')(BigNumber))
|
|
.should();
|
|
|
|
contract('ReentrancyGuard', function () {
|
|
beforeEach(async function () {
|
|
this.reentrancyMock = await ReentrancyMock.new();
|
|
(await this.reentrancyMock.counter()).should.be.bignumber.equal(0);
|
|
});
|
|
|
|
it('should not allow remote callback', async function () {
|
|
const attacker = await ReentrancyAttack.new();
|
|
await shouldFail.reverting(this.reentrancyMock.countAndCall(attacker.address));
|
|
});
|
|
|
|
// 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('should not allow local recursion', async function () {
|
|
await shouldFail.reverting(this.reentrancyMock.countLocalRecursive(10));
|
|
});
|
|
|
|
it('should not allow indirect local recursion', async function () {
|
|
await shouldFail.reverting(this.reentrancyMock.countThisRecursive(10));
|
|
});
|
|
});
|