openzeppelin-contracts/test/Pausable.test.js

const assertRevert = require('./helpers/assertRevert');
const PausableMock = artifacts.require('mocks/PausableMock.sol');

contract('Pausable', function (accounts) {
  it('can perform normal process in non-pause', async function () {
    let Pausable = await PausableMock.new();
    let count0 = await Pausable.count();
    assert.equal(count0, 0);

    await Pausable.normalProcess();
    let count1 = await Pausable.count();
    assert.equal(count1, 1);
  });

  it('can not perform normal process in pause', async function () {
    let Pausable = await PausableMock.new();
    await Pausable.pause();
    let count0 = await Pausable.count();
    assert.equal(count0, 0);

    try {
      await Pausable.normalProcess();
      assert.fail('should have thrown before');
    } catch (error) {
      assertRevert(error);
    }
    let count1 = await Pausable.count();
    assert.equal(count1, 0);
  });

  it('can not take drastic measure in non-pause', async function () {
    let Pausable = await PausableMock.new();
    try {
      await Pausable.drasticMeasure();
      assert.fail('should have thrown before');
    } catch (error) {
      assertRevert(error);
    }
    const drasticMeasureTaken = await Pausable.drasticMeasureTaken();
    assert.isFalse(drasticMeasureTaken);
  });

  it('can take a drastic measure in a pause', async function () {
    let Pausable = await PausableMock.new();
    await Pausable.pause();
    await Pausable.drasticMeasure();
    let drasticMeasureTaken = await Pausable.drasticMeasureTaken();

    assert.isTrue(drasticMeasureTaken);
  });

  it('should resume allowing normal process after pause is over', async function () {
    let Pausable = await PausableMock.new();
    await Pausable.pause();
    await Pausable.unpause();
    await Pausable.normalProcess();
    let count0 = await Pausable.count();

    assert.equal(count0, 1);
  });

  it('should prevent drastic measure after pause is over', async function () {
    let Pausable = await PausableMock.new();
    await Pausable.pause();
    await Pausable.unpause();
    try {
      await Pausable.drasticMeasure();
      assert.fail('should have thrown before');
    } catch (error) {
      assertRevert(error);
    }

    const drasticMeasureTaken = await Pausable.drasticMeasureTaken();
    assert.isFalse(drasticMeasureTaken);
  });
});