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/tests/mainnet/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 = { 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.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