import BigNumber from "bignumber.js";

import {
  calcExpectedReserveDataAfterDeposit,
  calcExpectedReserveDataAfterRedeem,
  calcExpectedUserDataAfterDeposit,
  calcExpectedUserDataAfterRedeem,
  calcExpectedReserveDataAfterBorrow,
  calcExpectedUserDataAfterBorrow,
  calcExpectedReserveDataAfterRepay,
  calcExpectedUserDataAfterRepay,
  calcExpectedUserDataAfterSetUseAsCollateral,
  calcExpectedUserDataAfterSwapRateMode,
  calcExpectedReserveDataAfterSwapRateMode,
  calcExpectedReserveDataAfterStableRateRebalance,
  calcExpectedUserDataAfterStableRateRebalance,
  calcExpectedUsersDataAfterRedirectInterest,
} from "./utils/calculations";
import {
  getReserveAddressFromSymbol,
  getReserveData,
  getUserData,
} from "./utils/helpers";

import {
  getMintableErc20,
  convertToCurrencyDecimals,
  getAToken,
} from "../../helpers/contracts-helpers";
import {
  MOCK_ETH_ADDRESS,
  ONE_YEAR,
  MAX_UINT_AMOUNT,
} from "../../helpers/constants";
import {TestEnv, SignerWithAddress} from "./make-suite";
import {BRE, increaseTime, timeLatest} from "../../helpers/misc-utils";

import chai from "chai";
import {ReserveData, UserReserveData} from "./utils/interfaces";
import {waitForTx} from "../__setup.spec";
import {ContractReceipt} from "ethers/contract";
import {ethers} from "ethers";
import {AToken} from "../../types/AToken";
import {tEthereumAddress} from "../../helpers/types";

const {expect} = chai;

const almostEqualOrEqual = function (
  this: any,
  expected: ReserveData | UserReserveData,
  actual: ReserveData | UserReserveData
) {
  const keys = Object.keys(actual);

  keys.forEach((key) => {
    if (
      key === "lastUpdateTimestamp" ||
      key === "marketStableRate" ||
      key === "symbol" ||
      key === "aTokenAddress" ||
      key === "initialATokenExchangeRate" ||
      key === "decimals"
    ) {
      // skipping consistency check on accessory data
      return;
    }

    this.assert(
      actual[key] != undefined,
      `Property ${key} is undefined in the actual data`
    );
    expect(
      expected[key] != undefined,
      `Property ${key} is undefined in the expected data`
    );

    if (actual[key] instanceof BigNumber) {
      const actualValue = (<BigNumber>actual[key]).decimalPlaces(
        0,
        BigNumber.ROUND_DOWN
      );
      const expectedValue = (<BigNumber>expected[key]).decimalPlaces(
        0,
        BigNumber.ROUND_DOWN
      );

      this.assert(
        actualValue.eq(expectedValue) ||
          actualValue.plus(1).eq(expectedValue) ||
          actualValue.eq(expectedValue.plus(1)) ||
          actualValue.plus(2).eq(expectedValue) ||
          actualValue.eq(expectedValue.plus(2)) ||
          actualValue.plus(3).eq(expectedValue) ||
          actualValue.eq(expectedValue.plus(3)),
        `expected #{act} to be almost equal or equal #{exp} for property ${key}`,
        `expected #{act} to be almost equal or equal #{exp} for property ${key}`,
        expectedValue.toFixed(0),
        actualValue.toFixed(0)
      );
    } else {
      this.assert(
        actual[key] !== null &&
          expected[key] !== null &&
          actual[key].toString() === expected[key].toString(),
        `expected #{act} to be equal #{exp} for property ${key}`,
        `expected #{act} to be equal #{exp} for property ${key}`,
        expected[key],
        actual[key]
      );
    }
  });
};

chai.use(function (chai: any, utils: any) {
  chai.Assertion.overwriteMethod("almostEqualOrEqual", function (
    original: any
  ) {
    return function (this: any, expected: ReserveData | UserReserveData) {
      const actual = (expected as ReserveData)
        ? <ReserveData>this._obj
        : <UserReserveData>this._obj;

      almostEqualOrEqual.apply(this, [expected, actual]);
    };
  });
});

interface ActionsConfig {
  skipIntegrityCheck: boolean;
}

export const configuration: ActionsConfig = <ActionsConfig>{};

export const mint = async (
  reserveSymbol: string,
  amount: string,
  user: SignerWithAddress
) => {
  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  if (MOCK_ETH_ADDRESS.toLowerCase() === reserve.toLowerCase()) {
    throw "Cannot mint ethereum. Mint action is most likely not needed in this story";
  }

  const token = await getMintableErc20(reserve);

  await waitForTx(
    await token
      .connect(user.signer)
      .mint(await convertToCurrencyDecimals(reserve, amount))
  );
};

export const approve = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  testEnv: TestEnv
) => {
  const {pool} = testEnv;
  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  if (MOCK_ETH_ADDRESS.toLowerCase() === reserve.toLowerCase()) {
    throw "Cannot mint ethereum. Mint action is most likely not needed in this story";
  }

  const token = await getMintableErc20(reserve);

  await token
    .connect(user.signer)
    .approve(pool.address, "100000000000000000000000000000");
};

export const deposit = async (
  reserveSymbol: string,
  amount: string,
  user: SignerWithAddress,
  sendValue: string,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const amountToDeposit = await convertToCurrencyDecimals(reserve, amount);

  const txOptions: any = {};

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, user.address, testEnv);

  if (MOCK_ETH_ADDRESS === reserve) {
    if (sendValue) {
      const valueToSend = await convertToCurrencyDecimals(reserve, sendValue);
      txOptions.value = valueToSend;
    } else {
      txOptions.value = amountToDeposit;
    }
  }
  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await await pool
        .connect(user.signer)
        .deposit(reserve, amountToDeposit, "0", txOptions)
    );

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
      timestamp,
    } = await getContractsData(reserve, user.address, testEnv);

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const expectedReserveData = calcExpectedReserveDataAfterDeposit(
      amountToDeposit.toString(),
      reserveDataBefore,
      txTimestamp
    );

    const expectedUserReserveData = calcExpectedUserDataAfterDeposit(
      amountToDeposit.toString(),
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      txTimestamp,
      timestamp,
      txCost
    );

    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserReserveData);

    // truffleAssert.eventEmitted(txResult, "Deposit", (ev: any) => {
    //   const {_reserve, _user, _amount} = ev;
    //   return (
    //     _reserve === reserve &&
    //     _user === user &&
    //     new BigNumber(_amount).isEqualTo(new BigNumber(amountToDeposit))
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      pool
        .connect(user.signer)
        .deposit(reserve, amountToDeposit, "0", txOptions),
      revertMessage
    ).to.be.reverted;
  }
};

export const redeem = async (
  reserveSymbol: string,
  amount: string,
  user: SignerWithAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {

  const {
    aTokenInstance,
    reserve,
    userData: userDataBefore,
    reserveData: reserveDataBefore,
  } = await getDataBeforeAction(reserveSymbol, user.address, testEnv);
  
  let amountToRedeem = "0";

  if (amount !== "-1") {
    amountToRedeem = (
      await convertToCurrencyDecimals(reserve, amount)
    ).toString();
  } else {
    amountToRedeem = MAX_UINT_AMOUNT;
  }

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await aTokenInstance.connect(user.signer).redeem(amountToRedeem)
    );

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
      timestamp,
    } = await getContractsData(reserve, user.address, testEnv);

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const expectedReserveData = calcExpectedReserveDataAfterRedeem(
      amountToRedeem,
      reserveDataBefore,
      userDataBefore,
      txTimestamp
    );

    const expectedUserData = calcExpectedUserDataAfterRedeem(
      amountToRedeem,
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      txTimestamp,
      timestamp,
      txCost
    );

    const actualAmountRedeemed = userDataBefore.currentATokenBalance.minus(
      expectedUserData.currentATokenBalance
    );

    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserData);

    // truffleAssert.eventEmitted(txResult, "Redeem", (ev: any) => {
    //   const {_from, _value} = ev;
    //   return (
    //     _from === user && new BigNumber(_value).isEqualTo(actualAmountRedeemed)
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      aTokenInstance.connect(user.signer).redeem(amountToRedeem),
      revertMessage
    ).to.be.reverted;
  }
};

export const borrow = async (
  reserveSymbol: string,
  amount: string,
  interestRateMode: string,
  user: SignerWithAddress,
  timeTravel: string,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, user.address, testEnv);

  const amountToBorrow = await convertToCurrencyDecimals(reserve, amount);

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await pool
        .connect(user.signer)
        .borrow(reserve, amountToBorrow, interestRateMode, "0")
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    if (timeTravel) {
      const secondsToTravel = new BigNumber(timeTravel)
        .multipliedBy(ONE_YEAR)
        .div(365)
        .toNumber();

      await increaseTime(secondsToTravel);
      // await advanceBlock(new Date().getTime());
      // TODO
      // await time.increase(secondsToTravel);
    }

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
      timestamp,
    } = await getContractsData(reserve, user.address, testEnv);

    const expectedReserveData = calcExpectedReserveDataAfterBorrow(
      amountToBorrow.toString(),
      interestRateMode,
      reserveDataBefore,
      userDataBefore,
      txTimestamp,
      timestamp
    );

    const expectedUserData = calcExpectedUserDataAfterBorrow(
      amountToBorrow.toString(),
      interestRateMode,
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      txTimestamp,
      timestamp,
      txCost
    );
    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserData);

    // truffleAssert.eventEmitted(txResult, "Borrow", (ev: any) => {
    //   const {
    //     _reserve,
    //     _user,
    //     _amount,
    //     _borrowRateMode,
    //     _borrowRate,
    //     _originationFee,
    //   } = ev;
    //   return (
    //     _reserve.toLowerCase() === reserve.toLowerCase() &&
    //     _user.toLowerCase() === user.toLowerCase() &&
    //     new BigNumber(_amount).eq(amountToBorrow) &&
    //     new BigNumber(_borrowRateMode).eq(expectedUserData.borrowRateMode) &&
    //     new BigNumber(_borrowRate).eq(expectedUserData.borrowRate) &&
    //     new BigNumber(_originationFee).eq(
    //       expectedUserData.originationFee.minus(userDataBefore.originationFee)
    //     )
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      pool
        .connect(user.signer)
        .borrow(reserve, amountToBorrow, interestRateMode, "0"),
      revertMessage
    ).to.be.reverted;
  }
};

export const repay = async (
  reserveSymbol: string,
  amount: string,
  user: SignerWithAddress,
  onBehalfOf: SignerWithAddress,
  sendValue: string,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;
  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, onBehalfOf.address, testEnv);

  let amountToRepay = "0";

  if (amount !== "-1") {
    amountToRepay = (
      await convertToCurrencyDecimals(reserve, amount)
    ).toString();
  } else {
    amountToRepay = ethers.utils.bigNumberify(MAX_UINT_AMOUNT).toString();
  }
  amountToRepay = "0x" + new BigNumber(amountToRepay).toString(16);

  const txOptions: any = {};

  if (MOCK_ETH_ADDRESS === reserve) {
    if (sendValue) {
      if (sendValue !== "-1") {
        const valueToSend = await convertToCurrencyDecimals(reserve, sendValue);
        txOptions.value =
          "0x" + new BigNumber(valueToSend.toString()).toString(16);
      } else {
        txOptions.value =
          "0x" +
          userDataBefore.currentBorrowBalance
            .plus((await convertToCurrencyDecimals(reserve, "0.1")).toString())
            .toString(16); //add 0.1 ETH to the repayment amount to cover for accrued interest during tx execution
      }
    } else {
      txOptions.value = amountToRepay;
    }
  }

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await pool
        .connect(user.signer)
        .repay(reserve, amountToRepay, onBehalfOf.address, txOptions)
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
      timestamp,
    } = await getContractsData(reserve, onBehalfOf.address, testEnv);

    const expectedReserveData = calcExpectedReserveDataAfterRepay(
      amountToRepay,
      reserveDataBefore,
      userDataBefore,
      txTimestamp,
      timestamp
    );

    const expectedUserData = calcExpectedUserDataAfterRepay(
      amountToRepay,
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      user.address,
      onBehalfOf.address,
      txTimestamp,
      timestamp,
      txCost
    );

    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserData);

    // truffleAssert.eventEmitted(txResult, "Repay", (ev: any) => {
    //   const {_reserve, _user, _repayer} = ev;

    //   return (
    //     _reserve.toLowerCase() === reserve.toLowerCase() &&
    //     _user.toLowerCase() === onBehalfOf.toLowerCase() &&
    //     _repayer.toLowerCase() === user.toLowerCase()
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      pool
        .connect(user.signer)
        .repay(reserve, amountToRepay, onBehalfOf.address, txOptions),
      revertMessage
    ).to.be.reverted;
  }
};

export const setUseAsCollateral = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  useAsCollateral: string,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, user.address, testEnv);

  const useAsCollateralBool = useAsCollateral.toLowerCase() === "true";

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await pool
        .connect(user.signer)
        .setUserUseReserveAsCollateral(reserve, useAsCollateralBool)
    );

    const {txCost} = await getTxCostAndTimestamp(txResult);

    const {userData: userDataAfter} = await getContractsData(
      reserve,
      user.address,
      testEnv
    );

    const expectedUserData = calcExpectedUserDataAfterSetUseAsCollateral(
      useAsCollateral.toLocaleLowerCase() === "true",
      reserveDataBefore,
      userDataBefore,
      txCost
    );

    expectEqual(userDataAfter, expectedUserData);
    // if (useAsCollateralBool) {
    //   truffleAssert.eventEmitted(txResult, 'ReserveUsedAsCollateralEnabled', (ev: any) => {
    //     const {_reserve, _user} = ev;
    //     return _reserve === reserve && _user === user;
    //   });
    // } else {
    //   truffleAssert.eventEmitted(txResult, 'ReserveUsedAsCollateralDisabled', (ev: any) => {
    //     const {_reserve, _user} = ev;
    //     return _reserve === reserve && _user === user;
    //   });
    // }
  } else if (expectedResult === "revert") {
    await expect(
      pool
        .connect(user.signer)
        .setUserUseReserveAsCollateral(reserve, useAsCollateralBool),
      revertMessage
    ).to.be.reverted;
  }
};

export const swapBorrowRateMode = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, user.address, testEnv);

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await pool.connect(user.signer).swapBorrowRateMode(reserve)
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
    } = await getContractsData(reserve, user.address, testEnv);

    const expectedReserveData = calcExpectedReserveDataAfterSwapRateMode(
      reserveDataBefore,
      userDataBefore,
      txTimestamp
    );

    const expectedUserData = calcExpectedUserDataAfterSwapRateMode(
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      txCost,
      txTimestamp
    );

    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserData);

    // truffleAssert.eventEmitted(txResult, "Swap", (ev: any) => {
    //   const {_user, _reserve, _newRateMode, _newRate} = ev;
    //   return (
    //     _user === user &&
    //     _reserve == reserve &&
    //     new BigNumber(_newRateMode).eq(expectedUserData.borrowRateMode) &&
    //     new BigNumber(_newRate).eq(expectedUserData.borrowRate)
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      pool.connect(user.signer).swapBorrowRateMode(reserve),
      revertMessage
    ).to.be.reverted;
  }
};

export const rebalanceStableBorrowRate = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  target: SignerWithAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {pool} = testEnv;

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {
    reserveData: reserveDataBefore,
    userData: userDataBefore,
  } = await getContractsData(reserve, target.address, testEnv);

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await pool
        .connect(user.signer)
        .rebalanceStableBorrowRate(reserve, target.address)
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const {
      reserveData: reserveDataAfter,
      userData: userDataAfter,
    } = await getContractsData(reserve, target.address, testEnv);

    const expectedReserveData = calcExpectedReserveDataAfterStableRateRebalance(
      reserveDataBefore,
      userDataBefore,
      txTimestamp
    );

    const expectedUserData = calcExpectedUserDataAfterStableRateRebalance(
      reserveDataBefore,
      expectedReserveData,
      userDataBefore,
      txCost,
      txTimestamp
    );

    expectEqual(reserveDataAfter, expectedReserveData);
    expectEqual(userDataAfter, expectedUserData);

    // truffleAssert.eventEmitted(txResult, 'RebalanceStableBorrowRate', (ev: any) => {
    //   const {_user, _reserve, _newStableRate} = ev;
    //   return (
    //     _user.toLowerCase() === target.toLowerCase() &&
    //     _reserve.toLowerCase() === reserve.toLowerCase() &&
    //     new BigNumber(_newStableRate).eq(expectedUserData.borrowRate)
    //   );
    // });
  } else if (expectedResult === "revert") {
    await expect(
      pool
        .connect(user.signer)
        .rebalanceStableBorrowRate(reserve, target.address),
      revertMessage
    ).to.be.reverted;
  }
};

export const redirectInterestStream = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  to: tEthereumAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {
    aTokenInstance,
    reserve,
    userData: fromDataBefore,
    reserveData: reserveDataBefore,
  } = await getDataBeforeAction(reserveSymbol, user.address, testEnv);

  const {userData: toDataBefore} = await getContractsData(reserve, to, testEnv);

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await aTokenInstance.connect(user.signer).redirectInterestStream(to)
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const {userData: fromDataAfter} = await getContractsData(
      reserve,
      user.address,
      testEnv
    );

    const {userData: toDataAfter} = await getContractsData(
      reserve,
      to,
      testEnv
    );

    const [
      expectedFromData,
      expectedToData,
    ] = calcExpectedUsersDataAfterRedirectInterest(
      reserveDataBefore,
      fromDataBefore,
      toDataBefore,
      user.address,
      to,
      true,
      txCost,
      txTimestamp
    );

    expectEqual(fromDataAfter, expectedFromData);
    expectEqual(toDataAfter, expectedToData);

    // truffleAssert.eventEmitted(txResult, 'InterestStreamRedirected', (ev: any) => {
    //   const {_from, _to} = ev;
    //   return _from === user
    //   && _to === (to === user ? NIL_ADDRESS : to);
    // });
  } else if (expectedResult === "revert") {
    await expect(
      aTokenInstance.connect(user.signer).redirectInterestStream(to),
      revertMessage
    ).to.be.reverted;
  }
};

export const redirectInterestStreamOf = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  from: tEthereumAddress,
  to: tEthereumAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {
    aTokenInstance,
    reserve,
    userData: fromDataBefore,
    reserveData: reserveDataBefore,
  } = await getDataBeforeAction(reserveSymbol, from, testEnv);

  const {userData: toDataBefore} = await getContractsData(
    reserve,
    user.address,
    testEnv
  );

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await aTokenInstance
        .connect(user.signer)
        .redirectInterestStreamOf(from, to)
    );

    const {txCost, txTimestamp} = await getTxCostAndTimestamp(txResult);

    const {userData: fromDataAfter} = await getContractsData(
      reserve,
      from,
      testEnv
    );

    const {userData: toDataAfter} = await getContractsData(
      reserve,
      to,
      testEnv
    );

    const [
      expectedFromData,
      exptectedToData,
    ] = calcExpectedUsersDataAfterRedirectInterest(
      reserveDataBefore,
      fromDataBefore,
      toDataBefore,
      from,
      to,
      from === user.address,
      txCost,
      txTimestamp
    );

    expectEqual(fromDataAfter, expectedFromData);
    expectEqual(toDataAfter, exptectedToData);

    // truffleAssert.eventEmitted(
    //   txResult,
    //   "InterestStreamRedirected",
    //   (ev: any) => {
    //     const {_from, _to} = ev;
    //     return (
    //       _from.toLowerCase() === from.toLowerCase() &&
    //       _to.toLowerCase() === to.toLowerCase()
    //     );
    //   }
    // );
  } else if (expectedResult === "revert") {
    await expect(
      aTokenInstance.connect(user.signer).redirectInterestStreamOf(from, to),
      revertMessage
    ).to.be.reverted;
  }
};

export const allowInterestRedirectionTo = async (
  reserveSymbol: string,
  user: SignerWithAddress,
  to: tEthereumAddress,
  expectedResult: string,
  testEnv: TestEnv,
  revertMessage?: string
) => {
  const {aTokenInstance} = await getDataBeforeAction(
    reserveSymbol,
    user.address,
    testEnv
  );

  if (expectedResult === "success") {
    const txResult = await waitForTx(
      await aTokenInstance.connect(user.signer).allowInterestRedirectionTo(to)
    );

    // truffleAssert.eventEmitted(
    //   txResult,
    //   "InterestRedirectionAllowanceChanged",
    //   (ev: any) => {
    //     const {_from, _to} = ev;
    //     return (
    //       _from.toLowerCase() === user.toLowerCase() &&
    //       _to.toLowerCase() === to.toLowerCase()
    //     );
    //   }
    // );
  } else if (expectedResult === "revert") {
    await expect(
      aTokenInstance.connect(user.signer).allowInterestRedirectionTo(to),
      revertMessage
    ).to.be.reverted;
  }
};

const expectEqual = (
  actual: UserReserveData | ReserveData,
  expected: UserReserveData | ReserveData
) => {
  if (!configuration.skipIntegrityCheck) {
    // @ts-ignore
    expect(actual).to.be.almostEqualOrEqual(expected);
  }
};

interface ActionData {
  reserve: string;
  reserveData: ReserveData;
  userData: UserReserveData;
  aTokenInstance: AToken;
}

const getDataBeforeAction = async (
  reserveSymbol: string,
  user: tEthereumAddress,
  testEnv: TestEnv
): Promise<ActionData> => {

  const reserve = await getReserveAddressFromSymbol(reserveSymbol);

  const {reserveData, userData} = await getContractsData(
    reserve,
    user,
    testEnv
  );
  const aTokenInstance = await getAToken(reserveData.aTokenAddress);
  return {
    reserve,
    reserveData,
    userData,
    aTokenInstance,
  };
};

const getTxCostAndTimestamp = async (tx: ContractReceipt) => {
  if (!tx.blockNumber || !tx.transactionHash || !tx.cumulativeGasUsed) {
    throw new Error("No tx blocknumber");
  }
  const txTimestamp = new BigNumber(
    (await BRE.ethers.provider.getBlock(tx.blockNumber)).timestamp
  );

  const txInfo = await BRE.ethers.provider.getTransaction(tx.transactionHash);
  const txCost = new BigNumber(tx.cumulativeGasUsed.toString()).multipliedBy(
    txInfo.gasPrice.toString()
  );

  return {txCost, txTimestamp};
};

const getContractsData = async (
  reserve: string,
  user: string,
  testEnv: TestEnv
) => {
  const {pool} = testEnv;
  const reserveData = await getReserveData(pool, reserve);
  const userData = await getUserData(pool, reserve, user);
  const timestamp = await timeLatest();

  return {
    reserveData,
    userData,
    timestamp: new BigNumber(timestamp),
  };
};