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