Pausable.test.js

const { assertRevert } = require('../helpers/assertRevert');
const expectEvent = require('../helpers/expectEvent');
const PausableMock = artifacts.require('PausableMock');

const BigNumber = web3.BigNumber;

require('chai')
  .use(require('chai-bignumber')(BigNumber))
  .should();

contract('Pausable', function () {
  beforeEach(async function () {
    this.Pausable = await PausableMock.new();
  });

  it('can perform normal process in non-pause', async function () {
    (await this.Pausable.count()).should.be.bignumber.equal(0);

    await this.Pausable.normalProcess();
    (await this.Pausable.count()).should.be.bignumber.equal(1);
  });

  it('can not perform normal process in pause', async function () {
    await this.Pausable.pause();
    (await this.Pausable.count()).should.be.bignumber.equal(0);

    await assertRevert(this.Pausable.normalProcess());
    (await this.Pausable.count()).should.be.bignumber.equal(0);
  });

  it('can not take drastic measure in non-pause', async function () {
    await assertRevert(this.Pausable.drasticMeasure());
    (await this.Pausable.drasticMeasureTaken()).should.equal(false);
  });

  it('can take a drastic measure in a pause', async function () {
    await this.Pausable.pause();
    await this.Pausable.drasticMeasure();
    (await this.Pausable.drasticMeasureTaken()).should.equal(true);
  });

  it('should resume allowing normal process after pause is over', async function () {
    await this.Pausable.pause();
    await this.Pausable.unpause();
    await this.Pausable.normalProcess();
    (await this.Pausable.count()).should.be.bignumber.equal(1);
  });

  it('should prevent drastic measure after pause is over', async function () {
    await this.Pausable.pause();
    await this.Pausable.unpause();

    await assertRevert(this.Pausable.drasticMeasure());

    (await this.Pausable.drasticMeasureTaken()).should.equal(false);
  });

  it('should log Pause and Unpause events appropriately', async function () {
    const setPauseLogs = (await this.Pausable.pause()).logs;
    expectEvent.inLogs(setPauseLogs, 'Paused');

    const setUnPauseLogs = (await this.Pausable.unpause()).logs;
    expectEvent.inLogs(setUnPauseLogs, 'Unpaused');
  });
});