ef347ffccc
* Consolidted ERC20 Interface and Implementation Files * Fixed CanReclaimToken's tests to use StandardTokenMock instead of BasicTokenMock * Changed token's variable type in TokenTimelock to ERC20 * Merged the StandardBurnableToken with BurnableToken since it now inherits from StandardToken; Fixed TokenTimelock so it uses SafeERC20 for ERC20 * Fixed variable type for _token in TokenTimelock constructor * Fixed linting warning in BurnableToken * Added back burnFrom tests.
36 lines
1.3 KiB
JavaScript
36 lines
1.3 KiB
JavaScript
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 })
|
|
);
|
|
});
|
|
});
|