dsa-connectors/test/mainnet/uniswapStake/uniswapStake.test.ts

import { expect } from "chai";
import hre from "hardhat";
const { web3, deployments, waffle, ethers } = hre;
const { provider, deployContract } = waffle

import { deployAndEnableConnector } from "../../../scripts/tests/deployAndEnableConnector";
import { buildDSAv2 } from "../../../scripts/tests/buildDSAv2";
import { encodeSpells } from "../../../scripts/tests/encodeSpells";
import { getMasterSigner } from "../../../scripts/tests/getMasterSigner";
import { addLiquidity } from "../../../scripts/tests/addLiquidity";

import addresses from "../../../scripts/constant/addresses";
import abis from "../../../scripts/constant/abis";
import { abi } from "@uniswap/v3-periphery/artifacts/contracts/NonfungiblePositionManager.sol/NonfungiblePositionManager.json"
import type { Signer, Contract } from "ethers";

import { ConnectV2UniswapV3Staker__factory, ConnectV2UniswapV3__factory } from "../../../typechain";

const FeeAmount = {
    LOW: 500,
    MEDIUM: 3000,
    HIGH: 10000,
}

const TICK_SPACINGS: Record<number, number> = {
    500: 10,
    3000: 60,
    10000: 200
}

const DAI_ADDR = "0x6b175474e89094c44da98b954eedeac495271d0f"
const ethAddress = "0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
const INST_ADDR = "0x6f40d4a6237c257fff2db00fa0510deeecd303eb"

let tokenIds: any[] = []
const abiCoder = ethers.utils.defaultAbiCoder

describe("UniswapV3", function () {
    const connectorStaker = "UniswapStaker-v1"
    const connectorUniswap = "UniswapV3-v1"

    let dsaWallet0: any
    let masterSigner: Signer;
    let instaConnectorsV2: any;
    let connector: any;
    let startTime: any, endTime: any;
    let nftManager: Contract;

    const wallets = provider.getWallets()
    const [wallet0, wallet1, wallet2, wallet3] = wallets
    before(async () => {
        await hre.network.provider.request({
            method: "hardhat_reset",
            params: [
                {
                    forking: {
                        // @ts-ignore
                        jsonRpcUrl: hre.config.networks.hardhat.forking.url,
                        blockNumber: 13300000,
                    },
                },
            ],
        });
        masterSigner = await getMasterSigner()
        instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.mainnet.core.connectorsV2);
        nftManager = await ethers.getContractAt(abi, "0xC36442b4a4522E871399CD717aBDD847Ab11FE88");
        connector = await deployAndEnableConnector({
            connectorName: connectorStaker,
            contractArtifact: ConnectV2UniswapV3Staker__factory,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })
        console.log("Connector address", connector.address)

        let uniswapConnector = await deployAndEnableConnector({
            connectorName: connectorUniswap,
            contractArtifact: ConnectV2UniswapV3__factory,
            signer: masterSigner,
            connectors: instaConnectorsV2
        });
    })

    it("Should have contracts deployed.", async function () {
        expect(!!instaConnectorsV2.address).to.be.true;
        expect(!!connector.address).to.be.true;
        expect(!!(await masterSigner.getAddress())).to.be.true;
    });

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

        it("Deposit ETH & DAI 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"));

            await addLiquidity("dai", dsaWallet0.address, ethers.utils.parseEther("100000"));
        });

        it("Deposit ETH & USDT & INST 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"));

            await addLiquidity("dai", dsaWallet0.address, ethers.utils.parseEther("100000"));
            await addLiquidity("usdt", dsaWallet0.address, ethers.utils.parseEther("100000"));
            await addLiquidity("inst", dsaWallet0.address, ethers.utils.parseEther("10000"));
        });
    });

    describe("Main", function () {
        const ethAmount = ethers.utils.parseEther("0.1") // 1 ETH
        const daiAmount = ethers.utils.parseEther("400") // 1 ETH
        const instAmount = ethers.utils.parseEther("50")

        it("Should mint successfully", async function () {
            const getIds = ["0", "0"]
            const setId = "0"

            const spells = [
                {
                    connector: connectorUniswap,
                    method: "mint",
                    args: [
                        DAI_ADDR,
                        ethAddress,
                        FeeAmount.MEDIUM,
                        getMinTick(TICK_SPACINGS[FeeAmount.MEDIUM]),
                        getMaxTick(TICK_SPACINGS[FeeAmount.MEDIUM]),
                        daiAmount,
                        ethAmount,
                        "500000000000000000",
                        getIds,
                        setId
                    ],
                },
                {
                    connector: connectorUniswap,
                    method: "mint",
                    args: [
                        INST_ADDR,
                        ethAddress,
                        FeeAmount.MEDIUM,
                        getMinTick(TICK_SPACINGS[FeeAmount.MEDIUM]),
                        getMaxTick(TICK_SPACINGS[FeeAmount.MEDIUM]),
                        instAmount,
                        ethAmount,
                        "500000000000000000",
                        getIds,
                        setId
                    ],
                }
            ]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
            let receipt = await tx.wait()

            let castEvent = new Promise((resolve, reject) => {
                dsaWallet0.on('LogCast', (origin: any, sender: any, value: any, targetNames: any, targets: any, eventNames: any, eventParams: any, event: any) => {
                    const params = abiCoder.decode(["uint256", "uint256", "uint256", "uint256", "int24", "int24"], eventParams[0]);
                    const params1 = abiCoder.decode(["uint256", "uint256", "uint256", "uint256", "int24", "int24"], eventParams[1]);
                    tokenIds.push(params[0]);
                    tokenIds.push(params1[0]);
                    event.removeListener();

                    resolve({
                        eventNames,
                    });
                });

                setTimeout(() => {
                    reject(new Error('timeout'));
                }, 60000)
            });

            let event = await castEvent

            let balance = await nftManager.connect(wallet0).balanceOf(dsaWallet0.address)
            console.log("Balance", balance.toString())
        });

        it("Should create incentive successfully", async function () {
            const spells = [
                {
                    connector: connectorStaker,
                    method: "createIncentive",
                    args: [
                        ethAddress,
                        "1000",
                        dsaWallet0.address,
                        "0xc2e9f25be6257c210d7adf0d4cd6e3e881ba25f8",
                        ethers.utils.parseEther("0.01")
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "createIncentive",
                    args: [
                        INST_ADDR,
                        "50",
                        dsaWallet0.address,
                        "0xcba27c8e7115b4eb50aa14999bc0866674a96ecb",
                        ethers.utils.parseEther("0.01")
                    ],
                }]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet0.address)
            let receipt = await tx.wait()

            let castEvent = new Promise((resolve, reject) => {
                dsaWallet0.on('LogCast', (origin: any, sender: any, value: any, targetNames: any, targets: any, eventNames: any, eventParams: any, event: any) => {
                    const params = abiCoder.decode(["bytes32", "address", "address", "uint256", "uint256", "uint256"], eventParams[0]);
                    const params1 = abiCoder.decode(["bytes32", "address", "address", "uint256", "uint256", "uint256"], eventParams[1]);
                    event.removeListener();

                    resolve({ start: [params[3], params1[3]], end: [params[4], params1[4]] });
                });

                setTimeout(() => {
                    reject(new Error('timeout'));
                }, 60000)
            });

            let event: any = await castEvent
            startTime = event.start;
            endTime = event.end;
        });

        it("Should stake successfully", async function () {
            const spells = [
                {
                    connector: connectorStaker,
                    method: "deposit",
                    args: [
                        tokenIds[0]
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "stake",
                    args: [
                        ethAddress,
                        startTime[0],
                        endTime[0],
                        dsaWallet0.address,
                        tokenIds[0]
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "deposit",
                    args: [
                        tokenIds[1]
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "stake",
                    args: [
                        INST_ADDR,
                        startTime[1],
                        endTime[1],
                        dsaWallet0.address,
                        tokenIds[1]
                    ],
                }
            ]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
            let receipt = await tx.wait()

            let balance = await nftManager.connect(wallet0).balanceOf(dsaWallet0.address)
            console.log("Balance", balance.toString())
        });

        it("Should claim rewards successfully", async function () {
            const spells = [
                {
                    connector: connectorStaker,
                    method: "claimRewards",
                    args: [
                        ethAddress,
                        "1000",
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "claimRewards",
                    args: [
                        INST_ADDR,
                        "1000",
                    ],
                }
            ]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
            let receipt = await tx.wait()
        });

        it("Should unstake successfully", async function () {
            const spells = [
                {
                    connector: connectorStaker,
                    method: "unstake",
                    args: [
                        ethAddress,
                        startTime[0],
                        endTime[0],
                        dsaWallet0.address,
                        tokenIds[0]
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "withdraw",
                    args: [
                        tokenIds[0],
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "unstake",
                    args: [
                        INST_ADDR,
                        startTime[1],
                        endTime[1],
                        dsaWallet0.address,
                        tokenIds[1]
                    ],
                },
                {
                    connector: connectorStaker,
                    method: "withdraw",
                    args: [
                        tokenIds[1]
                    ],
                }
            ]

            const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
            let receipt = await tx.wait()

            let balance = await nftManager.connect(wallet0).balanceOf(dsaWallet0.address)
            console.log("Balance", balance.toString())
        });
    })
})

const getMinTick = (tickSpacing: number) => Math.ceil(-887272 / tickSpacing) * tickSpacing
const getMaxTick = (tickSpacing: number) => Math.floor(887272 / tickSpacing) * tickSpacing