import { expect } from "chai"; import hre from "hardhat"; const { web3, deployments, waffle, ethers } = hre; const { provider, deployContract } = waffle; const ALCHEMY_ID = process.env.ALCHEMY_ID; 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 { addresses } from "../../../scripts/tests/polygon/addresses"; import { abis } from "../../../scripts/constant/abis"; import { tokens } from "../../../scripts/tests/polygon/tokens"; import type { Signer, Contract } from "ethers"; import { ConnectV2AaveV2Polygon__factory, ConnectV2PoolTogetherPolygon__factory } from "../../../typechain"; const DAI_TOKEN_ADDR = tokens.dai.address; // DAI Token // PoolTogether Address: https://docs.pooltogether.com/resources/networks/matic const USDC_PRIZE_POOL_ADDR = "0xEE06AbE9e2Af61cabcb13170e01266Af2DEFa946"; // USDC Prize Pool const PT_USDC_TICKET_ADDR = "0x473E484c722EF9ec6f63B509b07Bb9cfB258820b"; // PT USDC Ticket const PT_USDC_SPONGSOR_TICKET_ADDR = "0x19c0e557ee5a9b456f613ba3d025a4dc45b52c35"; // PT USDC Sponsor Ticket const USDC_POOL_FAUCET_ADDR = "0x6cbc003fE015D753180f072d904bA841b2415498"; // USDC POOL Faucet const POOL_TOKEN_ADDRESS = "0x25788a1a171ec66Da6502f9975a15B609fF54CF6"; // POOL Tocken const TOKEN_FAUCET_PROXY_FACTORY_ADDR = "0xeaa636304a7C8853324B6b603dCdE55F92dfbab1"; // TokenFaucetProxyFactory for claimAll // Community WETH Prize Pool (Rari): https://reference-app.pooltogether.com/pools/mainnet/0xa88ca010b32a54d446fc38091ddbca55750cbfc3/manage#stats const WETH_PRIZE_POOL_ADDR = "0xa88ca010b32a54d446fc38091ddbca55750cbfc3"; // Community WETH Prize Pool (Rari) const WETH_POOL_TICKET_ADDR = "0x9b5c30aeb9ce2a6a121cea9a85bc0d662f6d9b40"; // Community WETH Prize Pool Ticket (Rari) const prizePoolABI = [ "function calculateEarlyExitFee( address from, address controlledToken, uint256 amount) external returns ( uint256 exitFee, uint256 burnedCredit)", "function creditPlanOf( address controlledToken) external view returns ( uint128 creditLimitMantissa, uint128 creditRateMantissa)", ]; const connectorsABI = [ "function isConnectors(string[] calldata connectorNames) external view returns (bool, address[] memory)", ]; describe("PoolTogether", function() { const connectorName = "AAVEV2-TEST-A"; const ptConnectorName = "POOLTOGETHER-TEST-A"; let dsaWallet0: any; let masterSigner: Signer; let instaConnectorsV2: Contract; let connector: any; let ptConnector: Contract; const wallets = provider.getWallets(); const [wallet0, wallet1, wallet2, wallet3] = wallets; before(async () => { await hre.network.provider.request({ method: "hardhat_reset", params: [ { forking: { jsonRpcUrl: `https://polygon-mainnet.g.alchemy.com/v2/${ALCHEMY_ID}`, blockNumber: 18717337, }, }, ], }); masterSigner = await getMasterSigner(); instaConnectorsV2 = await ethers.getContractAt( abis.core.connectorsV2, addresses.core.connectorsV2 ); // Deploy and enable Compound Connector connector = await deployAndEnableConnector({ connectorName, contractArtifact: ConnectV2AaveV2Polygon__factory, signer: masterSigner, connectors: instaConnectorsV2, }); // Deploy and enable Pool Together Connector ptConnector = await deployAndEnableConnector({ connectorName: ptConnectorName, contractArtifact: ConnectV2PoolTogetherPolygon__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(!!ptConnector.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 1000 MATIC into DSA wallet", async function() { await wallet0.sendTransaction({ to: dsaWallet0.address, value: ethers.utils.parseEther("1000"), }); expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte( ethers.utils.parseEther("1000") ); }); }); describe("Main - USDC Prize Pool Test", function() { it("Should deposit 100 MATIC in AAVE V2", async function() { const amount = ethers.utils.parseEther("100"); // 100 MATIC const spells = [ { connector: connectorName, method: "deposit", args: [tokens.eth.address, amount, 0, 0], }, ]; const tx = await dsaWallet0.cast( ...encodeSpells(spells), wallet1.address ); const receipt = await tx.wait(); expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.lte( ethers.utils.parseEther("900") ); }); it("Should borrow 10 USDC from AAVE V2 and deposit USDC into USDC Prize Pool", async function() { const amount = ethers.utils.parseUnits("10", 6); // 10 USDC const setId = "83478237"; const spells = [ { connector: connectorName, method: "borrow", args: [tokens.usdc.address, amount, 2, 0, setId], }, { connector: ptConnectorName, method: "depositTo", args: [ USDC_PRIZE_POOL_ADDR, amount, PT_USDC_SPONGSOR_TICKET_ADDR, setId, 0, ], }, ]; // Before Spell let usdcToken = await ethers.getContractAt( abis.basic.erc20, tokens.usdc.address ); let usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect(usdcBalance, `USDC balance is 0`).to.be.eq( ethers.utils.parseUnits("0", 6) ); let cToken = await ethers.getContractAt( abis.basic.erc20, PT_USDC_SPONGSOR_TICKET_ADDR ); const balance = await cToken.balanceOf(dsaWallet0.address); expect(balance, `PoolTogether USDC Ticket balance is 0`).to.be.eq(0); // Run spell transaction const tx = await dsaWallet0 .connect(wallet0) .cast(...encodeSpells(spells), wallet1.address); const receipt = await tx.wait(); // After spell usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect( usdcBalance, `Expect USDC balance to still equal 0 since it was deposited into Prize Pool` ).to.be.eq(0); const balanceAfter = await cToken.balanceOf(dsaWallet0.address); expect( balanceAfter, `PoolTogether USDC Ticket balance equals 10` ).to.be.eq(ethers.utils.parseUnits("10", 6)); // ETH used for transaction expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.lte( ethers.utils.parseEther("900") ); }); it("Should wait 11 days, withdraw all PrizePool, get back 10 USDC, and claim POOL", async function() { const amount = ethers.utils.parseUnits("10", 6); // 10 USDC let prizePoolContract = new ethers.Contract( USDC_PRIZE_POOL_ADDR, prizePoolABI, ethers.provider ); // const { creditLimitMantissa, creditRateMantissa } = await prizePoolContract.creditPlanOf(PT_USDC_TICKET_ADDR); // console.log("CreditLimitMantiss: ", creditLimitMantissa.toString()); // console.log("CreditRateMantiss: ", creditRateMantissa.toString()); let earlyExitFee = await prizePoolContract.callStatic[ "calculateEarlyExitFee" ](dsaWallet0.address, PT_USDC_SPONGSOR_TICKET_ADDR, amount); expect( earlyExitFee.exitFee, "Exit Fee equal to 0 USDC because 0% fee for sponsorship ticket" ).to.be.eq(ethers.utils.parseUnits("0", 6)); const spells = [ { connector: ptConnectorName, method: "claim", args: [USDC_POOL_FAUCET_ADDR, 0], }, { connector: ptConnectorName, method: "withdrawInstantlyFrom", args: [ USDC_PRIZE_POOL_ADDR, amount, PT_USDC_SPONGSOR_TICKET_ADDR, earlyExitFee.exitFee, 0, 0, ], }, ]; // Before spell let usdcToken = await ethers.getContractAt( abis.basic.erc20, tokens.usdc.address ); let usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect(usdcBalance, `USDC balance equals 0`).to.be.eq( ethers.utils.parseEther("0") ); let cToken = await ethers.getContractAt( abis.basic.erc20, PT_USDC_SPONGSOR_TICKET_ADDR ); const balance = await cToken.balanceOf(dsaWallet0.address); expect(balance, `PoolTogether USDC Ticket is 10`).to.be.eq( ethers.utils.parseUnits("10", 6) ); let poolToken = await ethers.getContractAt( abis.basic.erc20, POOL_TOKEN_ADDRESS ); const poolBalance = await poolToken.balanceOf(dsaWallet0.address); expect(poolBalance, `POOL Token equals 0`).to.be.eq( ethers.utils.parseEther("0") ); // Increase time by 11 days so we get back all USDC without early withdrawal fee await ethers.provider.send("evm_increaseTime", [15 * 24 * 60 * 60]); earlyExitFee = await prizePoolContract.callStatic[ "calculateEarlyExitFee" ](dsaWallet0.address, PT_USDC_SPONGSOR_TICKET_ADDR, amount); expect( earlyExitFee.exitFee, "Exit Fee equal to 0 DAI because past 14 days" ).to.be.eq(0); // Run spell transaction const tx = await dsaWallet0 .connect(wallet0) .cast(...encodeSpells(spells), wallet1.address); const receipt = await tx.wait(); // After spell usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); console.log("USDC BALANCE: ", usdcBalance.toString()); console.log( "USDC BALANCE: ", ethers.utils.parseUnits("10", 6).toString() ); expect( usdcBalance, `USDC balance to be equal to 10, because of no early withdrawal fee` ).to.be.eq(ethers.utils.parseUnits("10", 6)); const balanceAfter = await cToken.balanceOf(dsaWallet0.address); expect(balanceAfter, `PoolTogether USDC Ticket to equal 0`).to.be.eq(0); const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address); console.log("POOL BALANCE AFTER:", poolBalanceAfter.toString()); expect( poolBalanceAfter, `POOL Token Balance to be greater than 0` ).to.be.gt(ethers.utils.parseEther("0")); }); it("Should deposit and withdraw all PrizePool, get back less than 10 USDC", async function() { const amount = ethers.utils.parseUnits("10", 6); // 10 USDC const exitFee = ethers.utils.parseUnits(".1", 6); // 1 USDC is 1% of 100 USDC const spells = [ { connector: ptConnectorName, method: "depositTo", args: [USDC_PRIZE_POOL_ADDR, amount, PT_USDC_TICKET_ADDR, 0, 0], }, { connector: ptConnectorName, method: "withdrawInstantlyFrom", args: [ USDC_PRIZE_POOL_ADDR, amount, PT_USDC_TICKET_ADDR, exitFee, 0, 0, ], }, ]; // Before spell let usdcToken = await ethers.getContractAt( abis.basic.erc20, tokens.usdc.address ); let usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect(usdcBalance, `USDC Balance equals 100`).to.be.eq( ethers.utils.parseUnits("10", 6) ); let cToken = await ethers.getContractAt( abis.basic.erc20, PT_USDC_TICKET_ADDR ); const balance = await cToken.balanceOf(dsaWallet0.address); expect(balance, `PoolTogether USDC Ticket equals 0`).to.be.eq(0); let poolToken = await ethers.getContractAt( abis.basic.erc20, POOL_TOKEN_ADDRESS ); const poolBalance = await poolToken.balanceOf(dsaWallet0.address); expect(poolBalance, `PoolTogether Token greater than 0`).to.be.gt(0); // Run spell transaction const tx = await dsaWallet0 .connect(wallet0) .cast(...encodeSpells(spells), wallet1.address); const receipt = await tx.wait(); // After spell usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect( usdcBalance, `USDC balance to be less than 10, because of early withdrawal fee` ).to.be.lt(ethers.utils.parseUnits("10", 6)); console.log("USDC BALANCE AFTER:", usdcBalance.toString()); const balanceAfter = await cToken.balanceOf(dsaWallet0.address); expect(balanceAfter, `PoolTogether USDC Ticket to equal 0`).to.be.eq(0); const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address); expect(poolBalanceAfter, `POOL Token Balance to greater than 0`).to.be.gt( ethers.utils.parseEther("0") ); }); it("Should deposit, wait 11 days, and withdraw all PrizePool, get 10 USDC, and claim all POOL using claimAll", async function() { const amount = ethers.utils.parseUnits("9.9", 6); // 9 USDC const depositSpells = [ { connector: ptConnectorName, method: "depositTo", args: [ USDC_PRIZE_POOL_ADDR, amount, PT_USDC_SPONGSOR_TICKET_ADDR, 0, 0, ], }, ]; // Before spell let usdcToken = await ethers.getContractAt( abis.basic.erc20, tokens.usdc.address ); let usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect(usdcBalance, `USDC balance less than 10`).to.be.lt( ethers.utils.parseUnits("10", 6) ); let cToken = await ethers.getContractAt( abis.basic.erc20, PT_USDC_SPONGSOR_TICKET_ADDR ); const balance = await cToken.balanceOf(dsaWallet0.address); expect(balance, `PoolTogether USDC Ticket equal 0`).to.be.eq(0); let poolToken = await ethers.getContractAt( abis.basic.erc20, POOL_TOKEN_ADDRESS ); const poolBalance = await poolToken.balanceOf(dsaWallet0.address); expect(poolBalance, `POOL Token is greater than 0`).to.be.gt( ethers.utils.parseEther("0") ); // Run spell transaction const tx = await dsaWallet0 .connect(wallet0) .cast(...encodeSpells(depositSpells), wallet1.address); const receipt = await tx.wait(); const prizePoolContract = new ethers.Contract( USDC_PRIZE_POOL_ADDR, prizePoolABI, ethers.provider ); let earlyExitFee = await prizePoolContract.callStatic[ "calculateEarlyExitFee" ](dsaWallet0.address, PT_USDC_SPONGSOR_TICKET_ADDR, amount); expect( earlyExitFee.exitFee, "Exit Fee equal to 0 USDC because fee 0%" ).to.be.eq(0); // Increase time by 11 days so we get back all DAI without early withdrawal fee await ethers.provider.send("evm_increaseTime", [11 * 24 * 60 * 60]); const withdrawSpells = [ { connector: ptConnectorName, method: "withdrawInstantlyFrom", args: [ USDC_PRIZE_POOL_ADDR, amount, PT_USDC_SPONGSOR_TICKET_ADDR, earlyExitFee.exitFee, 0, 0, ], }, { connector: ptConnectorName, method: "claimAll", args: [TOKEN_FAUCET_PROXY_FACTORY_ADDR, [USDC_POOL_FAUCET_ADDR]], }, ]; // Run spell transaction const tx2 = await dsaWallet0 .connect(wallet0) .cast(...encodeSpells(withdrawSpells), wallet1.address); const receipt2 = await tx2.wait(); // After spell usdcBalance = await usdcToken.balanceOf(dsaWallet0.address); expect(usdcBalance, `USDC balance equals 9.9`).to.be.eq( ethers.utils.parseUnits("9.9", 6) ); const balanceAfter = await cToken.balanceOf(dsaWallet0.address); expect(balanceAfter, `PoolTogether USDC Ticket equal 0`).to.be.eq(0); // Expect const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address); console.log("POOL BALANCE AFTER:", poolBalanceAfter.toString()); expect(poolBalanceAfter, `Pool Token to be greater than before`).to.be.gt( poolBalance ); }); // }) // NO WMATIC POOLS: https://reference-app.pooltogether.com/pools/polygon // describe("Main - WETH Prize Pool Test", function () { // it("Deposit 1 ETH into WETH Prize Pool and withdraw immediately", async function () { // const amount = ethers.utils.parseEther("1") // 1 ETH // const setId = "83478237" // const spells = [ // { // connector: ptConnectorName, // method: "depositTo", // args: [WETH_PRIZE_POOL_ADDR, amount, WETH_POOL_TICKET_ADDR, 0, setId] // }, // { // connector: ptConnectorName, // method: "withdrawInstantlyFrom", // args: [WETH_PRIZE_POOL_ADDR, amount, WETH_POOL_TICKET_ADDR, amount, setId, 0] // }, // ] // // Before Spell // const ethBalanceBefore = await ethers.provider.getBalance(dsaWallet0.address); // // Run spell transaction // const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address) // const receipt = await tx.wait() // // After spell // const ethBalanceAfter = await ethers.provider.getBalance(dsaWallet0.address); // // ETH used for transaction // expect(ethBalanceAfter, `ETH Balance less than before spell because of early withdrawal fee`).to.be.lte(ethBalanceBefore); // }); // it("Deposit 1 ETH into WETH Prize Pool, wait 14 days, then withdraw", async function () { // const amount = ethers.utils.parseEther("1") // 1 ETH // const depositSpell = [ // { // connector: ptConnectorName, // method: "depositTo", // args: [WETH_PRIZE_POOL_ADDR, amount, WETH_POOL_TICKET_ADDR, 0, 0] // } // ] // const withdrawSpell = [ // { // connector: ptConnectorName, // method: "withdrawInstantlyFrom", // args: [WETH_PRIZE_POOL_ADDR, amount, WETH_POOL_TICKET_ADDR, amount, 0, 0] // } // ] // // Before Deposit Spell // let ethBalanceBefore = await ethers.provider.getBalance(dsaWallet0.address); // // Run deposit spell transaction // const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(depositSpell), wallet1.address) // const receipt = await tx.wait() // // After Deposit spell // let ethBalanceAfter = await ethers.provider.getBalance(dsaWallet0.address); // expect(ethBalanceAfter, `ETH Balance less than before spell`).to.be.lte(ethBalanceBefore); // // Increase time by 11 days so we get back all ETH without early withdrawal fee // await ethers.provider.send("evm_increaseTime", [14*24*60*60]); // await ethers.provider.send("evm_mine"); // // Run withdraw spell transaction // const tx2 = await dsaWallet0.connect(wallet0).cast(...encodeSpells(withdrawSpell), wallet1.address) // const receipt2 = await tx.wait() // // After Deposit spell // ethBalanceAfter = await ethers.provider.getBalance(dsaWallet0.address); // expect(ethBalanceAfter, `ETH Balance equal to before spell because no early exit fee`).to.be.eq(ethBalanceBefore); // }); }); });