3682c6575c
* Error handling in ERC20 and ERC721 * Added message string for require. * Fixed solhint errors. * Updated PR as per issue #1709 * changes as per #1709 and openzeppelin forum. * Changes in require statement * Changes in require statement * build pipeline fix * Changes as per @nventuro's comment. * Update revert reason strings. * Fianal update of revert reason strings. * WIP: Updating reason strings in test cases * WIP: Added changes to ERC20 and ERC721 * Fixes linting errors in *.tes.js files * Achieved 100% code coverage * Updated the test cases with shouldFail.reverting.withMessage() * Fix package-lock. * address review comments * fix linter issues * fix remaining revert reasons
58 lines
2.4 KiB
JavaScript
58 lines
2.4 KiB
JavaScript
const { BN, ether, shouldFail } = require('openzeppelin-test-helpers');
|
|
|
|
const WhitelistCrowdsale = artifacts.require('WhitelistCrowdsaleImpl');
|
|
const SimpleToken = artifacts.require('SimpleToken');
|
|
|
|
contract('WhitelistCrowdsale', function ([_, wallet, whitelister, whitelisted, otherWhitelisted, other]) {
|
|
const rate = new BN(1);
|
|
const value = ether('42');
|
|
const tokenSupply = new BN('10').pow(new BN('22'));
|
|
|
|
beforeEach(async function () {
|
|
this.token = await SimpleToken.new({ from: whitelister });
|
|
this.crowdsale = await WhitelistCrowdsale.new(rate, wallet, this.token.address, { from: whitelister });
|
|
await this.token.transfer(this.crowdsale.address, tokenSupply, { from: whitelister });
|
|
});
|
|
|
|
async function purchaseShouldSucceed (crowdsale, beneficiary, value) {
|
|
await crowdsale.buyTokens(beneficiary, { from: beneficiary, value });
|
|
await crowdsale.sendTransaction({ from: beneficiary, value });
|
|
}
|
|
|
|
async function purchaseShouldFail (crowdsale, beneficiary, value) {
|
|
await shouldFail.reverting.withMessage(crowdsale.buyTokens(beneficiary, { from: beneficiary, value }),
|
|
'WhitelistCrowdsale: beneficiary doesn\'t have the Whitelisted role'
|
|
);
|
|
await shouldFail.reverting.withMessage(crowdsale.sendTransaction({ from: beneficiary, value }),
|
|
'WhitelistCrowdsale: beneficiary doesn\'t have the Whitelisted role'
|
|
);
|
|
}
|
|
|
|
context('with no whitelisted addresses', function () {
|
|
it('rejects all purchases', async function () {
|
|
await purchaseShouldFail(this.crowdsale, other, value);
|
|
await purchaseShouldFail(this.crowdsale, whitelisted, value);
|
|
});
|
|
});
|
|
|
|
context('with whitelisted addresses', function () {
|
|
beforeEach(async function () {
|
|
await this.crowdsale.addWhitelisted(whitelisted, { from: whitelister });
|
|
await this.crowdsale.addWhitelisted(otherWhitelisted, { from: whitelister });
|
|
});
|
|
|
|
it('accepts purchases with whitelisted beneficiaries', async function () {
|
|
await purchaseShouldSucceed(this.crowdsale, whitelisted, value);
|
|
await purchaseShouldSucceed(this.crowdsale, otherWhitelisted, value);
|
|
});
|
|
|
|
it('rejects purchases from whitelisted addresses with non-whitelisted beneficiaries', async function () {
|
|
await shouldFail(this.crowdsale.buyTokens(other, { from: whitelisted, value }));
|
|
});
|
|
|
|
it('rejects purchases with non-whitelisted beneficiaries', async function () {
|
|
await purchaseShouldFail(this.crowdsale, other, value);
|
|
});
|
|
});
|
|
});
|