Instadapp/aave-protocol-v2
1
0
Fork 0
mirror of https://github.com/Instadapp/aave-protocol-v2.git synced 2024-07-29 21:47:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add batch swaps unit tests

Browse Source
This commit is contained in:
Gerardo Nardelli 2020-11-03 15:37:06 -03:00
parent 94a6a8688c
commit 847ad3b12a
4 changed files with 905 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
contracts/adapters/UniswapLiquiditySwapAdapter.sol
View File

@ -159,7 +159,7 @@ contract UniswapLiquiditySwapAdapter is BaseUniswapAdapter, IFlashLoanReceiver {
* bytes32[] s List of s param for the permit signature
* @return SwapParams struct containing decoded params
*/
function _decodeParams(bytes memory params) internal returns (SwapParams memory) {
function _decodeParams(bytes memory params) internal pure returns (SwapParams memory) {
(
address[] memory assetToSwapToList,
uint256[] memory minAmountsToReceive,

2
contracts/adapters/UniswapRepayAdapter.sol
View File

@ -147,7 +147,7 @@ contract UniswapRepayAdapter is BaseUniswapAdapter, IFlashLoanReceiver {
* bytes32[] s List of s param for the permit signature
* @return RepayParams struct containing decoded params
*/
function _decodeParams(bytes memory params) internal returns (RepayParams memory) {
function _decodeParams(bytes memory params) internal pure returns (RepayParams memory) {
(
address[] memory assetToSwapToList,
LeftoverAction leftOverAction,

30
contracts/mocks/swap/MockUniswapV2Router02.sol
View File

@ -6,17 +6,17 @@ import {IERC20} from '@openzeppelin/contracts/token/ERC20/IERC20.sol';
import {MintableERC20} from '../tokens/MintableERC20.sol';
contract MockUniswapV2Router02 is IUniswapV2Router02 {
uint256 internal _amountToReturn;
uint256 internal _amountToSwap;
mapping(address => uint256) internal _amountToReturn;
mapping(address => uint256) internal _amountToSwap;
mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsIn;
mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsOut;
function setAmountToReturn(uint256 amount) public {
_amountToReturn = amount;
function setAmountToReturn(address reserve, uint256 amount) public {
_amountToReturn[reserve] = amount;
}
function setAmountToSwap(uint256 amount) public {
_amountToSwap = amount;
function setAmountToSwap(address reserve, uint256 amount) public {
_amountToSwap[reserve] = amount;
}
function swapExactTokensForTokens(
@ -28,29 +28,29 @@ contract MockUniswapV2Router02 is IUniswapV2Router02 {
) external override returns (uint256[] memory amounts) {
IERC20(path[0]).transferFrom(msg.sender, address(this), amountIn);
MintableERC20(path[1]).mint(_amountToReturn);
IERC20(path[1]).transfer(to, _amountToReturn);
MintableERC20(path[1]).mint(_amountToReturn[path[0]]);
IERC20(path[1]).transfer(to, _amountToReturn[path[0]]);
amounts = new uint[](path.length);
amounts[0] = amountIn;
amounts[1] = _amountToReturn;
amounts[1] = _amountToReturn[path[0]];
}
function swapTokensForExactTokens(
uint /* amountOut */,
uint amountOut,
uint /* amountInMax */,
address[] calldata path,
address to,
uint /* deadline */
) external override returns (uint256[] memory amounts) {
IERC20(path[0]).transferFrom(msg.sender, address(this), _amountToSwap);
IERC20(path[0]).transferFrom(msg.sender, address(this), _amountToSwap[path[0]]);
MintableERC20(path[1]).mint(_amountToReturn);
IERC20(path[1]).transfer(to, _amountToReturn);
MintableERC20(path[1]).mint(amountOut);
IERC20(path[1]).transfer(to, amountOut);
amounts = new uint[](path.length);
amounts[0] = _amountToSwap;
amounts[1] = _amountToReturn;
amounts[0] = _amountToSwap[path[0]];
amounts[1] = amountOut;
}
function setAmountOut(uint amountIn, address reserveIn, address reserveOut, uint amountOut) public {

917
test/uniswapAdapters.spec.ts
View File

File diff suppressed because it is too large Load Diff