aave-protocol-v2/test/helpers/make-suite.ts

149 lines
4.9 KiB
TypeScript
Raw Normal View History

import { evmRevert, evmSnapshot, DRE } from '../../helpers/misc-utils';
import { Signer } from 'ethers';
import {
getLendingPool,
getLendingPoolAddressesProvider,
getAaveProtocolDataProvider,
getAToken,
getMintableERC20,
2020-07-13 08:54:08 +00:00
getLendingPoolConfiguratorProxy,
getPriceOracle,
2020-10-08 13:41:48 +00:00
getLendingPoolAddressesProviderRegistry,
2020-10-29 11:44:21 +00:00
getWETHMocked,
getWETHGateway,
2020-10-15 17:19:02 +00:00
} from '../../helpers/contracts-getters';
import { tEthereumAddress } from '../../helpers/types';
import { LendingPool } from '../../types/LendingPool';
import { AaveProtocolDataProvider } from '../../types/AaveProtocolDataProvider';
import { MintableERC20 } from '../../types/MintableERC20';
import { AToken } from '../../types/AToken';
import { LendingPoolConfigurator } from '../../types/LendingPoolConfigurator';
2020-07-13 08:54:08 +00:00
import chai from 'chai';
// @ts-ignore
2020-07-13 08:54:08 +00:00
import bignumberChai from 'chai-bignumber';
import { almostEqual } from './almost-equal';
import { PriceOracle } from '../../types/PriceOracle';
import { LendingPoolAddressesProvider } from '../../types/LendingPoolAddressesProvider';
import { LendingPoolAddressesProviderRegistry } from '../../types/LendingPoolAddressesProviderRegistry';
import { getEthersSigners } from '../../helpers/contracts-helpers';
import { WETH9Mocked } from '../../types/WETH9Mocked';
import { WETHGateway } from '../../types/WETHGateway';
import { solidity } from 'ethereum-waffle';
2020-10-29 14:50:32 +00:00
chai.use(bignumberChai());
chai.use(almostEqual());
2020-10-29 14:50:32 +00:00
chai.use(solidity);
2020-06-10 12:47:39 +00:00
export interface SignerWithAddress {
signer: Signer;
address: tEthereumAddress;
}
export interface TestEnv {
deployer: SignerWithAddress;
users: SignerWithAddress[];
pool: LendingPool;
configurator: LendingPoolConfigurator;
oracle: PriceOracle;
helpersContract: AaveProtocolDataProvider;
weth: WETH9Mocked;
aWETH: AToken;
dai: MintableERC20;
aDai: AToken;
usdc: MintableERC20;
aave: MintableERC20;
2020-07-13 08:54:08 +00:00
addressesProvider: LendingPoolAddressesProvider;
2020-09-23 09:21:49 +00:00
registry: LendingPoolAddressesProviderRegistry;
wethGateway: WETHGateway;
}
2020-07-13 08:54:08 +00:00
let buidlerevmSnapshotId: string = '0x1';
const setBuidlerevmSnapshotId = (id: string) => {
if (DRE.network.name === 'hardhat') {
buidlerevmSnapshotId = id;
}
};
const testEnv: TestEnv = {
deployer: {} as SignerWithAddress,
users: [] as SignerWithAddress[],
pool: {} as LendingPool,
configurator: {} as LendingPoolConfigurator,
helpersContract: {} as AaveProtocolDataProvider,
oracle: {} as PriceOracle,
weth: {} as WETH9Mocked,
aWETH: {} as AToken,
dai: {} as MintableERC20,
aDai: {} as AToken,
usdc: {} as MintableERC20,
aave: {} as MintableERC20,
2020-07-13 08:54:08 +00:00
addressesProvider: {} as LendingPoolAddressesProvider,
2020-10-08 13:41:48 +00:00
registry: {} as LendingPoolAddressesProviderRegistry,
wethGateway: {} as WETHGateway,
} as TestEnv;
export async function initializeMakeSuite() {
const [_deployer, ...restSigners] = await getEthersSigners();
const deployer: SignerWithAddress = {
address: await _deployer.getAddress(),
signer: _deployer,
};
for (const signer of restSigners) {
testEnv.users.push({
signer,
address: await signer.getAddress(),
});
}
testEnv.deployer = deployer;
testEnv.pool = await getLendingPool();
2020-06-26 12:34:44 +00:00
testEnv.configurator = await getLendingPoolConfiguratorProxy();
2020-07-13 08:54:08 +00:00
testEnv.oracle = await getPriceOracle();
testEnv.addressesProvider = await getLendingPoolAddressesProvider();
2020-09-23 09:21:49 +00:00
testEnv.registry = await getLendingPoolAddressesProviderRegistry();
2020-07-13 08:54:08 +00:00
testEnv.helpersContract = await getAaveProtocolDataProvider();
2020-07-13 08:54:08 +00:00
const allTokens = await testEnv.helpersContract.getAllATokens();
const aDaiAddress = allTokens.find((aToken) => aToken.symbol === 'aDAI')?.tokenAddress;
const aWEthAddress = allTokens.find((aToken) => aToken.symbol === 'aWETH')?.tokenAddress;
const reservesTokens = await testEnv.helpersContract.getAllReservesTokens();
2020-06-26 12:34:44 +00:00
2020-07-13 08:54:08 +00:00
const daiAddress = reservesTokens.find((token) => token.symbol === 'DAI')?.tokenAddress;
const usdcAddress = reservesTokens.find((token) => token.symbol === 'USDC')?.tokenAddress;
const aaveAddress = reservesTokens.find((token) => token.symbol === 'AAVE')?.tokenAddress;
const wethAddress = reservesTokens.find((token) => token.symbol === 'WETH')?.tokenAddress;
2020-07-13 08:54:08 +00:00
if (!aDaiAddress || !aWEthAddress) {
process.exit(1);
}
if (!daiAddress || !usdcAddress || !aaveAddress || !wethAddress) {
process.exit(1);
}
testEnv.aDai = await getAToken(aDaiAddress);
testEnv.aWETH = await getAToken(aWEthAddress);
testEnv.dai = await getMintableERC20(daiAddress);
testEnv.usdc = await getMintableERC20(usdcAddress);
testEnv.aave = await getMintableERC20(aaveAddress);
2020-10-29 11:44:21 +00:00
testEnv.weth = await getWETHMocked(wethAddress);
testEnv.wethGateway = await getWETHGateway();
}
export function makeSuite(name: string, tests: (testEnv: TestEnv) => void) {
describe(name, () => {
2020-06-10 12:47:39 +00:00
before(async () => {
setBuidlerevmSnapshotId(await evmSnapshot());
2020-06-10 12:47:39 +00:00
});
tests(testEnv);
2020-06-10 12:47:39 +00:00
after(async () => {
await evmRevert(buidlerevmSnapshotId);
2020-06-10 12:47:39 +00:00
});
});
}