add safe ERC20 helpers
This commit is contained in:
22
contracts/token/SafeERC20.sol
Normal file
22
contracts/token/SafeERC20.sol
Normal file
@ -0,0 +1,22 @@
|
||||
pragma solidity ^0.4.11;
|
||||
|
||||
import './ERC20Basic.sol';
|
||||
import './ERC20.sol';
|
||||
|
||||
/**
|
||||
* @title SafeERC20
|
||||
* @dev Wrappers around ERC20 operations that throw on failure.
|
||||
*/
|
||||
library SafeERC20 {
|
||||
function safeTransfer(ERC20Basic token, address to, uint256 value) internal {
|
||||
assert(token.transfer(to, value));
|
||||
}
|
||||
|
||||
function safeTransferFrom(ERC20 token, address from, address to, uint256 value) internal {
|
||||
assert(token.transferFrom(from, to, value));
|
||||
}
|
||||
|
||||
function safeApprove(ERC20 token, address spender, uint256 value) internal {
|
||||
assert(token.approve(spender, value));
|
||||
}
|
||||
}
|
||||
27
test/SafeERC20.js
Normal file
27
test/SafeERC20.js
Normal file
@ -0,0 +1,27 @@
|
||||
import EVMThrow from './helpers/EVMThrow';
|
||||
|
||||
require('chai')
|
||||
.use(require('chai-as-promised'))
|
||||
.should();
|
||||
|
||||
const SafeERC20Helper = artifacts.require('./helpers/SafeERC20Helper.sol');
|
||||
|
||||
contract('SafeERC20', function () {
|
||||
|
||||
beforeEach(async function () {
|
||||
this.helper = await SafeERC20Helper.new();
|
||||
});
|
||||
|
||||
it('should throw on failed transfer', async function () {
|
||||
await this.helper.doFailingTransfer().should.be.rejectedWith(EVMThrow);
|
||||
});
|
||||
|
||||
it('should throw on failed transferFrom', async function () {
|
||||
await this.helper.doFailingTransferFrom().should.be.rejectedWith(EVMThrow);
|
||||
});
|
||||
|
||||
it('should throw on failed approve', async function () {
|
||||
await this.helper.doFailingApprove().should.be.rejectedWith(EVMThrow);
|
||||
});
|
||||
|
||||
});
|
||||
48
test/helpers/SafeERC20Helper.sol
Normal file
48
test/helpers/SafeERC20Helper.sol
Normal file
@ -0,0 +1,48 @@
|
||||
pragma solidity ^0.4.11;
|
||||
|
||||
import '../../contracts/token/ERC20.sol';
|
||||
import '../../contracts/token/SafeERC20.sol';
|
||||
|
||||
contract ERC20FailingMock is ERC20 {
|
||||
function transfer(address, uint256) returns (bool) {
|
||||
return false;
|
||||
}
|
||||
|
||||
function transferFrom(address, address, uint256) returns (bool) {
|
||||
return false;
|
||||
}
|
||||
|
||||
function approve(address, uint256) returns (bool) {
|
||||
return false;
|
||||
}
|
||||
|
||||
function balanceOf(address) constant returns (uint256) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
function allowance(address, address) constant returns (uint256) {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
contract SafeERC20Helper {
|
||||
using SafeERC20 for ERC20;
|
||||
|
||||
ERC20 token;
|
||||
|
||||
function SafeERC20Helper() {
|
||||
token = new ERC20FailingMock();
|
||||
}
|
||||
|
||||
function doFailingTransfer() {
|
||||
token.safeTransfer(0, 0);
|
||||
}
|
||||
|
||||
function doFailingTransferFrom() {
|
||||
token.safeTransferFrom(0, 0, 0);
|
||||
}
|
||||
|
||||
function doFailingApprove() {
|
||||
token.safeApprove(0, 0);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user