dsa-connectors/test/arbitrum/dsa-spell/dsa-spell.test.ts

181 lines
6.3 KiB
TypeScript
Raw Normal View History

2022-06-20 02:50:49 +00:00
import hre from "hardhat";
import axios from "axios";
import { expect } from "chai";
const { ethers } = hre; //check
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/arbitrum/addresses";
import { abis } from "../../../scripts/constant/abis";
import { ConnectV2DSASpellArbitrum__factory } from "../../../typechain";
2022-06-20 14:10:49 +00:00
import { Signer, Contract } from "ethers";
import { BigNumber } from "bignumber.js";
2022-06-20 02:50:49 +00:00
describe("DSA Spell", function () {
const connectorName = "dsa-spell-test";
let dsaWallet0: any;
let dsaWallet1: any;
let dsaWallet2: any;
let walletB: any;
let wallet0: any;
let masterSigner: Signer;
let instaConnectorsV2: Contract;
let connector: any;
before(async () => {
await hre.network.provider.request({
method: "hardhat_reset",
params: [
{
forking: {
// @ts-ignore
jsonRpcUrl: hre.config.networks.hardhat.forking.url
}
}
]
});
[wallet0] = await ethers.getSigners();
masterSigner = await getMasterSigner();
instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.core.connectorsV2);
connector = await deployAndEnableConnector({
connectorName,
contractArtifact: ConnectV2DSASpellArbitrum__factory,
signer: masterSigner,
connectors: instaConnectorsV2
});
console.log("\tConnector address", connector.address);
});
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;
walletB = await ethers.getSigner(dsaWallet0.address);
dsaWallet1 = await buildDSAv2(dsaWallet0.address);
expect(!!dsaWallet1.address).to.be.true;
console.log(`\t${dsaWallet1.address}`);
});
it("Deposit eth into DSA wallet 0", 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"));
});
it("Deposit eth into DSA wallet 1", async function () {
await wallet0.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 ETH = "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE";
2022-06-20 06:41:56 +00:00
let USDC = "0xFF970A61A04b1cA14834A43f5dE4533eBDDB5CC8";
let usdc = new ethers.Contract(USDC, abis.basic.erc20);
2022-06-20 14:10:49 +00:00
let aETH = "0xe50fA9b3c56FfB159cB0FCA61F5c9D750e8128c8";
let aEth = new ethers.Contract(aETH, abis.basic.aToken);
2022-06-20 02:50:49 +00:00
var abi = [
"function withdraw(address,uint256,address,uint256,uint256)",
2022-06-20 06:41:56 +00:00
"function deposit(address,uint256,uint256,uint256)",
"function borrow(address,uint256,uint256,uint256,uint256)"
2022-06-20 02:50:49 +00:00
];
function getCallData(spell: string, params: any) {
var iface = new ethers.utils.Interface(abi);
let data = iface.encodeFunctionData(spell, params);
2022-06-20 06:41:56 +00:00
return ethers.utils.hexlify(data);
2022-06-20 02:50:49 +00:00
}
it("should cast spells", async function () {
2022-06-20 06:41:56 +00:00
async function getArg(connectors: any, spells: any, params: any) {
let datas = [];
for (let i = 0; i < connectors.length; i++) {
datas.push(getCallData(spells[i], params[i]));
}
2022-06-20 02:50:49 +00:00
return [dsaWallet1.address, connectors, datas];
}
2022-06-20 06:41:56 +00:00
let connectors = ["BASIC-A", "AAVE-V3-A", "AAVE-V3-A"];
let methods = ["withdraw", "deposit", "borrow"];
let params = [
[ETH, ethers.utils.parseEther("2"), dsaWallet0.address, 0, 0],
[ETH, ethers.constants.MaxUint256, 0, 0],
[USDC, ethers.utils.parseUnits("1", 6), 2, 0, 0]
];
let arg = await getArg(connectors, methods, params);
2022-06-20 02:50:49 +00:00
const spells = [
{
connector: connectorName,
method: "castOnDSA",
args: arg
}
];
const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), await wallet0.getAddress());
const receipt = await tx.wait();
2022-06-20 06:41:56 +00:00
});
2022-06-20 02:50:49 +00:00
2022-06-20 06:41:56 +00:00
it("should check balances after cast on DSA", async function () {
2022-06-20 14:10:49 +00:00
expect(await ethers.provider.getBalance(dsaWallet1.address)).to.be.lte(0);
expect(await usdc.connect(wallet0).balanceOf(dsaWallet1.address)).to.be.gte(
new BigNumber(1).multipliedBy(1e6).toString()
);
expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(
new BigNumber(12).multipliedBy(1e18).toString()
);
2022-06-20 02:50:49 +00:00
});
2022-06-20 14:10:49 +00:00
it("should cast spell on the first successful", async function () {
2022-06-20 06:41:56 +00:00
async function getArg(connectors: any, spells: any, params: any) {
let datas = [];
for (let i = 0; i < connectors.length; i++) {
datas.push(getCallData(spells[i], params[i]));
}
2022-06-20 02:50:49 +00:00
return [connectors, datas];
}
2022-06-20 14:10:49 +00:00
let connectors = ["AAVE-V3-A"];
let methods = ["deposit"];
2022-06-20 06:41:56 +00:00
let params = [
2022-06-20 14:10:49 +00:00
[ETH, ethers.utils.parseEther("10"), 0, 0]
2022-06-20 06:41:56 +00:00
];
let arg = await getArg(connectors, methods, params);
2022-06-20 02:50:49 +00:00
const spells = [
{
connector: connectorName,
2022-06-20 14:10:49 +00:00
method: "castAny",
2022-06-20 02:50:49 +00:00
args: arg
}
];
const tx = await dsaWallet0
.connect(wallet0)
2022-06-20 14:10:49 +00:00
.cast(...encodeSpells(spells), await wallet0.getAddress());
2022-06-20 02:50:49 +00:00
const receipt = await tx.wait();
2022-06-20 06:41:56 +00:00
});
2022-06-20 02:50:49 +00:00
2022-06-20 06:41:56 +00:00
it("should check balances after spells on DSA", async function () {
2022-06-20 14:10:49 +00:00
expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.lte(
new BigNumber(2).multipliedBy(1e18).toString()
);
expect(await aEth.connect(wallet0).balanceOf(dsaWallet0.address)).to.be.gte(
new BigNumber(10).multipliedBy(1e18).toString()
);
2022-06-20 02:50:49 +00:00
});
});
});