163 lines
7.1 KiB
TypeScript
163 lines
7.1 KiB
TypeScript
/*
|
|
|
|
Copyright 2020 DODO ZOO.
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
*/
|
|
|
|
import { assert } from 'chai';
|
|
import { DODOContext, getDODOContext } from '../utils-v1/Context-route';
|
|
import { ProxyContext, getProxyContext } from '../utils/ProxyContext';
|
|
import { decimalStr, MAX_UINT256, fromWei, mweiStr } from '../utils-v1/Converter';
|
|
import { logGas } from '../utils-v1/Log';
|
|
import * as contracts from '../utils-v1/Contracts';
|
|
import { Contract } from 'web3-eth-contract';
|
|
|
|
let lp: string;
|
|
let trader: string;
|
|
|
|
async function initDODO_USDT(ctx: DODOContext): Promise<void> {
|
|
await ctx.setOraclePrice(ctx.DODO_USDT_ORACLE, mweiStr("0.1"));
|
|
lp = ctx.spareAccounts[0];
|
|
trader = ctx.spareAccounts[1];
|
|
|
|
let DODO = ctx.DODO;
|
|
let USDT = ctx.USDT;
|
|
let DODO_USDT = ctx.DODO_USDT;
|
|
await ctx.approvePair(DODO, USDT, DODO_USDT.options.address, lp);
|
|
await ctx.approvePair(DODO, USDT, DODO_USDT.options.address, trader);
|
|
|
|
await ctx.mintToken(DODO, USDT, lp, decimalStr("10000000"), mweiStr("1000000"));
|
|
await ctx.mintToken(DODO, USDT, trader, decimalStr("1000"), mweiStr("1000"));
|
|
|
|
await DODO_USDT.methods
|
|
.depositBaseTo(lp, decimalStr("10000000"))
|
|
.send(ctx.sendParam(lp));
|
|
await DODO_USDT.methods
|
|
.depositQuoteTo(lp, mweiStr("1000000"))
|
|
.send(ctx.sendParam(lp));
|
|
}
|
|
|
|
async function initWETH_USDC(ctx: DODOContext): Promise<void> {
|
|
await ctx.setOraclePrice(ctx.WETH_USDC_ORACLE, mweiStr("450"));
|
|
lp = ctx.spareAccounts[0];
|
|
trader = ctx.spareAccounts[1];
|
|
|
|
let WETH = ctx.WETH;
|
|
let USDC = ctx.USDC;
|
|
let WETH_USDC = ctx.WETH_USDC;
|
|
|
|
await ctx.approvePair(WETH, USDC, WETH_USDC.options.address, lp);
|
|
await ctx.mintToken(null, USDC, lp, decimalStr("0"), mweiStr("3600"));
|
|
await ctx.mintToken(null, USDC, trader, decimalStr("0"), mweiStr("100"));
|
|
await WETH.methods.deposit().send(ctx.sendParam(lp, '8'));
|
|
|
|
await WETH_USDC.methods
|
|
.depositBaseTo(lp, decimalStr("8"))
|
|
.send(ctx.sendParam(lp));
|
|
await WETH_USDC.methods
|
|
.depositQuoteTo(lp, mweiStr("3600"))
|
|
.send(ctx.sendParam(lp));
|
|
}
|
|
|
|
describe("AddLiquidity", () => {
|
|
let snapshotId1: string;
|
|
let snapshotId2: string;
|
|
let ctxV1: DODOContext;
|
|
let ctxV2: ProxyContext;
|
|
let DODO_LP: Contract;
|
|
let WETH_LP: Contract;
|
|
let USDT_LP: Contract;
|
|
let USDC_LP: Contract;
|
|
|
|
before(async () => {
|
|
let ETH = await contracts.newContract(
|
|
contracts.WETH_CONTRACT_NAME
|
|
);
|
|
ctxV1 = await getDODOContext(ETH.options.address);
|
|
ctxV2 = await getProxyContext(ETH.options.address);
|
|
await initDODO_USDT(ctxV1);
|
|
await initWETH_USDC(ctxV1);
|
|
var dodo_dlp = await ctxV1.DODO_USDT.methods._BASE_CAPITAL_TOKEN_().call();
|
|
var usdt_dlp = await ctxV1.DODO_USDT.methods._QUOTE_CAPITAL_TOKEN_().call();
|
|
DODO_LP = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, dodo_dlp);
|
|
USDT_LP = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, usdt_dlp);
|
|
|
|
var weth_dlp = await ctxV1.WETH_USDC.methods._BASE_CAPITAL_TOKEN_().call();
|
|
var usdc_dlp = await ctxV1.WETH_USDC.methods._QUOTE_CAPITAL_TOKEN_().call();
|
|
WETH_LP = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, weth_dlp);
|
|
USDC_LP = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, usdc_dlp);
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
snapshotId1 = await ctxV1.EVM.snapshot();
|
|
snapshotId2 = await ctxV2.EVM.snapshot();
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await ctxV1.EVM.reset(snapshotId1);
|
|
await ctxV2.EVM.reset(snapshotId2);
|
|
});
|
|
|
|
describe("AddLiquidity", () => {
|
|
it("AddLiquidity", async () => {
|
|
var b_DODO = await ctxV1.DODO.methods.balanceOf(trader).call()
|
|
var b_USDT = await ctxV1.USDT.methods.balanceOf(trader).call()
|
|
var dodo_lp = await DODO_LP.methods.balanceOf(trader).call()
|
|
var usdt_lp = await USDT_LP.methods.balanceOf(trader).call()
|
|
console.log("Before DODO:" + fromWei(b_DODO, 'ether') + "; USDT:" + fromWei(b_USDT, 'mwei'));
|
|
console.log("dodo_lp:" + dodo_lp + " usdt_lp:" + usdt_lp);
|
|
await ctxV1.DODO.methods.approve(ctxV2.SmartApprove.options.address, MAX_UINT256).send(ctxV2.sendParam(trader));
|
|
await ctxV1.USDT.methods.approve(ctxV2.SmartApprove.options.address, MAX_UINT256).send(ctxV2.sendParam(trader));
|
|
var tx = await logGas(await ctxV2.DODOProxy.methods.addLiquidityToV1(
|
|
trader,
|
|
ctxV1.DODO_USDT.options.address,
|
|
decimalStr("100"),
|
|
mweiStr("100"),
|
|
0,
|
|
0,
|
|
0,
|
|
Math.floor(new Date().getTime() / 1000 + 60 * 10)
|
|
), ctxV2.sendParam(trader), "addLiquidity");
|
|
var dodo_lp = await DODO_LP.methods.balanceOf(trader).call()
|
|
var usdt_lp = await USDT_LP.methods.balanceOf(trader).call()
|
|
var a_DODO = await ctxV1.DODO.methods.balanceOf(trader).call()
|
|
var a_USDT = await ctxV1.USDT.methods.balanceOf(trader).call()
|
|
console.log("After DODO:" + fromWei(a_DODO, 'ether') + "; USDT:" + fromWei(a_USDT, 'mwei'));
|
|
assert.equal(dodo_lp,decimalStr("100"));
|
|
assert.equal(usdt_lp,mweiStr("100"));
|
|
assert.equal(a_DODO,decimalStr("900"));
|
|
assert.equal(a_USDT,mweiStr("900"));
|
|
});
|
|
|
|
it("AddLiquidity - ETH", async () => {
|
|
var b_WETH = await ctxV1.WETH.methods.balanceOf(trader).call()
|
|
var b_USDC = await ctxV1.USDC.methods.balanceOf(trader).call()
|
|
var b_ETH = await ctxV2.Web3.eth.getBalance(trader);
|
|
var weth_lp = await WETH_LP.methods.balanceOf(trader).call()
|
|
var usdc_lp = await USDC_LP.methods.balanceOf(trader).call()
|
|
console.log("Before WETH:" + fromWei(b_WETH, 'ether') + "; USDC:" + fromWei(b_USDC, 'mwei') + "; ETH:" + fromWei(b_ETH, 'ether'));
|
|
console.log("weth_lp:" + weth_lp + " usdc_lp:" + usdc_lp);
|
|
await ctxV1.USDC.methods.approve(ctxV2.SmartApprove.options.address, MAX_UINT256).send(ctxV2.sendParam(trader));
|
|
var tx = await logGas(await ctxV2.DODOProxy.methods.addLiquidityToV1(
|
|
trader,
|
|
ctxV1.WETH_USDC.options.address,
|
|
decimalStr("1"),
|
|
mweiStr("100"),
|
|
0,
|
|
0,
|
|
1,
|
|
Math.floor(new Date().getTime() / 1000 + 60 * 10)
|
|
), ctxV2.sendParam(trader,"1"), "addLiquidity - eth");
|
|
var weth_lp = await WETH_LP.methods.balanceOf(trader).call()
|
|
var usdc_lp = await USDC_LP.methods.balanceOf(trader).call()
|
|
var a_WETH = await ctxV1.WETH.methods.balanceOf(trader).call()
|
|
var a_USDC = await ctxV1.USDC.methods.balanceOf(trader).call()
|
|
var a_ETH = await ctxV2.Web3.eth.getBalance(trader);
|
|
console.log("After WETH:" + fromWei(a_WETH, 'ether') + "; USDC:" + fromWei(a_USDC, 'mwei') + "; ETH:" + fromWei(a_ETH, 'ether'));
|
|
assert.equal(weth_lp,decimalStr("1"));
|
|
assert.equal(usdc_lp,mweiStr("100"));
|
|
});
|
|
});
|
|
});
|