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

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

const CanReclaimToken = artifacts.require('CanReclaimToken');
const StandardTokenMock = artifacts.require('StandardTokenMock');

contract('CanReclaimToken', function ([_, owner, anyone]) {
  let token = null;
  let canReclaimToken = null;

  beforeEach(async function () {
    // Create contract and token
    token = await StandardTokenMock.new(owner, 100, { from: owner });
    canReclaimToken = await CanReclaimToken.new({ from: owner });

    // Force token into contract
    await token.transfer(canReclaimToken.address, 10, { from: owner });
    const startBalance = await token.balanceOf(canReclaimToken.address);
    assert.equal(startBalance, 10);
  });

  it('should allow owner to reclaim tokens', async function () {
    const ownerStartBalance = await token.balanceOf(owner);
    await canReclaimToken.reclaimToken(token.address, { from: owner });
    const ownerFinalBalance = await token.balanceOf(owner);
    const finalBalance = await token.balanceOf(canReclaimToken.address);
    assert.equal(finalBalance, 0);
    assert.equal(ownerFinalBalance - ownerStartBalance, 10);
  });

  it('should allow only owner to reclaim tokens', async function () {
    await expectThrow(
      canReclaimToken.reclaimToken(token.address, { from: anyone })
    );
  });
});