mirror of
https://github.com/Instadapp/dsa-connectors.git
synced 2024-07-29 22:37:00 +00:00
chore: add standadized functions
This commit is contained in:
Dimitri
parent
14008b3bf9
commit
ceb90406cd
|
|
@ -16,6 +16,8 @@ import type { Signer, Contract, BigNumber } from "ethers";
|
|||
|
||||
import { ConnectV2mStable__factory, IERC20Minimal__factory, IERC20Minimal } from "../../../typechain";
|
||||
|
||||
import { executeAndAssertDeposit, executeAndAssertSwap, executeAndAssertWithdraw } from "./mstable.utils";
|
||||
|
||||
import {
|
||||
fundWallet,
|
||||
getToken,
|
||||
|
|
@ -23,12 +25,13 @@ import {
|
|||
DEAD_ADDRESS,
|
||||
calcMinOut,
|
||||
ONE_DAY,
|
||||
increaseTime
|
||||
increaseTime,
|
||||
DEFAULT_DECIMALS,
|
||||
connectorName,
|
||||
toEther
|
||||
} from "./mstable.helpers";
|
||||
|
||||
describe("MStable", async () => {
|
||||
const connectorName = "MStable";
|
||||
|
||||
let dsaWallet0: Contract;
|
||||
let masterSigner: Signer;
|
||||
let instaConnectorsV2: Contract;
|
||||
|
|
@ -36,17 +39,20 @@ describe("MStable", async () => {
|
|||
|
||||
let mtaToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("MTA").tokenAddress, provider);
|
||||
let mUsdToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("mUSD").tokenAddress, provider);
|
||||
let daiToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("DAI").tokenAddress, provider);
|
||||
let fraxToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("FRAX").tokenAddress, provider);
|
||||
let imUsdToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("imUSD").tokenAddress, provider);
|
||||
let imUsdVault: IERC20Minimal = IERC20Minimal__factory.connect(getToken("imUSDVault").tokenAddress, provider);
|
||||
|
||||
let daiToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("DAI").tokenAddress, provider);
|
||||
let usdcToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("USDC").tokenAddress, provider);
|
||||
let fraxToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("FRAX").tokenAddress, provider);
|
||||
|
||||
const wallets = provider.getWallets();
|
||||
const [wallet0, wallet1, wallet2, wallet3] = wallets;
|
||||
|
||||
const toEther = (amount: BigNumber) => ethers.utils.formatEther(amount);
|
||||
describe("DSA wallet", async () => {
|
||||
const fundAmount = simpleToExactAmount(10000);
|
||||
|
||||
before(async () => {
|
||||
const setup = async () => {
|
||||
await hre.network.provider.request({
|
||||
method: "hardhat_reset",
|
||||
params: [
|
||||
|
|
@ -70,159 +76,71 @@ describe("MStable", async () => {
|
|||
});
|
||||
|
||||
console.log("Connector address", connector.address);
|
||||
});
|
||||
it("should deploy", async () => {
|
||||
expect(instaConnectorsV2.address).to.be.properAddress;
|
||||
expect(connector.address).to.be.properAddress;
|
||||
expect(await masterSigner.getAddress()).to.be.properAddress;
|
||||
});
|
||||
describe("DSA wallet", async () => {
|
||||
it("Should build DSA v2", async () => {
|
||||
|
||||
dsaWallet0 = await buildDSAv2(wallet0.address);
|
||||
expect(dsaWallet0.address).to.be.properAddress;
|
||||
});
|
||||
it("Deposit ETH and tokens into DSA Wallet", async () => {
|
||||
|
||||
await wallet0.sendTransaction({
|
||||
to: dsaWallet0.address,
|
||||
value: simpleToExactAmount(10)
|
||||
});
|
||||
|
||||
const fundAmount = simpleToExactAmount(10000);
|
||||
|
||||
expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("10"));
|
||||
|
||||
await fundWallet("mUSD", fundAmount, dsaWallet0.address);
|
||||
await fundWallet("DAI", fundAmount, dsaWallet0.address);
|
||||
await fundWallet("FRAX", fundAmount, dsaWallet0.address);
|
||||
};
|
||||
|
||||
describe("Deploy", async () => {
|
||||
before(async () => {
|
||||
await setup();
|
||||
});
|
||||
|
||||
it("Should deploy properly", async () => {
|
||||
expect(instaConnectorsV2.address).to.be.properAddress;
|
||||
expect(connector.address).to.be.properAddress;
|
||||
expect(await masterSigner.getAddress()).to.be.properAddress;
|
||||
|
||||
expect(dsaWallet0.address).to.be.properAddress;
|
||||
});
|
||||
it("Should fund the wallet", async () => {
|
||||
expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("10"));
|
||||
|
||||
expect(await mUsdToken.balanceOf(dsaWallet0.address)).to.be.gte(fundAmount);
|
||||
expect(await daiToken.balanceOf(dsaWallet0.address)).to.be.gte(fundAmount);
|
||||
expect(await fraxToken.balanceOf(dsaWallet0.address)).to.be.gte(fundAmount);
|
||||
|
||||
});
|
||||
it("Should not have vault tokens prior", async () => {
|
||||
// No deposits prior
|
||||
expect(await imUsdToken.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
expect(await imUsdVault.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
});
|
||||
});
|
||||
|
||||
describe("Main", async () => {
|
||||
describe("Main SAVE", async () => {
|
||||
before(async () => {
|
||||
await setup();
|
||||
});
|
||||
it("Should deposit mUSD to Vault successfully", async () => {
|
||||
const depositAmount = simpleToExactAmount(100);
|
||||
|
||||
const mUsdBalanceBefore = await mUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("mUSD balance before: ", toEther(mUsdBalanceBefore));
|
||||
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method: "deposit",
|
||||
args: [mUsdToken.address, depositAmount]
|
||||
}
|
||||
];
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const mUsdBalanceAfter = await mUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("mUSD balance after: ", toEther(mUsdBalanceAfter));
|
||||
|
||||
const imUsdBalance = await imUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD balance: ", toEther(imUsdBalance));
|
||||
|
||||
const imUsdVaultBalance = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance: ", toEther(imUsdVaultBalance));
|
||||
|
||||
// Should have something in the vault but no imUSD
|
||||
expect(await imUsdToken.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
expect(await imUsdVault.balanceOf(dsaWallet0.address)).to.be.gt(0);
|
||||
expect(mUsdBalanceAfter).to.eq(mUsdBalanceBefore.sub(depositAmount));
|
||||
await executeAndAssertDeposit("deposit", mUsdToken, depositAmount, dsaWallet0, wallet0);
|
||||
});
|
||||
it("Should deposit DAI to Vault successfully (mUSD bAsset)", async () => {
|
||||
const depositAmount = simpleToExactAmount(100);
|
||||
const minOut = calcMinOut(depositAmount, 0.02);
|
||||
|
||||
const daiBalanceBefore = await daiToken.balanceOf(dsaWallet0.address);
|
||||
console.log("DAI balance before: ", toEther(daiBalanceBefore));
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method: "depositViaMint",
|
||||
args: [daiToken.address, depositAmount, minOut]
|
||||
}
|
||||
];
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const daiBalanceAfter = await daiToken.balanceOf(dsaWallet0.address);
|
||||
console.log("DAI balance after: ", toEther(daiBalanceAfter));
|
||||
|
||||
const imUsdVaultBalanceAfter = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance after: ", toEther(imUsdVaultBalanceAfter));
|
||||
|
||||
expect(imUsdVaultBalanceAfter).to.be.gt(imUsdVaultBalanceBefore);
|
||||
expect(await imUsdToken.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
expect(daiBalanceAfter).to.eq(daiBalanceBefore.sub(depositAmount));
|
||||
await executeAndAssertDeposit("depositViaMint", daiToken, depositAmount, dsaWallet0, wallet0, [minOut]);
|
||||
});
|
||||
it("Should deposit FRAX to Vault successfully (via Feeder Pool)", async () => {
|
||||
const depositAmount = simpleToExactAmount(100);
|
||||
const minOut = calcMinOut(depositAmount, 0.02);
|
||||
const path = getToken("FRAX").feederPool;
|
||||
|
||||
const fraxBalanceBefore = await fraxToken.balanceOf(dsaWallet0.address);
|
||||
console.log("FRAX balance before: ", toEther(fraxBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method: "depositViaSwap",
|
||||
args: [fraxToken.address, depositAmount, minOut, getToken("FRAX").feederPool]
|
||||
}
|
||||
];
|
||||
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const fraxBalanceAfter = await fraxToken.balanceOf(dsaWallet0.address);
|
||||
console.log("FRAX balance after: ", toEther(fraxBalanceAfter));
|
||||
|
||||
const imUsdVaultBalanceAfter = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance after: ", toEther(imUsdVaultBalanceAfter));
|
||||
|
||||
expect(imUsdVaultBalanceAfter).to.be.gt(imUsdVaultBalanceBefore);
|
||||
expect(await imUsdToken.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
expect(fraxBalanceAfter).to.eq(fraxBalanceBefore.sub(depositAmount));
|
||||
await executeAndAssertDeposit("depositViaSwap", fraxToken, depositAmount, dsaWallet0, wallet0, [minOut, path]);
|
||||
});
|
||||
it("Should withdraw from Vault to mUSD", async () => {
|
||||
const withdrawAmount = simpleToExactAmount(100);
|
||||
|
||||
const mUsdBalanceBefore = await mUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("mUSD balance before: ", toEther(mUsdBalanceBefore));
|
||||
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method: "withdraw",
|
||||
args: [withdrawAmount]
|
||||
}
|
||||
];
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const imUsdVaultBalanceAfter = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance after: ", toEther(imUsdVaultBalanceAfter));
|
||||
|
||||
const mUsdBalanceAfter = await mUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("mUSD balance after: ", toEther(mUsdBalanceAfter));
|
||||
|
||||
expect(imUsdVaultBalanceAfter).to.be.eq(imUsdVaultBalanceBefore.sub(withdrawAmount));
|
||||
expect(mUsdBalanceAfter).to.gt(mUsdBalanceBefore);
|
||||
await executeAndAssertWithdraw("withdraw", mUsdToken, withdrawAmount, dsaWallet0, wallet0, [withdrawAmount]);
|
||||
});
|
||||
it("Should withdraw from Vault to DAI (mUSD bAsset)", async () => {
|
||||
const withdrawAmount = simpleToExactAmount(100);
|
||||
|
|
@ -305,5 +223,50 @@ describe("MStable", async () => {
|
|||
expect(mtaBalanceAfter).to.be.gt(mtaBalanceBefore);
|
||||
});
|
||||
});
|
||||
describe("Main SWAP", async () => {
|
||||
before(async () => {
|
||||
await setup();
|
||||
});
|
||||
it("Should swap mUSD to bAsset (redeem)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
await executeAndAssertSwap("swap", mUsdToken, 18, daiToken, 18, swapAmount, dsaWallet0, wallet0);
|
||||
});
|
||||
it("Should swap mUSD to fAsset (via feeder pool)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
const path = getToken("FRAX").feederPool;
|
||||
await executeAndAssertSwap("swapViaFeeder", mUsdToken, 18, fraxToken, 18, swapAmount, dsaWallet0, wallet0, [
|
||||
path
|
||||
]);
|
||||
});
|
||||
it("Should swap bAsset to mUSD (mint)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
await executeAndAssertSwap("swap", daiToken, 18, mUsdToken, 18, swapAmount, dsaWallet0, wallet0);
|
||||
});
|
||||
it("Should swap bAsset to bAsset (swap)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
await executeAndAssertSwap("swap", daiToken, 18, usdcToken, 6, swapAmount, dsaWallet0, wallet0);
|
||||
});
|
||||
it("Should swap bAsset to fAsset (via feeder)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
const path = getToken("FRAX").feederPool;
|
||||
await executeAndAssertSwap("swapViaFeeder", daiToken, 18, fraxToken, 18, swapAmount, dsaWallet0, wallet0, [
|
||||
path
|
||||
]);
|
||||
});
|
||||
it("Should swap fAsset to bAsset (via feeder)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
const path = getToken("FRAX").feederPool;
|
||||
await executeAndAssertSwap("swapViaFeeder", fraxToken, 18, daiToken, 18, swapAmount, dsaWallet0, wallet0, [
|
||||
path
|
||||
]);
|
||||
});
|
||||
it("Should swap fAsset to mUSD (via feeder)", async () => {
|
||||
const swapAmount = simpleToExactAmount(100);
|
||||
const path = getToken("FRAX").feederPool;
|
||||
await executeAndAssertSwap("swapViaFeeder", fraxToken, 18, mUsdToken, 18, swapAmount, dsaWallet0, wallet0, [
|
||||
path
|
||||
]);
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
|
|
|
|||
137
test/polygon/mstable/mstable.utils.ts
Normal file
137
test/polygon/mstable/mstable.utils.ts
Normal file
|
|
@ -0,0 +1,137 @@
|
|||
import hre from "hardhat";
|
||||
import { ethers } from "hardhat";
|
||||
import { assert, expect } from "chai";
|
||||
|
||||
import {
|
||||
DEFAULT_DECIMALS,
|
||||
DEAD_ADDRESS,
|
||||
toEther,
|
||||
connectorName,
|
||||
simpleToExactAmount,
|
||||
getToken
|
||||
} from "./mstable.helpers";
|
||||
|
||||
import { IERC20Minimal, IERC20Minimal__factory } from "../../../typechain";
|
||||
import { BigNumber, Contract, Wallet } from "ethers";
|
||||
|
||||
import { encodeSpells } from "../../../scripts/tests/encodeSpells";
|
||||
|
||||
const provider = hre.waffle.provider;
|
||||
|
||||
let imUsdToken: IERC20Minimal = IERC20Minimal__factory.connect(getToken("imUSD").tokenAddress, provider);
|
||||
let imUsdVault: IERC20Minimal = IERC20Minimal__factory.connect(getToken("imUSDVault").tokenAddress, provider);
|
||||
|
||||
export const executeAndAssertSwap = async (
|
||||
method: string,
|
||||
tokenFrom: IERC20Minimal,
|
||||
tokenFromDecimals: number,
|
||||
tokenTo: IERC20Minimal,
|
||||
tokenToDecimals: number,
|
||||
swapAmount: BigNumber,
|
||||
dsaWallet0: Contract,
|
||||
wallet0: Wallet,
|
||||
args?: any[]
|
||||
) => {
|
||||
const diffFrom = ethers.BigNumber.from(10).pow(DEFAULT_DECIMALS - tokenFromDecimals);
|
||||
const diffTo = ethers.BigNumber.from(10).pow(DEFAULT_DECIMALS - tokenToDecimals);
|
||||
|
||||
const tokenFromBalanceBefore = (await tokenFrom.balanceOf(dsaWallet0.address)).mul(diffFrom);
|
||||
console.log("Token From balance before: ", toEther(tokenFromBalanceBefore));
|
||||
|
||||
const tokenToBalanceBefore = (await tokenTo.balanceOf(dsaWallet0.address)).mul(diffTo);
|
||||
console.log("Token To balance before: ", toEther(tokenToBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method,
|
||||
args: [tokenFrom.address, tokenTo.address, swapAmount, 1, ...(args ? args : [])]
|
||||
}
|
||||
];
|
||||
|
||||
console.log("Swapping...", toEther(swapAmount));
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const tokenFromBalanceAfter = (await tokenFrom.balanceOf(dsaWallet0.address)).mul(diffFrom);
|
||||
console.log("Token From balance after: ", toEther(tokenFromBalanceAfter));
|
||||
|
||||
const tokenToBalanceAfter = (await tokenTo.balanceOf(dsaWallet0.address)).mul(diffTo);
|
||||
console.log("Token To balance after: ", toEther(tokenToBalanceAfter));
|
||||
|
||||
expect(tokenFromBalanceAfter).to.be.eq(tokenFromBalanceBefore.sub(swapAmount));
|
||||
expect(tokenToBalanceAfter).to.be.gt(tokenToBalanceBefore);
|
||||
};
|
||||
|
||||
export const executeAndAssertDeposit = async (
|
||||
method: string,
|
||||
tokenFrom: IERC20Minimal,
|
||||
depositAmount: BigNumber,
|
||||
dsaWallet0: Contract,
|
||||
wallet0: Wallet,
|
||||
args?: any[]
|
||||
) => {
|
||||
const FromBalanceBefore = await tokenFrom.balanceOf(dsaWallet0.address);
|
||||
console.log("Balance before: ", toEther(FromBalanceBefore));
|
||||
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method,
|
||||
args: [tokenFrom.address, depositAmount, ...(args ? args : [])]
|
||||
}
|
||||
];
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const FromBalanceAfter = await tokenFrom.balanceOf(dsaWallet0.address);
|
||||
console.log("Balance after: ", toEther(FromBalanceAfter));
|
||||
|
||||
const imUsdBalance = await imUsdToken.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD balance: ", toEther(imUsdBalance));
|
||||
|
||||
const imUsdVaultBalance = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance: ", toEther(imUsdVaultBalance));
|
||||
|
||||
// Should have something in the vault but no imUSD
|
||||
expect(await imUsdToken.balanceOf(dsaWallet0.address)).to.be.eq(0);
|
||||
expect(await imUsdVault.balanceOf(dsaWallet0.address)).to.be.gt(imUsdVaultBalanceBefore);
|
||||
expect(FromBalanceAfter).to.eq(FromBalanceBefore.sub(depositAmount));
|
||||
};
|
||||
|
||||
export const executeAndAssertWithdraw = async (
|
||||
method: string,
|
||||
tokenFrom: IERC20Minimal,
|
||||
withdrawAmount: BigNumber,
|
||||
dsaWallet0: Contract,
|
||||
wallet0: Wallet,
|
||||
args: any[]
|
||||
) => {
|
||||
const mUsdBalanceBefore = await tokenFrom.balanceOf(dsaWallet0.address);
|
||||
console.log("Balance before: ", toEther(mUsdBalanceBefore));
|
||||
|
||||
const imUsdVaultBalanceBefore = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance before: ", toEther(imUsdVaultBalanceBefore));
|
||||
|
||||
const spells = [
|
||||
{
|
||||
connector: connectorName,
|
||||
method,
|
||||
args
|
||||
}
|
||||
];
|
||||
|
||||
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), DEAD_ADDRESS);
|
||||
|
||||
const imUsdVaultBalanceAfter = await imUsdVault.balanceOf(dsaWallet0.address);
|
||||
console.log("imUSD Vault balance after: ", toEther(imUsdVaultBalanceAfter));
|
||||
|
||||
const mUsdBalanceAfter = await tokenFrom.balanceOf(dsaWallet0.address);
|
||||
console.log("Balance after: ", toEther(mUsdBalanceAfter));
|
||||
|
||||
expect(imUsdVaultBalanceAfter).to.be.eq(imUsdVaultBalanceBefore.sub(withdrawAmount));
|
||||
expect(mUsdBalanceAfter).to.gt(mUsdBalanceBefore);
|
||||
};
|
||||
Loading…
Reference in New Issue
Block a user