You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.3 KiB
34 lines
1.3 KiB
const { expectThrow } = require('../helpers/expectThrow');
|
|
|
|
const CanReclaimToken = artifacts.require('CanReclaimToken');
|
|
const BasicTokenMock = artifacts.require('BasicTokenMock');
|
|
|
|
contract('CanReclaimToken', function ([_, owner, anyone]) {
|
|
let token = null;
|
|
let canReclaimToken = null;
|
|
|
|
beforeEach(async function () {
|
|
// Create contract and token
|
|
token = await BasicTokenMock.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 })
|
|
);
|
|
});
|
|
});
|
|
|