Instadapp/aave-protocol-v2
1
0
Fork 0
mirror of https://github.com/Instadapp/aave-protocol-v2.git synced 2024-07-29 21:47:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: fixed the accrued to treasury test

Browse Source
This commit is contained in:
The3D 2021-05-08 18:53:56 +02:00
parent ce052c669f
commit 42a07e75ab
1 changed files with 38 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
test-suites/test-aave/mint-to-treasury.spec.ts
View File

@ -1,79 +1,69 @@
import { makeSuite, TestEnv } from './helpers/make-suite'; import { makeSuite, TestEnv } from './helpers/make-suite';
import { ProtocolErrors, RateMode } from '../../helpers/types'; import { RateMode } from '../../helpers/types';
import { APPROVAL_AMOUNT_LENDING_POOL, oneEther, ONE_YEAR, RAY } from '../../helpers/constants'; import { APPROVAL_AMOUNT_LENDING_POOL, ONE_YEAR } from '../../helpers/constants';
import { convertToCurrencyDecimals } from '../../helpers/contracts-helpers'; import { convertToCurrencyDecimals } from '../../helpers/contracts-helpers';
import { parseEther, parseUnits } from 'ethers/lib/utils';
import { BigNumber } from 'bignumber.js'; import { BigNumber } from 'bignumber.js';
import { MockFlashLoanReceiver } from '../../types/MockFlashLoanReceiver'; import { advanceTimeAndBlock, waitForTx } from '../../helpers/misc-utils';
import { getMockFlashLoanReceiver } from '../../helpers/contracts-getters';
import { advanceTimeAndBlock, timeLatest, waitForTx } from '../../helpers/misc-utils';
import './helpers/utils/math'; import './helpers/utils/math';
import { calcCompoundedInterest } from './helpers/utils/calculations';
import {getTxCostAndTimestamp } from './helpers/actions';
const { expect } = require('chai'); const { expect } = require('chai');
makeSuite('Mint to treasury', (testEnv: TestEnv) => { makeSuite('Mint to treasury', (testEnv: TestEnv) => {
let _mockFlashLoanReceiver = {} as MockFlashLoanReceiver;
const {
LP_IS_PAUSED,
INVALID_FROM_BALANCE_AFTER_TRANSFER,
INVALID_TO_BALANCE_AFTER_TRANSFER,
} = ProtocolErrors;
before(async () => {
_mockFlashLoanReceiver = await getMockFlashLoanReceiver();
});
it('User 0 deposits 1000 DAI. Borrower borrows 100 DAI. Clock moved forward one year. Calculates and verifies the amount earned by the treasury', async () => { it('User 0 deposits 1000 DAI. Borrower borrows 100 DAI. Clock moved forward one year. Calculates and verifies the amount earned by the treasury', async () => {
const { users, pool, dai, helpersContract } = testEnv; const { users, pool, dai, helpersContract } = testEnv;
const amountDAItoDeposit = await convertToCurrencyDecimals(dai.address, '1000'); const amountDAItoDeposit = await convertToCurrencyDecimals(dai.address, '1000');
const amountDAItoBorrow = await convertToCurrencyDecimals(dai.address, '100'); const amountDAItoBorrow = await convertToCurrencyDecimals(dai.address, '100');
await dai.connect(users[0].signer).mint(amountDAItoDeposit); await waitForTx(await dai.connect(users[0].signer).mint(amountDAItoDeposit));
// user 0 deposits 1000 DAI // user 0 deposits 1000 DAI
await dai.connect(users[0].signer).approve(pool.address, APPROVAL_AMOUNT_LENDING_POOL); await waitForTx(
await dai.connect(users[0].signer).approve(pool.address, APPROVAL_AMOUNT_LENDING_POOL)
);
await waitForTx(
await pool await pool
.connect(users[0].signer) .connect(users[0].signer)
.deposit(dai.address, amountDAItoDeposit, users[0].address, '0'); .deposit(dai.address, amountDAItoDeposit, users[0].address, '0')
);
const borrowTx = await waitForTx(await pool await waitForTx(
await pool
.connect(users[0].signer) .connect(users[0].signer)
.borrow(dai.address, amountDAItoBorrow, RateMode.Variable, '0', users[0].address)); .borrow(dai.address, amountDAItoBorrow, RateMode.Variable, '0', users[0].address)
);
const {txTimestamp : lastUpdateTimestamp} = await getTxCostAndTimestamp(borrowTx);
const {
currentLiquidityRate: liquidityRate,
currentVariableBorrowRate: variableBorrowRate,
} = await pool.getReserveData(dai.address);
const { reserveFactor } = await helpersContract.getReserveConfigurationData(dai.address); const { reserveFactor } = await helpersContract.getReserveConfigurationData(dai.address);
await advanceTimeAndBlock(parseInt(ONE_YEAR)); await advanceTimeAndBlock(parseInt(ONE_YEAR));
await dai.connect(users[0].signer).mint(amountDAItoDeposit); await waitForTx(await dai.connect(users[0].signer).mint(amountDAItoDeposit));
const depositTx = await waitForTx(await pool await waitForTx(
await pool
.connect(users[0].signer) .connect(users[0].signer)
.deposit(dai.address, amountDAItoDeposit, users[0].address, '0')); .deposit(dai.address, amountDAItoDeposit, users[0].address, '0')
);
const { liquidityIndex, variableBorrowIndex } = await pool.getReserveData(dai.address);
const {txTimestamp : currTimestamp} = await getTxCostAndTimestamp(depositTx); const amountBorrowedBN = new BigNumber(amountDAItoBorrow.toString());
const liquidityIndexBN = new BigNumber(liquidityIndex.toString());
const variableBorrowIndexBN = new BigNumber(variableBorrowIndex.toString());
const interestNormalized = calcCompoundedInterest(new BigNumber(variableBorrowRate.toString()), currTimestamp, lastUpdateTimestamp).minus(RAY); const expectedAccruedToTreasury = amountBorrowedBN
.rayMul(variableBorrowIndexBN)
.minus(amountBorrowedBN)
const expectedReserveFactor = new BigNumber(amountDAItoBorrow.toString())
.rayMul(interestNormalized)
.times(reserveFactor.toString()) .times(reserveFactor.toString())
.div(10000); .div(10000)
.rayDiv(liquidityIndexBN)
.toFixed(0);
const reserveData = await pool.getReserveData(dai.address); const { accruedToTreasury } = await pool.getReserveData(dai.address);
console.log(reserveData.accruedToTreasury.toString(), expectedReserveFactor.toString()); expect(accruedToTreasury.toString()).to.be.bignumber.almostEqual(
expectedAccruedToTreasury,
'Invalid amount accrued to treasury'
);
}); });
}); });