openzeppelin-contracts/test/crowdsale/WhitelistedCrowdsale.test.js

const { ether } = require('../helpers/ether');
const { expectThrow } = require('../helpers/expectThrow');

const BigNumber = web3.BigNumber;

require('chai')
  .should();

const WhitelistedCrowdsale = artifacts.require('WhitelistedCrowdsaleImpl');
const SimpleToken = artifacts.require('SimpleToken');

contract('WhitelistedCrowdsale', function ([_, wallet, authorized, unauthorized, anotherAuthorized]) {
  const rate = 1;
  const value = ether(42);
  const tokenSupply = new BigNumber('1e22');

  describe('single user whitelisting', function () {
    beforeEach(async function () {
      this.token = await SimpleToken.new();
      this.crowdsale = await WhitelistedCrowdsale.new(rate, wallet, this.token.address);
      await this.token.transfer(this.crowdsale.address, tokenSupply);
      await this.crowdsale.addAddressToWhitelist(authorized);
    });

    describe('accepting payments', function () {
      it('should accept payments to whitelisted (from whichever buyers)', async function () {
        await this.crowdsale.sendTransaction({ value, from: authorized });
        await this.crowdsale.buyTokens(authorized, { value: value, from: authorized });
        await this.crowdsale.buyTokens(authorized, { value: value, from: unauthorized });
      });

      it('should reject payments to not whitelisted (from whichever buyers)', async function () {
        await expectThrow(this.crowdsale.sendTransaction({ value, from: unauthorized }));
        await expectThrow(this.crowdsale.buyTokens(unauthorized, { value: value, from: unauthorized }));
        await expectThrow(this.crowdsale.buyTokens(unauthorized, { value: value, from: authorized }));
      });

      it('should reject payments to addresses removed from whitelist', async function () {
        await this.crowdsale.removeAddressFromWhitelist(authorized);
        await expectThrow(this.crowdsale.buyTokens(authorized, { value: value, from: authorized }));
      });
    });

    describe('reporting whitelisted', function () {
      it('should correctly report whitelisted addresses', async function () {
        (await this.crowdsale.whitelist(authorized)).should.equal(true);
        (await this.crowdsale.whitelist(unauthorized)).should.equal(false);
      });
    });
  });

  describe('many user whitelisting', function () {
    beforeEach(async function () {
      this.token = await SimpleToken.new();
      this.crowdsale = await WhitelistedCrowdsale.new(rate, wallet, this.token.address);
      await this.token.transfer(this.crowdsale.address, tokenSupply);
      await this.crowdsale.addAddressesToWhitelist([authorized, anotherAuthorized]);
    });

    describe('accepting payments', function () {
      it('should accept payments to whitelisted (from whichever buyers)', async function () {
        await this.crowdsale.buyTokens(authorized, { value: value, from: authorized });
        await this.crowdsale.buyTokens(authorized, { value: value, from: unauthorized });
        await this.crowdsale.buyTokens(anotherAuthorized, { value: value, from: authorized });
        await this.crowdsale.buyTokens(anotherAuthorized, { value: value, from: unauthorized });
      });

      it('should reject payments to not whitelisted (with whichever buyers)', async function () {
        await expectThrow(this.crowdsale.send(value));
        await expectThrow(this.crowdsale.buyTokens(unauthorized, { value: value, from: unauthorized }));
        await expectThrow(this.crowdsale.buyTokens(unauthorized, { value: value, from: authorized }));
      });

      it('should reject payments to addresses removed from whitelist', async function () {
        await this.crowdsale.removeAddressFromWhitelist(anotherAuthorized);
        await this.crowdsale.buyTokens(authorized, { value: value, from: authorized });
        await expectThrow(this.crowdsale.buyTokens(anotherAuthorized, { value: value, from: authorized }));
      });
    });

    describe('reporting whitelisted', function () {
      it('should correctly report whitelisted addresses', async function () {
        (await this.crowdsale.whitelist(authorized)).should.equal(true);
        (await this.crowdsale.whitelist(anotherAuthorized)).should.equal(true);
        (await this.crowdsale.whitelist(unauthorized)).should.equal(false);
      });
    });
  });
});