openzeppelin-contracts/test/ownership/DelayedClaimable.test.js

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

const DelayedClaimable = artifacts.require('DelayedClaimable');

contract('DelayedClaimable', function (accounts) {
  beforeEach(async function () {
    this.delayedClaimable = await DelayedClaimable.new();
  });

  it('can set claim blocks', async function () {
    await this.delayedClaimable.transferOwnership(accounts[2]);
    await this.delayedClaimable.setLimits(0, 1000);
    const end = await this.delayedClaimable.end();
    assert.equal(end, 1000);
    const start = await this.delayedClaimable.start();
    assert.equal(start, 0);
  });

  it('changes pendingOwner after transfer successful', async function () {
    await this.delayedClaimable.transferOwnership(accounts[2]);
    await this.delayedClaimable.setLimits(0, 1000);
    const end = await this.delayedClaimable.end();
    assert.equal(end, 1000);
    const start = await this.delayedClaimable.start();
    assert.equal(start, 0);
    const pendingOwner = await this.delayedClaimable.pendingOwner();
    assert.equal(pendingOwner, accounts[2]);
    await this.delayedClaimable.claimOwnership({ from: accounts[2] });
    const owner = await this.delayedClaimable.owner();
    assert.equal(owner, accounts[2]);
  });

  it('changes pendingOwner after transfer fails', async function () {
    await this.delayedClaimable.transferOwnership(accounts[1]);
    await this.delayedClaimable.setLimits(100, 110);
    const end = await this.delayedClaimable.end();
    assert.equal(end, 110);
    const start = await this.delayedClaimable.start();
    assert.equal(start, 100);
    const pendingOwner = await this.delayedClaimable.pendingOwner();
    assert.equal(pendingOwner, accounts[1]);
    await assertRevert(this.delayedClaimable.claimOwnership({ from: accounts[1] }));
    const owner = await this.delayedClaimable.owner();
    assert.isTrue(owner !== accounts[1]);
  });

  it('set end and start invalid values fail', async function () {
    await this.delayedClaimable.transferOwnership(accounts[1]);
    await assertRevert(this.delayedClaimable.setLimits(1001, 1000));
  });
});