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

const { BN, ether, shouldFail } = require('openzeppelin-test-helpers');

const IndividuallyCappedCrowdsaleImpl = artifacts.require('IndividuallyCappedCrowdsaleImpl');
const SimpleToken = artifacts.require('SimpleToken');
const { shouldBehaveLikePublicRole } = require('../behaviors/access/roles/PublicRole.behavior');

contract('IndividuallyCappedCrowdsale', function (
  [_, capper, otherCapper, wallet, alice, bob, charlie, other, ...otherAccounts]) {
  const rate = new BN(1);
  const capAlice = ether('10');
  const capBob = ether('2');
  const lessThanCapAlice = ether('6');
  const lessThanCapBoth = ether('1');
  const tokenSupply = new BN('10').pow(new BN('22'));

  beforeEach(async function () {
    this.token = await SimpleToken.new();
    this.crowdsale = await IndividuallyCappedCrowdsaleImpl.new(rate, wallet, this.token.address, { from: capper });
  });

  describe('capper role', function () {
    beforeEach(async function () {
      this.contract = this.crowdsale;
      await this.contract.addCapper(otherCapper, { from: capper });
    });

    shouldBehaveLikePublicRole(capper, otherCapper, otherAccounts, 'capper');
  });

  describe('individual caps', function () {
    it('sets a cap when the sender is a capper', async function () {
      await this.crowdsale.setCap(alice, capAlice, { from: capper });
      (await this.crowdsale.getCap(alice)).should.be.bignumber.equal(capAlice);
    });

    it('reverts when a non-capper sets a cap', async function () {
      await shouldFail.reverting.withMessage(this.crowdsale.setCap(alice, capAlice, { from: other }),
        'CapperRole: caller does not have the Capper role'
      );
    });

    context('with individual caps', function () {
      beforeEach(async function () {
        await this.crowdsale.setCap(alice, capAlice, { from: capper });
        await this.crowdsale.setCap(bob, capBob, { from: capper });
        await this.token.transfer(this.crowdsale.address, tokenSupply);
      });

      describe('accepting payments', function () {
        it('should accept payments within cap', async function () {
          await this.crowdsale.buyTokens(alice, { value: lessThanCapAlice });
          await this.crowdsale.buyTokens(bob, { value: lessThanCapBoth });
        });

        it('should reject payments outside cap', async function () {
          await this.crowdsale.buyTokens(alice, { value: capAlice });
          await shouldFail.reverting.withMessage(this.crowdsale.buyTokens(alice, { value: 1 }),
            'IndividuallyCappedCrowdsale: beneficiary\'s cap exceeded'
          );
        });

        it('should reject payments that exceed cap', async function () {
          await shouldFail.reverting.withMessage(this.crowdsale.buyTokens(alice, { value: capAlice.addn(1) }),
            'IndividuallyCappedCrowdsale: beneficiary\'s cap exceeded'
          );
          await shouldFail.reverting.withMessage(this.crowdsale.buyTokens(bob, { value: capBob.addn(1) }),
            'IndividuallyCappedCrowdsale: beneficiary\'s cap exceeded'
          );
        });

        it('should manage independent caps', async function () {
          await this.crowdsale.buyTokens(alice, { value: lessThanCapAlice });
          await shouldFail.reverting.withMessage(this.crowdsale.buyTokens(bob, { value: lessThanCapAlice }),
            'IndividuallyCappedCrowdsale: beneficiary\'s cap exceeded'
          );
        });

        it('should default to a cap of zero', async function () {
          await shouldFail.reverting.withMessage(this.crowdsale.buyTokens(charlie, { value: lessThanCapBoth }),
            'IndividuallyCappedCrowdsale: beneficiary\'s cap exceeded'
          );
        });
      });

      describe('reporting state', function () {
        it('should report correct cap', async function () {
          (await this.crowdsale.getCap(alice)).should.be.bignumber.equal(capAlice);
        });

        it('should report actual contribution', async function () {
          await this.crowdsale.buyTokens(alice, { value: lessThanCapAlice });
          (await this.crowdsale.getContribution(alice)).should.be.bignumber.equal(lessThanCapAlice);
        });
      });
    });
  });
});