dsa-connectors/test/mainnet/compound-import/compound-import.test.js

const { expect, should } = require("chai");
const { ethers } = require('hardhat');
const { provider, deployContract} = waffle
const { Signer, Contract } = require("ethers");

const { buildDSAv2 } = require("../../../scripts/tests/buildDSAv2");
const { cEthAddress, cDaiAddress, daiAddress, comptrollerAddress } = require("./constants.js");
const cEthAbi = require("./ABIs/cEthAbi");
const cDaiAbi = require("./ABIs/cDaiAbi");
const comptrollerAbi = require("./ABIs/comptrollerAbi");
const { addresses } = require("../../../scripts/tests/mainnet/addresses");
const { deployAndEnableConnector } = require("../../../scripts/tests/deployAndEnableConnector");
const { abis } = require("../../../scripts/constant/abis");
const { getMasterSigner } = require("../../../scripts/tests/getMasterSigner");
const { parseEther, parseUnits } = require("ethers/lib/utils");
const { encodeSpells } = require("../../../scripts/tests/encodeSpells");
const encodeFlashcastData = require("../../../scripts/tests/encodeFlashcastData").default;
const { ConnectV2CompoundImport__factory } = require("../../../typechain");
const { ConnectV2InstaPoolV4__factory } = require("../../../typechain");


describe('Import Compound', function () {
    // const connectorName = "COMPOUND-IMPORT-ABC";
    const connectorName = "COMPOUND-IMPORT-C";
    const instapoolConnector = "INSTAPOOL-C";
    let dsaWallet0; // signers
    let cEth, cDai, comptroller, Dai; // contracts
    let masterSigner = Signer;
    let connector, connector2;
    let wallet0, wallet1, wallet2, wallet3;
    const wallets = provider.getWallets()


    before(async () => {
        // create (reset) mainnet fork
        await hre.network.provider.request({
            method: "hardhat_reset",
            params: [
                {
                    forking: {
                        jsonRpcUrl: hre.config.networks.hardhat.forking.url,
                        blockNumber: 13300000,
                    },
                },
            ],
        });
        
        // deploy and enable connector contract
        masterSigner = await getMasterSigner()
        instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.core.connectorsV2);

        connector = await deployAndEnableConnector({
            connectorName,
            contractArtifact: ConnectV2CompoundImport__factory,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })
        console.log("Connector address", connector.address);

        connector2 = await deployAndEnableConnector({
            connectorName: instapoolConnector,
            contractArtifact: ConnectV2InstaPoolV4__factory,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })
        console.log("Connector2 address", connector2.address);

        [wallet0, wallet1, wallet2, wallet3] = wallets

        // get an account
        await hre.network.provider.request({
            method: "hardhat_impersonateAccount",
            params: ["0x10a25c6886AE02fde87C5561CDD331d941d0771a"],
        });
        wallet0 = await ethers.getSigner("0x10a25c6886AE02fde87C5561CDD331d941d0771a");

        await hre.network.provider.send("hardhat_setBalance", [
            "0x10a25c6886AE02fde87C5561CDD331d941d0771a",
            parseEther('100000').toHexString()
        ]);

        cEth = new ethers.Contract(cEthAddress, cEthAbi, ethers.provider);
        cDai = new ethers.Contract(cDaiAddress, cDaiAbi, ethers.provider);
        const tokenArtifact = await artifacts.readArtifact("@openzeppelin/contracts/token/ERC20/IERC20.sol:IERC20");
        Dai = new ethers.Contract(daiAddress, tokenArtifact.abi, ethers.provider);
        comptroller = new ethers.Contract(comptrollerAddress, comptrollerAbi, ethers.provider);

        // deposit ether to Compound
        await cEth.connect(wallet0).mint({
            value: parseEther('10')
        });

        // enter markets with deposits
        const cTokens = [cEth.address];
        await comptroller.connect(wallet0).enterMarkets(cTokens);

        // borrow dai from Compound
        await cDai.connect(wallet0).borrow(parseUnits('1000'));
    });

    describe('Deployment', async () => {
        it('Should set correct name', async () => {
            await expect(await connector.name()).to.eq('Compound-Import-v2');
        });
    });

    describe("DSA wallet setup", async () => {
        it("Should build DSA v2", async () => {
            dsaWallet0 = await buildDSAv2(wallet0.address);
            console.log(dsaWallet0.address);
            expect(!!dsaWallet0.address).to.be.true;
        });

        it("Deposit ETH into DSA wallet", async function () {
            await wallet0.sendTransaction({
                to: dsaWallet0.address,
                value: ethers.utils.parseEther("10")
            });
            expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("10"));
        });
    });

    describe('Compound position migration', async () => {
        it('Should migrate Compound position', async () => {
            const flashSpells = [
                {
                    connector: instapoolConnector,
                    method: 'flashPayback',
                    args: [Dai.address, parseUnits('1000.9'), 0, 0],
                }
            ]

            const spells = [
                {
                    connector: instapoolConnector,
                    method: "flashBorrowAndCast",
                    args: [Dai.address, parseUnits('1000'), 0, encodeFlashcastData(flashSpells), "0x"]
                }
            ]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
            const receipt = await tx.wait();
        })
        // take flash loan of dai through spell
        // call contract function
        // repay flash loan of dai
        // check if import was successful
    })
});


// deploy the connector on mainnet fork
// build a new dsa in tests

// create a Compound position
    // deposit some ether in Compound
    // borrow some DAI

// migrate the Compound position
    // cast the migrate spell

// check if migration was successful
    // check the balance of DSA contract address in ERC20 tokens