274 lines
9.7 KiB
TypeScript
274 lines
9.7 KiB
TypeScript
/*
|
|
|
|
Copyright 2020 DODO ZOO.
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
*/
|
|
|
|
import * as assert from 'assert';
|
|
import BigNumber from 'bignumber.js';
|
|
import { DODOContext, getDODOContext } from '../utils-v1/Context-route';
|
|
import { decimalStr,MAX_UINT256,fromWei,mweiStr} from '../utils-v1/Converter';
|
|
import { logGas } from '../utils-v1/Log';
|
|
import { DODOHelper } from '../utils-v1/dodoHelper';
|
|
|
|
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("10000"), mweiStr("1000"));
|
|
await ctx.mintToken(DODO,USDT,trader, decimalStr("100"), mweiStr("0"));
|
|
|
|
await DODO_USDT.methods
|
|
.depositBaseTo(lp, decimalStr("10000"))
|
|
.send(ctx.sendParam(lp));
|
|
await DODO_USDT.methods
|
|
.depositQuoteTo(lp, mweiStr("1000"))
|
|
.send(ctx.sendParam(lp));
|
|
}
|
|
|
|
async function initUSDT_USDC(ctx: DODOContext): Promise<void> {
|
|
await ctx.setOraclePrice(ctx.USDT_USDC_ORACLE,decimalStr("1"));
|
|
lp = ctx.spareAccounts[0];
|
|
trader = ctx.spareAccounts[1];
|
|
|
|
let USDT = ctx.USDT;
|
|
let USDC = ctx.USDC;
|
|
let USDT_USDC = ctx.USDT_USDC;
|
|
|
|
await ctx.approvePair(USDT,USDC,USDT_USDC.options.address,lp);
|
|
await ctx.mintToken(USDT,USDC,lp, mweiStr("1000"), mweiStr("1000"));
|
|
|
|
await USDT_USDC.methods
|
|
.depositBaseTo(lp, mweiStr("1000"))
|
|
.send(ctx.sendParam(lp));
|
|
await USDT_USDC.methods
|
|
.depositQuoteTo(lp, mweiStr("1000"))
|
|
.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(WETH,USDC,lp, decimalStr("1000"), mweiStr("450000"));
|
|
|
|
await WETH_USDC.methods
|
|
.depositBaseTo(lp, decimalStr("1000"))
|
|
.send(ctx.sendParam(lp));
|
|
await WETH_USDC.methods
|
|
.depositQuoteTo(lp, mweiStr("450000"))
|
|
.send(ctx.sendParam(lp));
|
|
}
|
|
|
|
//mock sdk logic
|
|
async function calcRoute(ctx: DODOContext,fromTokenAmount:string,slippage:number,routes:any[],pairs:any[]) {
|
|
let swapAmount = fromTokenAmount
|
|
let directions:number[] = []
|
|
let dodoPairs:string[] = []
|
|
|
|
|
|
for (let i = 0; i < pairs.length; i++) {
|
|
let curPair = pairs[i]
|
|
dodoPairs.push(curPair.pair)
|
|
let curContact = pairs[i].pairContract
|
|
if (curPair.base === routes[i].address) {
|
|
directions[i] = 0;
|
|
swapAmount = await curContact.methods.querySellBaseToken(swapAmount).call();
|
|
console.log(i + "-swapAmount:",swapAmount);
|
|
} else {
|
|
directions[i] = 1;
|
|
swapAmount = await ctx.DODOSellHelper.methods.querySellQuoteToken(curPair.pair,swapAmount).call();
|
|
console.log(i + "-swapAmount:",swapAmount);
|
|
}
|
|
}
|
|
|
|
let toAmount = new BigNumber(swapAmount).multipliedBy(1-slippage).toFixed(0, BigNumber.ROUND_DOWN)
|
|
|
|
console.log("minAmount:",toAmount);
|
|
// console.log("dodoPairs",dodoPairs);
|
|
// console.log("directions",directions);
|
|
|
|
return ctx.SmartSwap.methods.dodoSwap(
|
|
routes[0].address,
|
|
routes[routes.length-1].address,
|
|
fromTokenAmount,
|
|
toAmount,
|
|
dodoPairs,
|
|
directions
|
|
)
|
|
}
|
|
|
|
describe("Trader", () => {
|
|
let snapshotId: string;
|
|
let ctx: DODOContext;
|
|
|
|
before(async () => {
|
|
ctx = await getDODOContext();
|
|
await initDODO_USDT(ctx);
|
|
await initUSDT_USDC(ctx);
|
|
await initWETH_USDC(ctx);
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
snapshotId = await ctx.EVM.snapshot();
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await ctx.EVM.reset(snapshotId);
|
|
});
|
|
|
|
describe("route calc test", () => {
|
|
it("DODO to USDT directly swap", async () => {
|
|
var b_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var b_USDT = await ctx.USDT.methods.balanceOf(trader).call()
|
|
console.log("Before DODO:" + fromWei(b_DODO,'ether') + "; USDT:" + fromWei(b_USDT,'mwei'));
|
|
//approve DODO entry
|
|
await ctx.DODO.methods.approve(ctx.SmartApprove.options.address,MAX_UINT256).send(ctx.sendParam(trader))
|
|
//set route path
|
|
var routes = [{
|
|
address: ctx.DODO.options.address,
|
|
decimals: 18
|
|
},
|
|
{
|
|
address: ctx.USDT.options.address,
|
|
decimals: 6
|
|
}];
|
|
|
|
var pairs = [{
|
|
pair: ctx.DODO_USDT.options.address,
|
|
base: ctx.DODO.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.DODO_USDT
|
|
/**************/
|
|
}];
|
|
|
|
var tx = await logGas(await calcRoute(ctx,decimalStr('10'),0.1,routes,pairs), ctx.sendParam(trader), "directly swap")
|
|
// console.log(tx.events['OrderHistory']);
|
|
var a_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var a_USDT = await ctx.USDT.methods.balanceOf(trader).call()
|
|
console.log("After DODO:" + fromWei(a_DODO,'ether') + "; USDT:" + fromWei(a_USDT,'mwei'));
|
|
console.log("===============================================")
|
|
var c_DODO = await ctx.DODO.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_USDT = await ctx.USDT.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
console.log("Contract DODO:" + fromWei(c_DODO,'ether') + "; USDT:" + fromWei(c_USDT,'mwei'));
|
|
});
|
|
|
|
|
|
it("DODO to USDC two hops swap", async () => {
|
|
var b_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var b_USDC = await ctx.USDC.methods.balanceOf(trader).call()
|
|
console.log("Before DODO:" + fromWei(b_DODO,'ether') + "; USDC:" + fromWei(b_USDC,'mwei'));
|
|
//approve DODO entry
|
|
await ctx.DODO.methods.approve(ctx.SmartApprove.options.address,MAX_UINT256).send(ctx.sendParam(trader))
|
|
//set route path
|
|
var routes = [{
|
|
address: ctx.DODO.options.address,
|
|
decimals: 18
|
|
},{
|
|
address: ctx.USDT.options.address,
|
|
decimals: 6
|
|
},{
|
|
address: ctx.USDC.options.address,
|
|
decimals: 6
|
|
}];
|
|
|
|
var pairs = [{
|
|
pair: ctx.DODO_USDT.options.address,
|
|
base: ctx.DODO.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.DODO_USDT
|
|
/**************/
|
|
},{
|
|
pair: ctx.USDT_USDC.options.address,
|
|
base: ctx.USDT.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.USDT_USDC
|
|
/**************/
|
|
}];
|
|
|
|
var tx = await logGas(await calcRoute(ctx,decimalStr('10'),0.1,routes,pairs), ctx.sendParam(trader), "tow hops swap")
|
|
// console.log(tx.events['Swapped']);
|
|
var a_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var a_USDC = await ctx.USDC.methods.balanceOf(trader).call()
|
|
console.log("After DODO:" + fromWei(a_DODO,'ether') + "; USDC:" + fromWei(a_USDC,'mwei'));
|
|
console.log("===============================================")
|
|
var c_DODO = await ctx.DODO.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_USDT = await ctx.USDT.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_USDC = await ctx.USDC.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
console.log("Contract DODO:" + fromWei(c_DODO,'ether') + "; USDT:" + fromWei(c_USDT,'mwei') + "; USDC:" + fromWei(c_USDC,'mwei'));
|
|
});
|
|
|
|
it("DODO to WETH three hops swap", async () => {
|
|
var b_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var b_WETH = await ctx.WETH.methods.balanceOf(trader).call()
|
|
console.log("Before DODO:" + fromWei(b_DODO,'ether') + "; WETH:" + fromWei(b_WETH,'ether'));
|
|
//approve DODO entry
|
|
await ctx.DODO.methods.approve(ctx.SmartApprove.options.address,MAX_UINT256).send(ctx.sendParam(trader))
|
|
//set route path
|
|
var routes = [{
|
|
address: ctx.DODO.options.address,
|
|
decimals: 18
|
|
},{
|
|
address: ctx.USDT.options.address,
|
|
decimals: 6
|
|
},{
|
|
address: ctx.USDC.options.address,
|
|
decimals: 6
|
|
},{
|
|
address: ctx.WETH.options.address,
|
|
decimals: 18
|
|
}];
|
|
|
|
var pairs = [{
|
|
pair: ctx.DODO_USDT.options.address,
|
|
base: ctx.DODO.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.DODO_USDT
|
|
/**************/
|
|
},{
|
|
pair: ctx.USDT_USDC.options.address,
|
|
base: ctx.USDT.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.USDT_USDC
|
|
/**************/
|
|
},{
|
|
pair: ctx.WETH_USDC.options.address,
|
|
base: ctx.WETH.options.address,
|
|
/*only for test*/
|
|
pairContract: ctx.WETH_USDC
|
|
/**************/
|
|
}];
|
|
|
|
var tx = await logGas(await calcRoute(ctx,decimalStr('10'),0.1,routes,pairs), ctx.sendParam(trader), "three hops swap")
|
|
console.log(tx.events['TestAmount']);
|
|
var a_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var a_WETH = await ctx.WETH.methods.balanceOf(trader).call()
|
|
console.log("After DODO:" + fromWei(a_DODO,'ether') + "; WETH:" + fromWei(a_WETH,'ether'));
|
|
console.log("===============================================")
|
|
var c_DODO = await ctx.DODO.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_USDT = await ctx.USDT.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_USDC = await ctx.USDC.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
var c_WETH = await ctx.WETH.methods.balanceOf(ctx.SmartSwap.options.address).call()
|
|
console.log("Contract DODO:" + fromWei(c_DODO,'ether') + "; USDT:" + fromWei(c_USDT,'mwei') + "; USDC:" + fromWei(c_USDC,'mwei') + "; WETH:" + fromWei(c_WETH,'ether'));
|
|
});
|
|
});
|
|
});
|