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 logic to set usageAsCollateralEnabled for the user if his balance was 0 before swap

Browse Source
This commit is contained in:
andyk 2020-09-16 13:30:57 +03:00
parent 1b61edc6ff
commit 83e0318bc6
2 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
contracts/lendingpool/LendingPoolLiquidationManager.sol
View File

@ -514,6 +514,11 @@ contract LendingPoolLiquidationManager is VersionedInitializable {
address(vars.toReserveAToken), address(vars.toReserveAToken),
vars.amountToReceive vars.amountToReceive
); );
if (vars.toReserveAToken.balanceOf(msg.sender) == 0) {
usersConfig[msg.sender].setUsingAsCollateral(toReserve.id, true);
}
vars.toReserveAToken.mint(msg.sender, vars.amountToReceive, toReserve.liquidityIndex); vars.toReserveAToken.mint(msg.sender, vars.amountToReceive, toReserve.liquidityIndex);
toReserve.updateInterestRates( toReserve.updateInterestRates(
toAsset, toAsset,

17
test/collateral-swap.spec.ts
View File

@ -83,6 +83,7 @@ makeSuite('LendingPool SwapDeposit function', (testEnv: TestEnv) => {
.deposit(weth.address, amountToDeposit, await signer.getAddress(), '0'); .deposit(weth.address, amountToDeposit, await signer.getAddress(), '0');
} }
}); });
it('User tries to swap more then he can, revert expected', async () => { it('User tries to swap more then he can, revert expected', async () => {
const {pool, weth, dai} = testEnv; const {pool, weth, dai} = testEnv;
await expect( await expect(
@ -96,19 +97,6 @@ makeSuite('LendingPool SwapDeposit function', (testEnv: TestEnv) => {
).to.be.revertedWith('55'); ).to.be.revertedWith('55');
}); });
it('User tries to swap asset on equal asset, revert expected', async () => {
const {pool, weth} = testEnv;
await expect(
pool.swapLiquidity(
_mockSwapAdapter.address,
weth.address,
weth.address,
ethers.utils.parseEther('0.1'),
'0x10'
)
).to.be.revertedWith('56');
});
it('User tries to swap more then available on the reserve', async () => { it('User tries to swap more then available on the reserve', async () => {
const {pool, weth, dai, users, aEth, deployer} = testEnv; const {pool, weth, dai, users, aEth, deployer} = testEnv;
@ -185,6 +173,9 @@ makeSuite('LendingPool SwapDeposit function', (testEnv: TestEnv) => {
reserveBalanceDAIBefore.add(amountToReturn).toString(), reserveBalanceDAIBefore.add(amountToReturn).toString(),
'was received incorrect amount if reserve funds' 'was received incorrect amount if reserve funds'
); );
expect(
(await pool.getUserReserveData(dai.address, userAddress)).usageAsCollateralEnabled
).to.be.equal(true, 'usage as collateral was not enabled on destination reserve for the user');
}); });
it('User tries to drop HF below one', async () => { it('User tries to drop HF below one', async () => {