openzeppelin-contracts/test/access/manager/AuthorityUtils.test.js

require('@openzeppelin/test-helpers');

const AuthorityUtils = artifacts.require('$AuthorityUtils');
const NotAuthorityMock = artifacts.require('NotAuthorityMock');
const AuthorityNoDelayMock = artifacts.require('AuthorityNoDelayMock');
const AuthorityDelayMock = artifacts.require('AuthorityDelayMock');
const AuthorityNoResponse = artifacts.require('AuthorityNoResponse');

contract('AuthorityUtils', function (accounts) {
  const [user, other] = accounts;

  beforeEach(async function () {
    this.mock = await AuthorityUtils.new();
  });

  describe('canCallWithDelay', function () {
    describe('when authority does not have a canCall function', function () {
      beforeEach(async function () {
        this.authority = await NotAuthorityMock.new();
      });

      it('returns (immediate = 0, delay = 0)', async function () {
        const { immediate, delay } = await this.mock.$canCallWithDelay(
          this.authority.address,
          user,
          other,
          '0x12345678',
        );
        expect(immediate).to.equal(false);
        expect(delay).to.be.bignumber.equal('0');
      });
    });

    describe('when authority has no delay', function () {
      beforeEach(async function () {
        this.authority = await AuthorityNoDelayMock.new();
        this.immediate = true;
        await this.authority._setImmediate(this.immediate);
      });

      it('returns (immediate, delay = 0)', async function () {
        const { immediate, delay } = await this.mock.$canCallWithDelay(
          this.authority.address,
          user,
          other,
          '0x12345678',
        );
        expect(immediate).to.equal(this.immediate);
        expect(delay).to.be.bignumber.equal('0');
      });
    });

    describe('when authority replies with a delay', function () {
      beforeEach(async function () {
        this.authority = await AuthorityDelayMock.new();
        this.immediate = true;
        this.delay = web3.utils.toBN(42);
        await this.authority._setImmediate(this.immediate);
        await this.authority._setDelay(this.delay);
      });

      it('returns (immediate, delay)', async function () {
        const { immediate, delay } = await this.mock.$canCallWithDelay(
          this.authority.address,
          user,
          other,
          '0x12345678',
        );
        expect(immediate).to.equal(this.immediate);
        expect(delay).to.be.bignumber.equal(this.delay);
      });
    });

    describe('when authority replies with empty data', function () {
      beforeEach(async function () {
        this.authority = await AuthorityNoResponse.new();
      });

      it('returns (immediate = 0, delay = 0)', async function () {
        const { immediate, delay } = await this.mock.$canCallWithDelay(
          this.authority.address,
          user,
          other,
          '0x12345678',
        );
        expect(immediate).to.equal(false);
        expect(delay).to.be.bignumber.equal('0');
      });
    });
  });
});