SafeERC20Helper.sol 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. pragma solidity ^0.5.2;
  2. import "../token/ERC20/IERC20.sol";
  3. import "../token/ERC20/SafeERC20.sol";
  4. contract ERC20FailingMock {
  5. uint256 private _allowance;
  6. // IERC20's functions are not pure, but these mock implementations are: to prevent Solidity from issuing warnings,
  7. // we write to a dummy state variable.
  8. uint256 private _dummy;
  9. function transfer(address, uint256) public returns (bool) {
  10. _dummy = 0;
  11. return false;
  12. }
  13. function transferFrom(address, address, uint256) public returns (bool) {
  14. _dummy = 0;
  15. return false;
  16. }
  17. function approve(address, uint256) public returns (bool) {
  18. _dummy = 0;
  19. return false;
  20. }
  21. function allowance(address, address) public view returns (uint256) {
  22. require(_dummy == 0);
  23. return 0;
  24. }
  25. }
  26. contract ERC20SucceedingMock {
  27. uint256 private _allowance;
  28. // IERC20's functions are not pure, but these mock implementations are: to prevent Solidity from issuing warnings,
  29. // we write to a dummy state variable.
  30. uint256 private _dummy;
  31. function transfer(address, uint256) public returns (bool) {
  32. _dummy = 0;
  33. return true;
  34. }
  35. function transferFrom(address, address, uint256) public returns (bool) {
  36. _dummy = 0;
  37. return true;
  38. }
  39. function approve(address, uint256) public returns (bool) {
  40. _dummy = 0;
  41. return true;
  42. }
  43. function setAllowance(uint256 allowance_) public {
  44. _allowance = allowance_;
  45. }
  46. function allowance(address, address) public view returns (uint256) {
  47. return _allowance;
  48. }
  49. }
  50. contract SafeERC20Helper {
  51. using SafeERC20 for IERC20;
  52. IERC20 private _failing;
  53. IERC20 private _succeeding;
  54. constructor () public {
  55. _failing = IERC20(address(new ERC20FailingMock()));
  56. _succeeding = IERC20(address(new ERC20SucceedingMock()));
  57. }
  58. function doFailingTransfer() public {
  59. _failing.safeTransfer(address(0), 0);
  60. }
  61. function doFailingTransferFrom() public {
  62. _failing.safeTransferFrom(address(0), address(0), 0);
  63. }
  64. function doFailingApprove() public {
  65. _failing.safeApprove(address(0), 0);
  66. }
  67. function doFailingIncreaseAllowance() public {
  68. _failing.safeIncreaseAllowance(address(0), 0);
  69. }
  70. function doFailingDecreaseAllowance() public {
  71. _failing.safeDecreaseAllowance(address(0), 0);
  72. }
  73. function doSucceedingTransfer() public {
  74. _succeeding.safeTransfer(address(0), 0);
  75. }
  76. function doSucceedingTransferFrom() public {
  77. _succeeding.safeTransferFrom(address(0), address(0), 0);
  78. }
  79. function doSucceedingApprove(uint256 amount) public {
  80. _succeeding.safeApprove(address(0), amount);
  81. }
  82. function doSucceedingIncreaseAllowance(uint256 amount) public {
  83. _succeeding.safeIncreaseAllowance(address(0), amount);
  84. }
  85. function doSucceedingDecreaseAllowance(uint256 amount) public {
  86. _succeeding.safeDecreaseAllowance(address(0), amount);
  87. }
  88. function setAllowance(uint256 allowance_) public {
  89. ERC20SucceedingMock(address(_succeeding)).setAllowance(allowance_);
  90. }
  91. function allowance() public view returns (uint256) {
  92. return _succeeding.allowance(address(0), address(0));
  93. }
  94. }