aave-protocol-v2/contracts/mocks/flashloan/MockFlashLoanReceiver.sol

67 lines
2.2 KiB
Solidity
Raw Normal View History

// SPDX-License-Identifier: agpl-3.0
pragma solidity ^0.6.8;
2020-08-20 10:44:51 +00:00
import {SafeMath} from '@openzeppelin/contracts/math/SafeMath.sol';
import {IERC20} from '@openzeppelin/contracts/token/ERC20/IERC20.sol';
2020-08-20 10:44:51 +00:00
import {FlashLoanReceiverBase} from '../../flashloan/base/FlashLoanReceiverBase.sol';
import {MintableERC20} from '../tokens/MintableERC20.sol';
import {SafeERC20} from '@openzeppelin/contracts/token/ERC20/SafeERC20.sol';
import {ILendingPoolAddressesProvider} from '../../interfaces/ILendingPoolAddressesProvider.sol';
contract MockFlashLoanReceiver is FlashLoanReceiverBase {
2020-07-13 08:54:08 +00:00
using SafeMath for uint256;
2020-08-12 17:36:58 +00:00
using SafeERC20 for IERC20;
2020-07-13 08:54:08 +00:00
ILendingPoolAddressesProvider internal _provider;
2020-07-13 08:54:08 +00:00
event ExecutedWithFail(address _reserve, uint256 _amount, uint256 _fee);
event ExecutedWithSuccess(address _reserve, uint256 _amount, uint256 _fee);
2020-08-25 12:50:07 +00:00
bool _failExecution;
uint256 _amountToApprove;
2020-07-13 08:54:08 +00:00
constructor(ILendingPoolAddressesProvider provider) public FlashLoanReceiverBase(provider) {}
2020-07-13 08:54:08 +00:00
2020-08-25 12:50:07 +00:00
function setFailExecutionTransfer(bool fail) public {
_failExecution = fail;
}
function setAmountToApprove(uint256 amountToApprove) public {
_amountToApprove = amountToApprove;
}
function amountToApprove() public view returns (uint256) {
return _amountToApprove;
2020-07-13 08:54:08 +00:00
}
function executeOperation(
address reserve,
uint256 amount,
uint256 fee,
bytes memory params
2020-07-13 08:54:08 +00:00
) public override {
//mint to this contract the specific amount
MintableERC20 token = MintableERC20(reserve);
2020-07-13 08:54:08 +00:00
//check the contract has the specified balance
2020-08-25 12:50:07 +00:00
require(amount <= IERC20(reserve).balanceOf(address(this)), 'Invalid balance for the contract');
uint256 amountToReturn = (_amountToApprove != 0) ? _amountToApprove : amount.add(fee);
2020-07-13 08:54:08 +00:00
2020-08-25 12:50:07 +00:00
if (_failExecution) {
emit ExecutedWithFail(reserve, amount, fee);
2020-07-13 08:54:08 +00:00
return;
}
2020-07-13 08:54:08 +00:00
//execution does not fail - mint tokens and return them to the _destination
//note: if the reserve is eth, the mock contract must receive at least _fee ETH before calling executeOperation
token.mint(fee);
2020-07-13 13:19:47 +00:00
2020-08-25 12:50:07 +00:00
IERC20(reserve).approve(_addressesProvider.getLendingPool(), amountToReturn);
2020-07-13 13:19:47 +00:00
emit ExecutedWithSuccess(reserve, amount, fee);
2020-07-13 08:54:08 +00:00
}
2020-06-02 14:16:22 +00:00
}