Updated flashloan updateInterestRates

aave · Nov 23, 2020 · 584a567 · 584a567
1 parent 13ce62e
commit 584a567
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/contracts/protocol/lendingpool/LendingPool.sol b/contracts/protocol/lendingpool/LendingPool.sol
@@ -508,7 +508,7 @@ contract LendingPool is VersionedInitializable, ILendingPool, LendingPoolStorage
         _reserves[vars.currentAsset].updateInterestRates(
           vars.currentAsset,
           vars.currentATokenAddress,
-          vars.currentPremium,
+          vars.currentAmountPlusPremium,
           0
         );
 

diff --git a/test/flashloan.spec.ts b/test/flashloan.spec.ts
@@ -252,6 +252,8 @@ makeSuite('LendingPool FlashLoan function', (testEnv: TestEnv) => {
 
     await _mockFlashLoanReceiver.setFailExecutionTransfer(false);
 
+    const reserveDataBefore = await helpersContract.getReserveData(usdc.address);
+
     const flashloanAmount = await convertToCurrencyDecimals(usdc.address, '500');
 
     await pool.flashLoan(
@@ -264,6 +266,9 @@ makeSuite('LendingPool FlashLoan function', (testEnv: TestEnv) => {
       '0'
     );
 
+    const reserveDataAfter = helpersContract.getReserveData(usdc.address);
+
+
     const reserveData = await helpersContract.getReserveData(usdc.address);
     const userData = await helpersContract.getUserReserveData(usdc.address, depositor.address);
 

diff --git a/test/helpers/utils/calculations.ts b/test/helpers/utils/calculations.ts
@@ -1194,7 +1194,7 @@ const calcCompoundedInterest = (
     .plus(thirdTerm);
 };
 
-const calcExpectedInterestRates = (
+export const calcExpectedInterestRates = (
   reserveSymbol: string,
   marketStableRate: BigNumber,
   utilizationRate: BigNumber,
@@ -1253,7 +1253,7 @@ const calcExpectedInterestRates = (
   return [liquidityRate, stableBorrowRate, variableBorrowRate];
 };
 
-const calcExpectedOverallBorrowRate = (
+export const calcExpectedOverallBorrowRate = (
   totalStableDebt: BigNumber,
   totalVariableDebt: BigNumber,
   currentVariableBorrowRate: BigNumber,
@@ -1274,7 +1274,7 @@ const calcExpectedOverallBorrowRate = (
   return overallBorrowRate;
 };
 
-const calcExpectedUtilizationRate = (
+export const calcExpectedUtilizationRate = (
   totalStableDebt: BigNumber,
   totalVariableDebt: BigNumber,
   totalLiquidity: BigNumber