aave-protocol-v2/test-suites/test-aave/borrow-repay-same-tx.ts

import { TestEnv, makeSuite } from './helpers/make-suite';
import {
  APPROVAL_AMOUNT_LENDING_POOL,
  MAX_UINT_AMOUNT,
  RAY,
  MAX_BORROW_CAP,
  MAX_SUPPLY_CAP,
} from '../../helpers/constants';
import { ProtocolErrors } from '../../helpers/types';
import {
  BorrowRepayTestMock,
  BorrowRepayTestMockFactory,
  MintableERC20,
  WETH9,
  WETH9Mocked,
} from '../../types';
import { parseEther } from '@ethersproject/units';
import { BigNumber } from '@ethersproject/bignumber';
import { waitForTx } from '../../helpers/misc-utils';
import { getFirstSigner } from '../../helpers/contracts-getters';

const { expect } = require('chai');

makeSuite('Borrow/repay in the same tx', (testEnv: TestEnv) => {
  const { VL_SAME_BLOCK_BORROW_REPAY } = ProtocolErrors;
  const unitParse = async (token: WETH9Mocked | MintableERC20, nb: string) =>
    BigNumber.from(nb).mul(BigNumber.from('10').pow((await token.decimals()) - 3));

  let testContract: BorrowRepayTestMock;

  it('Deploys the test contract', async () => {
    const { weth, dai, pool } = testEnv;

    testContract = await (
      await new BorrowRepayTestMockFactory(await getFirstSigner())
    ).deploy(pool.address, weth.address, dai.address);
  });

  it('Executes a test borrow/repay in the same transaction at variable (revert expected)', async () => {
    await expect(testContract.executeBorrowRepayVariable()).to.be.revertedWith(
      VL_SAME_BLOCK_BORROW_REPAY,
      'Borrow/repay in the same transaction did not revert as expected'
    );
  });

  it('Executes a test borrow/repay in the same transaction at stabke (revert expected)', async () => {
    await expect(testContract.executeBorrowRepayStable()).to.be.revertedWith(
      VL_SAME_BLOCK_BORROW_REPAY,
      'Borrow/repay in the same transaction did not revert as expected'
    );
  });
});