openzeppelin-contracts/test/Pausable.js

'use strict';

const assertJump = require('./helpers/assertJump');
const PausableMock = artifacts.require('helpers/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) {
      assertJump(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) {
      assertJump(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) {
      assertJump(error);
    }

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

});