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 * bytes32[] s List of s param for the permit signature
* @return SwapParams struct containing decoded params * @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, address[] memory assetToSwapToList,
uint256[] memory minAmountsToReceive, 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 * bytes32[] s List of s param for the permit signature
* @return RepayParams struct containing decoded params * @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, address[] memory assetToSwapToList,
LeftoverAction leftOverAction, 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'; import {MintableERC20} from '../tokens/MintableERC20.sol';
contract MockUniswapV2Router02 is IUniswapV2Router02 { contract MockUniswapV2Router02 is IUniswapV2Router02 {
uint256 internal _amountToReturn; mapping(address => uint256) internal _amountToReturn;
uint256 internal _amountToSwap; mapping(address => uint256) internal _amountToSwap;
mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsIn; mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsIn;
mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsOut; mapping(address => mapping(address => mapping(uint256 => uint256))) internal _amountsOut;
function setAmountToReturn(uint256 amount) public { function setAmountToReturn(address reserve, uint256 amount) public {
_amountToReturn = amount; _amountToReturn[reserve] = amount;
} }
function setAmountToSwap(uint256 amount) public { function setAmountToSwap(address reserve, uint256 amount) public {
_amountToSwap = amount; _amountToSwap[reserve] = amount;
} }
function swapExactTokensForTokens( function swapExactTokensForTokens(
@ -28,29 +28,29 @@ contract MockUniswapV2Router02 is IUniswapV2Router02 {
) external override returns (uint256[] memory amounts) { ) external override returns (uint256[] memory amounts) {
IERC20(path[0]).transferFrom(msg.sender, address(this), amountIn); IERC20(path[0]).transferFrom(msg.sender, address(this), amountIn);
MintableERC20(path[1]).mint(_amountToReturn); MintableERC20(path[1]).mint(_amountToReturn[path[0]]);
IERC20(path[1]).transfer(to, _amountToReturn); IERC20(path[1]).transfer(to, _amountToReturn[path[0]]);
amounts = new uint[](path.length); amounts = new uint[](path.length);
amounts[0] = amountIn; amounts[0] = amountIn;
amounts[1] = _amountToReturn; amounts[1] = _amountToReturn[path[0]];
} }
function swapTokensForExactTokens( function swapTokensForExactTokens(
uint /* amountOut */, uint amountOut,
uint /* amountInMax */, uint /* amountInMax */,
address[] calldata path, address[] calldata path,
address to, address to,
uint /* deadline */ uint /* deadline */
) external override returns (uint256[] memory amounts) { ) 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); MintableERC20(path[1]).mint(amountOut);
IERC20(path[1]).transfer(to, _amountToReturn); IERC20(path[1]).transfer(to, amountOut);
amounts = new uint[](path.length); amounts = new uint[](path.length);
amounts[0] = _amountToSwap; amounts[0] = _amountToSwap[path[0]];
amounts[1] = _amountToReturn; amounts[1] = amountOut;
} }
function setAmountOut(uint amountIn, address reserveIn, address reserveOut, uint amountOut) public { 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