4544df47da
* All tests now use account names, and dont use accounts[0] (except ERC721) * Added account names to some missing contracts.
35 lines
1.3 KiB
JavaScript
35 lines
1.3 KiB
JavaScript
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 })
|
|
);
|
|
});
|
|
});
|