From 68d5980f74a7c70eab302077d022e685bdd156fa Mon Sep 17 00:00:00 2001
From: Lasse Herskind <16536249+LHerskind@users.noreply.github.com>
Date: Wed, 24 Nov 2021 10:18:39 +0000
Subject: [PATCH] test: Add test with >128 reserves

---
 test-suites/test-aave/many-asset.spec.ts | 164 +++++++++++++++++++++++
 1 file changed, 164 insertions(+)
 create mode 100644 test-suites/test-aave/many-asset.spec.ts

diff --git a/test-suites/test-aave/many-asset.spec.ts b/test-suites/test-aave/many-asset.spec.ts
new file mode 100644
index 00000000..3408e549
--- /dev/null
+++ b/test-suites/test-aave/many-asset.spec.ts
@@ -0,0 +1,164 @@
+import { TestEnv, makeSuite } from './helpers/make-suite';
+import {
+  APPROVAL_AMOUNT_LENDING_POOL,
+  MAX_UINT_AMOUNT,
+  RAY,
+  ZERO_ADDRESS,
+} from '../../helpers/constants';
+import {
+  convertToCurrencyDecimals,
+  getContractAddressWithJsonFallback,
+} from '../../helpers/contracts-helpers';
+import { eContractid, ProtocolErrors, RateMode } from '../../helpers/types';
+import { strategyWETH } from '../../markets/aave/reservesConfigs';
+import {
+  deployATokenImplementations,
+  deployMintableERC20,
+  deployValidationLogic,
+} from '../../helpers/contracts-deployments';
+import { getATokenExtraParams } from '../../helpers/init-helpers';
+import { ConfigNames } from '../../helpers/configuration';
+import { zeroAddress } from 'hardhat/node_modules/ethereumjs-util';
+import AaveConfig from '../../markets/aave';
+import { getATokensAndRatesHelper } from '../../helpers/contracts-getters';
+import { config } from 'process';
+import { parseEther } from '@ethersproject/units';
+import exp from 'constants';
+
+const { expect } = require('chai');
+
+makeSuite('Adding > 128 asset to many-asset configured pool', (testEnv: TestEnv) => {
+  const tokens = {};
+
+  before('setup', async () => {
+    const { pool, dai, configurator, deployer, oracle } = testEnv;
+    const atokenAndRatesDeployer = await getATokensAndRatesHelper();
+
+    const reservesCount = await pool.getReservesList();
+
+    const daiData = await pool.getReserveData(dai.address);
+    const poolName = ConfigNames.Aave;
+
+    const {
+      ATokenNamePrefix,
+      StableDebtTokenNamePrefix,
+      VariableDebtTokenNamePrefix,
+      SymbolPrefix,
+    } = AaveConfig;
+
+    // Create tokens
+    let tokenInitParams = [];
+    let reserveInitParams = [];
+
+    const aTokenImplAddress = await getContractAddressWithJsonFallback(
+      eContractid.AToken,
+      poolName
+    );
+    const stableDebtImplAddress = await getContractAddressWithJsonFallback(
+      eContractid.StableDebtToken,
+      poolName
+    );
+    const variableDebtImplAddress = await getContractAddressWithJsonFallback(
+      eContractid.VariableDebtToken,
+      poolName
+    );
+
+    const daiPrice = await oracle.getAssetPrice(dai.address);
+
+    // All assets will be initialized with the same price as Dai.
+    for (let i = reservesCount.length; i < 140; i++) {
+      const tokenName = `RealT-${i}`;
+      const token = await deployMintableERC20([tokenName, tokenName, '18']);
+      tokens[tokenName] = token;
+
+      tokenInitParams.push({
+        aTokenImpl: aTokenImplAddress,
+        stableDebtTokenImpl: stableDebtImplAddress,
+        variableDebtTokenImpl: variableDebtImplAddress,
+        underlyingAssetDecimals: 18,
+        interestRateStrategyAddress: daiData.interestRateStrategyAddress,
+        underlyingAsset: token.address,
+        treasury: ZERO_ADDRESS,
+        incentivesController: ZERO_ADDRESS,
+        underlyingAssetName: tokenName,
+        aTokenName: `${ATokenNamePrefix} ${tokenName}`,
+        aTokenSymbol: `a${SymbolPrefix}${tokenName}`,
+        variableDebtTokenName: `${VariableDebtTokenNamePrefix} ${SymbolPrefix}${tokenName}`,
+        variableDebtTokenSymbol: `variableDebt${SymbolPrefix}${tokenName}`,
+        stableDebtTokenName: `${StableDebtTokenNamePrefix} ${tokenName}`,
+        stableDebtTokenSymbol: `stableDebt${SymbolPrefix}${tokenName}`,
+        params: '0x10',
+      });
+
+      reserveInitParams.push({
+        asset: token.address,
+        baseLTV: 5000,
+        liquidationThreshold: 8000,
+        liquidationBonus: 10500,
+        reserveFactor: 1000,
+        stableBorrowingEnabled: false,
+        borrowingEnabled: true,
+      });
+
+      if (tokenInitParams.length == 2 || i == 139) {
+        await configurator.batchInitReserve(tokenInitParams);
+
+        for (let j = 0; j < reserveInitParams.length; j++) {
+          let params = reserveInitParams[j];
+          await configurator
+            .connect(deployer.signer)
+            .configureReserveAsCollateral(
+              params.asset,
+              params.baseLTV,
+              params.liquidationThreshold,
+              params.liquidationBonus
+            );
+          await configurator.enableBorrowingOnReserve(params.asset, false);
+          await oracle.setAssetPrice(params.asset, daiPrice);
+        }
+        tokenInitParams = [];
+        reserveInitParams = [];
+      }
+    }
+  });
+
+  it('Check that pool contains >128 assets', async () => {
+    const { pool } = testEnv;
+    const reservesCount = await pool.getReservesList();
+    expect(reservesCount.length).to.be.gt(128);
+  });
+
+  it('Add collateral and borrow asset 130', async () => {
+    const {
+      users: [, user, funder],
+      weth,
+      pool,
+    } = testEnv;
+
+    const realT130 = tokens[`RealT-${130}`];
+
+    const assetdata = await pool.getReserveData(realT130.address);
+    expect(assetdata.id).to.be.gt(128);
+
+    await realT130.connect(funder.signer).mint(parseEther('100'));
+    await realT130.connect(funder.signer).approve(pool.address, MAX_UINT_AMOUNT);
+    await pool
+      .connect(funder.signer)
+      .deposit(realT130.address, parseEther('100'), funder.address, 0);
+
+    await weth.connect(user.signer).mint(parseEther('1'));
+    await weth.connect(user.signer).approve(pool.address, MAX_UINT_AMOUNT);
+    await pool.connect(user.signer).deposit(weth.address, parseEther('1'), user.address, 0);
+
+    const userDataBefore = await pool.getUserAccountData(user.address);
+    await pool
+      .connect(user.signer)
+      .borrow(realT130.address, parseEther('1'), RateMode.Variable, 0, user.address);
+    const userDataAfter = await pool.getUserAccountData(user.address);
+
+    expect(userDataAfter.totalCollateralETH).to.be.eq(userDataBefore.totalCollateralETH);
+    expect(userDataBefore.totalDebtETH).to.be.eq(0);
+    expect(userDataAfter.totalDebtETH).to.be.gt(userDataBefore.totalDebtETH);
+    expect(userDataAfter.healthFactor).to.be.lt(userDataBefore.healthFactor);
+  });
+});