Fixes missing param at deployDefaultReserveInterestRateStrategy. Update WalletBalanceProvider to add provider argument. Minor fixes migration script.

.gitlab-ci.yml

@@ -1,5 +1,6 @@
 stages:
   - test
+  - deploy-mainnet-fork
 
 test:
   stage: test
@@ -12,3 +13,14 @@ test:
   after_script:
     - docker-compose -f docker-compose.test.yml run contracts-env npm run ci:clean
     - docker-compose -f docker-compose.test.yml down
+deploy-mainnet-fork:
+  stage: deploy-mainnet-fork
+  tags:
+    - aave-build-runner
+  before_script:
+    - docker-compose build
+  script:
+    - docker-compose run contracts-env npm aave:main:fork
+  after_script:
+    - docker-compose -f docker-compose.test.yml run contracts-env npm run ci:clean
+    - docker-compose -f docker-compose.test.yml down

contracts/misc/WalletBalanceProvider.sol

@@ -24,13 +24,8 @@ contract WalletBalanceProvider {
   using SafeERC20 for IERC20;
   using ReserveConfiguration for ReserveConfiguration.Map;
 
-  ILendingPoolAddressesProvider internal immutable _provider;
   address constant MOCK_ETH_ADDRESS = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
 
-  constructor(ILendingPoolAddressesProvider provider) public {
-    _provider = provider;
-  }
-
   /**
     @dev Fallback function, don't accept any ETH
     **/
@@ -80,12 +75,12 @@ contract WalletBalanceProvider {
   /**
     @dev provides balances of user wallet for all reserves available on the pool
     */
-  function getUserWalletBalances(address user)
+  function getUserWalletBalances(address provider, address user)
     external
     view
     returns (address[] memory, uint256[] memory)
   {
-    ILendingPool pool = ILendingPool(_provider.getLendingPool());
+    ILendingPool pool = ILendingPool(ILendingPoolAddressesProvider(provider).getLendingPool());
 
     address[] memory reserves = pool.getReservesList();
     address[] memory reservesWithEth = new address[](reserves.length + 1);

helpers/contracts-deployments.ts

@@ -244,14 +244,11 @@ export const deployMockFlashLoanReceiver = async (
     verify
   );
 
-export const deployWalletBalancerProvider = async (
-  addressesProvider: tEthereumAddress,
-  verify?: boolean
-) =>
+export const deployWalletBalancerProvider = async (verify?: boolean) =>
   withSaveAndVerify(
-    await new WalletBalanceProviderFactory(await getFirstSigner()).deploy(addressesProvider),
+    await new WalletBalanceProviderFactory(await getFirstSigner()).deploy(),
     eContractid.WalletBalanceProvider,
-    [addressesProvider],
+    [],
     verify
   );
 
@@ -288,7 +285,7 @@ export const deployMintableDelegationERC20 = async (
     verify
   );
 export const deployDefaultReserveInterestRateStrategy = async (
-  args: [tEthereumAddress, string, string, string, string, string],
+  args: [tEthereumAddress, string, string, string, string, string, string],
   verify: boolean
 ) =>
   withSaveAndVerify(

tasks/dev/5_initialize.ts

@@ -22,7 +22,10 @@ import {
 } from '../../helpers/init-helpers';
 import { getAllTokenAddresses } from '../../helpers/mock-helpers';
 import { ZERO_ADDRESS } from '../../helpers/constants';
-import {getAllMockedTokens, getLendingPoolAddressesProvider} from '../../helpers/contracts-getters';
+import {
+  getAllMockedTokens,
+  getLendingPoolAddressesProvider,
+} from '../../helpers/contracts-getters';
 import { insertContractAddressInDb } from '../../helpers/contracts-helpers';
 
 task('dev:initialize-lending-pool', 'Initialize lending pool configuration.')
@@ -75,7 +78,7 @@ task('dev:initialize-lending-pool', 'Initialize lending pool configuration.')
       mockFlashLoanReceiver.address
     );
 
-    await deployWalletBalancerProvider(addressesProvider.address, verify);
+    await deployWalletBalancerProvider(verify);
 
     await insertContractAddressInDb(eContractid.AaveProtocolDataProvider, testHelpers.address);
 

tasks/dev/6_wallet_balance_provider.ts

@@ -8,6 +8,5 @@ task('dev:wallet-balance-provider', 'Initialize lending pool configuration.')
   .setAction(async ({ verify }, localBRE) => {
     await localBRE.run('set-DRE');
 
-    const addressesProvider = await getLendingPoolAddressesProvider();
-    await deployWalletBalancerProvider(addressesProvider.address, verify);
+    await deployWalletBalancerProvider(verify);
   });

tasks/full/5_initialize.ts

@@ -15,7 +15,10 @@ import {
   enableReservesAsCollateralByHelper,
 } from '../../helpers/init-helpers';
 import { exit } from 'process';
-import {getLendingPoolAddressesProvider} from '../../helpers/contracts-getters';
+import {
+  getAaveProtocolDataProvider,
+  getLendingPoolAddressesProvider,
+} from '../../helpers/contracts-getters';
 import { ZERO_ADDRESS } from '../../helpers/constants';
 
 task('full:initialize-lending-pool', 'Initialize lending pool configuration.')
@@ -32,7 +35,7 @@ task('full:initialize-lending-pool', 'Initialize lending pool configuration.')
 
       const addressesProvider = await getLendingPoolAddressesProvider();
 
-      const testHelpers = await deployAaveProtocolDataProvider(addressesProvider.address, verify);
+      const testHelpers = await getAaveProtocolDataProvider();
 
       const admin = await addressesProvider.getPoolAdmin();
       if (!reserveAssets) {
@@ -48,7 +51,7 @@ task('full:initialize-lending-pool', 'Initialize lending pool configuration.')
         await addressesProvider.setLendingPoolCollateralManager(collateralManager.address)
       );
 
-      await deployWalletBalancerProvider(addressesProvider.address, verify);
+      await deployWalletBalancerProvider(verify);
 
       const wethAddress = await getWethAddress(poolConfig);
       const lendingPoolAddress = await addressesProvider.getLendingPool();

test/__setup.spec.ts

@@ -226,7 +226,7 @@ const buildTestEnv = async (deployer: Signer, secondaryWallet: Signer) => {
   const mockFlashLoanReceiver = await deployMockFlashLoanReceiver(addressesProvider.address);
   await insertContractAddressInDb(eContractid.MockFlashLoanReceiver, mockFlashLoanReceiver.address);
 
-  await deployWalletBalancerProvider(addressesProvider.address);
+  await deployWalletBalancerProvider();
 
   await deployWETHGateway([mockTokens.WETH.address, lendingPoolAddress]);