Add TokenKillable
This commit is contained in:
30
contracts/lifecycle/TokenKillable.sol
Normal file
30
contracts/lifecycle/TokenKillable.sol
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
pragma solidity ^0.4.8;
|
||||||
|
|
||||||
|
|
||||||
|
import "../ownership/Ownable.sol";
|
||||||
|
import "../token/ERC20Basic.sol";
|
||||||
|
|
||||||
|
/// @title TokenKillable:
|
||||||
|
/// @author Remco Bloemen <remco@2π.com>
|
||||||
|
///.Base contract that can be killed by owner. All funds in contract including
|
||||||
|
/// listed tokens will be sent to the owner
|
||||||
|
contract TokenKillable is Ownable {
|
||||||
|
|
||||||
|
/// @notice Terminate contract and refund to owner
|
||||||
|
/// @param tokens List of addresses of ERC20 or ERC20Basic token contracts to
|
||||||
|
// refund
|
||||||
|
/// @notice The called token contracts could try to re-enter this contract.
|
||||||
|
// Only supply token contracts you
|
||||||
|
function kill(address[] tokens) onlyOwner {
|
||||||
|
|
||||||
|
// Transfer tokens to owner
|
||||||
|
for(uint i = 0; i < tokens.length; i++) {
|
||||||
|
ERC20Basic token = ERC20Basic(tokens[i]);
|
||||||
|
uint256 balance = token.balanceOf(this);
|
||||||
|
token.transfer(owner, balance);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Transfer Eth to owner and terminate contract
|
||||||
|
selfdestruct(owner);
|
||||||
|
}
|
||||||
|
}
|
||||||
32
test/TokenKillable.js
Normal file
32
test/TokenKillable.js
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
'use strict';
|
||||||
|
|
||||||
|
var TokenKillable = artifacts.require('../contracts/lifecycle/TokenKillable.sol');
|
||||||
|
var StandardTokenMock = artifacts.require("./helpers/StandardTokenMock.sol");
|
||||||
|
require('./helpers/transactionMined.js');
|
||||||
|
|
||||||
|
contract('TokenKillable', function(accounts) {
|
||||||
|
|
||||||
|
it('should send balance to owner after death', async function() {
|
||||||
|
let killable = await TokenKillable.new({from: accounts[0], value: web3.toWei('10','ether')});
|
||||||
|
let owner = await killable.owner();
|
||||||
|
let initBalance = web3.eth.getBalance(owner);
|
||||||
|
await killable.kill([], {from: owner});
|
||||||
|
let newBalance = web3.eth.getBalance(owner);
|
||||||
|
assert.isTrue(newBalance > initBalance);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should send tokens to owner after death', async function() {
|
||||||
|
let killable = await TokenKillable.new({from: accounts[0], value: web3.toWei('10','ether')});
|
||||||
|
let owner = await killable.owner();
|
||||||
|
let token = await StandardTokenMock.new(killable.address, 100);
|
||||||
|
let initContractBalance = await token.balanceOf(killable.address);
|
||||||
|
let initOwnerBalance = await token.balanceOf(owner);
|
||||||
|
assert.equal(initContractBalance, 100);
|
||||||
|
assert.equal(initOwnerBalance, 0);
|
||||||
|
await killable.kill([token.address], {from: owner});
|
||||||
|
let newContractBalance = await token.balanceOf(killable.address);
|
||||||
|
let newOwnerBalance = await token.balanceOf(owner);
|
||||||
|
assert.equal(newContractBalance, 0);
|
||||||
|
assert.equal(newOwnerBalance, 100);
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user