aave-protocol-v2/contracts/misc/AaveProtocolTestHelpers.sol

pragma solidity ^0.6.8;
pragma experimental ABIEncoderV2;

import {ILendingPoolAddressesProvider} from "../interfaces/ILendingPoolAddressesProvider.sol";
import {IERC20Detailed} from "../interfaces/IERC20Detailed.sol";
import {LendingPool} from "../lendingpool/LendingPool.sol";
import {AToken} from "../tokenization/AToken.sol";
import "@nomiclabs/buidler/console.sol";

contract AaveProtocolTestHelpers {
    struct TokenData {
        string symbol;
        address tokenAddress;
    }

    ILendingPoolAddressesProvider public immutable ADDRESSES_PROVIDER;

    constructor(ILendingPoolAddressesProvider addressesProvider) public {
        ADDRESSES_PROVIDER = addressesProvider;
    }

    function getAllReservesTokens() external view returns(TokenData[] memory) {
        LendingPool pool = LendingPool(payable(ADDRESSES_PROVIDER.getLendingPool()));
        console.log("Getting reserves...");

        address[] memory reserves = pool.getReserves();
        console.log("Reserves retrivied");

        TokenData[] memory reservesTokens = new TokenData[](reserves.length);
        for (uint256 i = 0; i < reserves.length; i++) {
            reservesTokens[i] = TokenData({
                symbol: (reserves[i] == 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE) ? "ETH" : IERC20Detailed(reserves[i]).symbol(),
                tokenAddress: reserves[i]
            });
        }
        return reservesTokens;
    }

    function getAllATokens() external view returns(TokenData[] memory) {
        LendingPool pool = LendingPool(payable(ADDRESSES_PROVIDER.getLendingPool()));
    
        console.log("Getting all the tokens...");
        address[] memory reserves = pool.getReserves();
    
        console.log("Reserves retrivied");
        TokenData[] memory aTokens = new TokenData[](reserves.length);
        for (uint256 i = 0; i < reserves.length; i++) {
            console.log("Getting configuration for reserve %s", i);
            (,,,,address aTokenAddress,,,,) = pool.getReserveConfigurationData(reserves[i]);
            aTokens[i] = TokenData({
                symbol: AToken(aTokenAddress).symbol(),
                tokenAddress: aTokenAddress
            });
        }
        return aTokens;
    }
}