mirror of
https://github.com/Instadapp/dsa-connectors-2.0.git
synced 2024-07-29 21:57:39 +00:00
358 lines
13 KiB
TypeScript
358 lines
13 KiB
TypeScript
|
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<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.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
|