mirror of
				https://github.com/Instadapp/dsa-connectors.git
				synced 2024-07-29 22:37:00 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			337 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
			
		
		
	
	
			337 lines
		
	
	
		
			12 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 { addresses } from "../../../scripts/tests/mainnet/addresses";
 | |
| import { abis } from "../../../scripts/constant/abis";
 | |
| import { tokens } from "../../../scripts/tests/mainnet/tokens";
 | |
| import { ConnectV2BMakerDAO__factory } from "../../../typechain";
 | |
| import type { Signer, Contract } from "ethers";
 | |
| 
 | |
| describe("B.Maker", function () {
 | |
|     const connectorName = "B.MAKER-TEST-A"
 | |
| 
 | |
|     let dsaWallet0: any;
 | |
|     let dsaWallet1: any;
 | |
|     let masterSigner: Signer;
 | |
|     let instaConnectorsV2: Contract;
 | |
|     let connector: any;
 | |
|     let manager: any;
 | |
|     let vat: any;
 | |
|     let dai: any;
 | |
| 
 | |
|     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: 12696000,
 | |
|                     },
 | |
|                 },
 | |
|             ],
 | |
|         });
 | |
|         masterSigner = await getMasterSigner()
 | |
|         instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.core.connectorsV2);
 | |
|         connector = await deployAndEnableConnector({
 | |
|             connectorName,
 | |
|             contractArtifact: ConnectV2BMakerDAO__factory,
 | |
|             signer: masterSigner,
 | |
|             connectors: instaConnectorsV2
 | |
|         })
 | |
| 
 | |
|         manager = await ethers.getContractAt("BManagerLike", "0x3f30c2381CD8B917Dd96EB2f1A4F96D91324BBed")
 | |
|         vat = await ethers.getContractAt("../artifacts/contracts/mainnet/connectors/b.protocol/makerdao/interface.sol:VatLike", await manager.vat())
 | |
|         dai = await ethers.getContractAt("../artifacts/contracts/mainnet/common/interfaces.sol:TokenInterface", tokens.dai.address)
 | |
| 
 | |
|         console.log("Connector address", connector.address)
 | |
|     })
 | |
| 
 | |
|     it("test veryClose.", async function () {
 | |
|         expect(veryClose(1000001, 1000000)).to.be.true
 | |
|         expect(veryClose(1000000, 1000001)).to.be.true
 | |
|         expect(veryClose(1003000, 1000001)).to.be.false
 | |
|         expect(veryClose(1000001, 1000300)).to.be.false
 | |
|     });
 | |
| 
 | |
|     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;
 | |
|         expect(await connector.name()).to.be.equal("B.MakerDAO-v1.0");
 | |
|     });
 | |
| 
 | |
|     describe("DSA wallet setup", function () {
 | |
|         it("Should build DSA v2", async function () {
 | |
|             dsaWallet0 = await buildDSAv2(wallet0.address)
 | |
|             expect(!!dsaWallet0.address).to.be.true;
 | |
| 
 | |
|             dsaWallet1 = await buildDSAv2(wallet1.address)
 | |
|             expect(!!dsaWallet1.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"));
 | |
| 
 | |
|             await wallet1.sendTransaction({
 | |
|                 to: dsaWallet1.address,
 | |
|                 value: ethers.utils.parseEther("10")
 | |
|             });
 | |
|             expect(await ethers.provider.getBalance(dsaWallet1.address)).to.be.gte(ethers.utils.parseEther("10"));
 | |
|         });
 | |
|     });
 | |
| 
 | |
|     describe("Main", function () {
 | |
|         let vault: any;
 | |
|         let ilk: any;
 | |
|         let urn: any;
 | |
| 
 | |
|         it("Should open ETH-A vault Maker", async function () {
 | |
|             vault = Number(await manager.cdpi()) + 1
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "open",
 | |
|                     args: ["ETH-A"]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             expect(await manager.owns(vault)).to.be.equal(dsaWallet0.address)
 | |
| 
 | |
|             ilk = await manager.ilks(vault)
 | |
|             expect(ilk).to.be.equal("0x4554482d41000000000000000000000000000000000000000000000000000000")
 | |
| 
 | |
|             urn = await manager.urns(vault)
 | |
|         });
 | |
| 
 | |
|         it("Should deposit", async function () {
 | |
|             const amount = ethers.utils.parseEther("7") // 7 ETH
 | |
|             const setId = "83478237"
 | |
| 
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "deposit",
 | |
|                     args: [vault, amount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("3"))
 | |
| 
 | |
|             const urnData = await vat.urns(ilk, urn)
 | |
|             expect(urnData[0]).to.be.equal(amount) // ink
 | |
|             expect(urnData[1]).to.be.equal("0") // art        
 | |
| 
 | |
|         });
 | |
| 
 | |
|         it("Should withdraw", async function () {
 | |
|             const amount = ethers.utils.parseEther("1") // 1 ETH
 | |
|             const setId = "83478237"
 | |
| 
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "withdraw",
 | |
|                     args: [vault, amount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("4"))
 | |
| 
 | |
|             const urnData = await vat.urns(ilk, urn)
 | |
|             expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
 | |
|             expect(urnData[1]).to.be.equal("0") // art        
 | |
| 
 | |
|         });
 | |
| 
 | |
|         it("Should borrow", async function () {
 | |
|             const amount = ethers.utils.parseEther("6000") // 6000 dai
 | |
|             const setId = "83478237"
 | |
| 
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "borrow",
 | |
|                     args: [vault, amount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             const urnData = await vat.urns(ilk, urn)
 | |
|             expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
 | |
|             expect(urnData[1]).to.be.equal(await daiToArt(vat, ilk, amount)) // art
 | |
| 
 | |
|             expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(amount)
 | |
|         });
 | |
| 
 | |
|         it("Should repay", async function () {
 | |
|             const amount = ethers.utils.parseEther("500") // 500 dai
 | |
|             const setId = "83478237"
 | |
| 
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "payback",
 | |
|                     args: [vault, amount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             const urnData = await vat.urns(ilk, urn)
 | |
|             expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
 | |
|             expect(urnData[1]).to.be.equal(await daiToArt(vat, ilk, ethers.utils.parseEther("5500"))) // art        
 | |
|             expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(ethers.utils.parseEther("5500"))
 | |
|         });
 | |
| 
 | |
|         it("Should depositAndBorrow", async function () {
 | |
|             const borrowAmount = ethers.utils.parseEther("1000") // 1000 dai
 | |
|             const depositAmount = ethers.utils.parseEther("1") // 1 dai
 | |
| 
 | |
|             const setId = "83478237"
 | |
| 
 | |
|             const spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "depositAndBorrow",
 | |
|                     args: [vault, depositAmount, borrowAmount, 0, 0, 0, 0]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
 | |
|             const receipt = await tx.wait()
 | |
| 
 | |
|             const urnData = await vat.urns(ilk, urn)
 | |
|             expect(urnData[0]).to.be.equal(ethers.utils.parseEther("7")) // ink
 | |
|             expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(ethers.utils.parseEther("6500"))
 | |
|             // calculation is not precise as the jug was dripped
 | |
|             expect(veryClose(urnData[1], await daiToArt(vat, ilk, ethers.utils.parseEther("6500")))).to.be.true
 | |
|             expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("1"))
 | |
|         });
 | |
| 
 | |
|         it("Should close", async function () {
 | |
|             // open a new vault
 | |
|             const newVault = vault + 1
 | |
|             let spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "open",
 | |
|                     args: ["ETH-A"]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             let tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
 | |
|             let receipt = await tx.wait()
 | |
| 
 | |
|             expect(await manager.owns(newVault)).to.be.equal(dsaWallet1.address)
 | |
| 
 | |
|             ilk = await manager.ilks(newVault)
 | |
|             expect(ilk).to.be.equal("0x4554482d41000000000000000000000000000000000000000000000000000000")
 | |
| 
 | |
|             urn = await manager.urns(newVault)
 | |
| 
 | |
|             // deposit and borrow
 | |
|             const borrowAmount = ethers.utils.parseEther("6000") // 6000 dai
 | |
|             const depositAmount = ethers.utils.parseEther("5") // 5 ETH
 | |
| 
 | |
|             spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "depositAndBorrow",
 | |
|                     args: [newVault, depositAmount, borrowAmount, 0, 0, 0, 0]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
 | |
|             receipt = await tx.wait()
 | |
| 
 | |
|             const setId = 0
 | |
| 
 | |
|             // repay borrow
 | |
|             spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "payback",
 | |
|                     args: [newVault, borrowAmount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
 | |
|             receipt = await tx.wait()
 | |
| 
 | |
|             // withdraw deposit
 | |
|             spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "withdraw",
 | |
|                     args: [newVault, depositAmount, 0, setId]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
 | |
|             receipt = await tx.wait()
 | |
| 
 | |
|             // close
 | |
|             spells = [
 | |
|                 {
 | |
|                     connector: connectorName,
 | |
|                     method: "close",
 | |
|                     args: [newVault]
 | |
|                 }
 | |
|             ]
 | |
| 
 | |
|             tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
 | |
|             receipt = await tx.wait()
 | |
| 
 | |
|             expect(await manager.owns(newVault)).not.to.be.equal(dsaWallet1.address)
 | |
|         });
 | |
|     })
 | |
| })
 | |
| 
 | |
| async function daiToArt(vat: any, ilk: any, dai: any) {
 | |
|     const ilks = await vat.ilks(ilk)
 | |
|     const rate = ilks[1] // second parameter
 | |
|     const _1e27 = ethers.utils.parseEther("1000000000") // 1e9 * 1e18
 | |
|     const art = dai.mul(_1e27).div(rate)
 | |
| 
 | |
|     return art.add(1)
 | |
| }
 | |
| 
 | |
| function veryClose(n1: any, n2: any) {
 | |
|     n1 = web3.utils.toBN(n1)
 | |
|     n2 = web3.utils.toBN(n2)
 | |
| 
 | |
|     let _10000 = web3.utils.toBN(10000)
 | |
|     let _9999 = web3.utils.toBN(9999)
 | |
| 
 | |
|     if (n1.mul(_10000).lt(n2.mul(_9999))) return false
 | |
|     if (n2.mul(_10000).lt(n1.mul(_9999))) return false
 | |
| 
 | |
|     return true
 | |
| }
 | 
