275 lines
13 KiB
TypeScript
275 lines
13 KiB
TypeScript
/*
|
|
|
|
Copyright 2020 DODO ZOO.
|
|
SPDX-License-Identifier: Apache-2.0
|
|
|
|
*/
|
|
|
|
import BigNumber from 'bignumber.js';
|
|
import { DODOContext, getDODOContext } from '../utils-v1/ProxyContextV1';
|
|
import { decimalStr, MAX_UINT256, fromWei, mweiStr } from '../utils-v1/Converter';
|
|
import { logGas } from '../utils-v1/Log';
|
|
import * as contracts from '../utils-v1/Contracts';
|
|
import { assert } from 'chai';
|
|
|
|
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 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 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("1000000"), mweiStr("1000000"));
|
|
|
|
await USDT_USDC.methods
|
|
.depositBaseTo(lp, mweiStr("1000000"))
|
|
.send(ctx.sendParam(lp));
|
|
await USDT_USDC.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 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));
|
|
}
|
|
|
|
async function initIncentive(ctx: DODOContext): Promise<void> {
|
|
var blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
await ctx.DODOIncentive.methods.switchIncentive(blockNum + 1).send(ctx.sendParam(ctx.Deployer));
|
|
await ctx.mintToken(ctx.DODO, null, ctx.DODOIncentive.options.address, decimalStr("10000"), mweiStr("0"));
|
|
}
|
|
|
|
//mock sdk logic
|
|
async function calcRoute(ctx: DODOContext, fromTokenAmount: string, slippage: number, routes: any[], pairs: any[], isIncentive: boolean) {
|
|
let tmpDirections: number[] = []
|
|
let strDirections: string = ''
|
|
let dodoPairs: string[] = []
|
|
|
|
for (let i = 0; i < pairs.length; i++) {
|
|
let curPair = pairs[i]
|
|
dodoPairs.push(curPair.pair)
|
|
if (routes[i].address == '0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE') {
|
|
tmpDirections[i] = 0;
|
|
} else if (curPair.base === routes[i].address) {
|
|
tmpDirections[i] = 0;
|
|
} else {
|
|
tmpDirections[i] = 1;
|
|
}
|
|
}
|
|
|
|
var [returmAmount,] = await ctx.DODOSwapCalcHelper.methods.calcReturnAmountV1(
|
|
fromTokenAmount,
|
|
dodoPairs,
|
|
tmpDirections,
|
|
).call();
|
|
// console.log("returnAmount:", returmAmount)
|
|
// console.log("localAmount:", swapAmount)
|
|
// console.log("midPrices:", midPrices)
|
|
|
|
|
|
let toAmount = new BigNumber(returmAmount).multipliedBy(1 - slippage).toFixed(0, BigNumber.ROUND_DOWN)
|
|
let deadline = Math.floor(new Date().getTime() / 1000 + 60 * 10);
|
|
|
|
for (let i = tmpDirections.length - 1; i >= 0; i--) {
|
|
strDirections += tmpDirections[i].toString()
|
|
}
|
|
|
|
return ctx.DODOV1Proxy02.methods.dodoSwapV1(
|
|
routes[0].address,
|
|
routes[routes.length - 1].address,
|
|
fromTokenAmount,
|
|
toAmount,
|
|
dodoPairs,
|
|
parseInt(strDirections, 2),
|
|
isIncentive,
|
|
deadline
|
|
)
|
|
}
|
|
|
|
describe("Trader", () => {
|
|
let snapshotId: string;
|
|
let ctx: DODOContext;
|
|
|
|
before(async () => {
|
|
let ETH = await contracts.newContract(
|
|
contracts.WETH_CONTRACT_NAME
|
|
);
|
|
ctx = await getDODOContext(ETH.options.address);
|
|
await initDODO_USDT(ctx);
|
|
await initUSDT_USDC(ctx);
|
|
await initWETH_USDC(ctx);
|
|
await initIncentive(ctx);
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
snapshotId = await ctx.EVM.snapshot();
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await ctx.EVM.reset(snapshotId);
|
|
});
|
|
|
|
describe("route calc with incentive test", () => {
|
|
it("incentive-switch with trade", async () => {
|
|
await ctx.DODOIncentive.methods.changePerReward(decimalStr("10")).send(ctx.sendParam(ctx.Deployer));
|
|
var totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
var totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
var blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("Init - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum);
|
|
|
|
//Aim to increase block
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("Close BlockNumber:", blockNum + 1)
|
|
await ctx.DODOIncentive.methods.switchIncentive(0).send(ctx.sendParam(ctx.Deployer));
|
|
totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("Close incentive - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum)
|
|
//Aim to increase block
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
await ctx.DODOIncentive.methods.switchIncentive(blockNum + 1).send(ctx.sendParam(ctx.Deployer));
|
|
console.log("Open BlockNumber:", blockNum + 1)
|
|
//Aim to increase block
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.DODOIncentive.methods.changePerReward(decimalStr("10")).send(ctx.sendParam(ctx.Deployer));
|
|
totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("End incentive - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum)
|
|
assert(totalReward, decimalStr("100"));
|
|
});
|
|
|
|
it("incentive-changeBoost with trade", async () => {
|
|
await ctx.DODOIncentive.methods.changePerReward(decimalStr("10")).send(ctx.sendParam(ctx.Deployer));
|
|
var totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
var totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
var blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("Init - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum);
|
|
|
|
//Aim to increase block
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("Change BlockNumber:", blockNum + 1)
|
|
await ctx.DODOIncentive.methods.changePerReward(decimalStr("20")).send(ctx.sendParam(ctx.Deployer));
|
|
totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("change incentive - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum)
|
|
|
|
//Aim to increase block
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.mintToken(ctx.DODO, null, lp, decimalStr("100"), mweiStr("0"));
|
|
await ctx.DODOIncentive.methods.changePerReward(decimalStr("10")).send(ctx.sendParam(ctx.Deployer));
|
|
totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
blockNum = await ctx.Web3.eth.getBlockNumber();
|
|
console.log("End incentive - Total Reward:" + fromWei(totalReward, 'ether') + "; Total distribution:" + fromWei(totalDistribution, 'ether') + "; BlockNumber:" + blockNum)
|
|
assert(totalReward, decimalStr("140"));
|
|
});
|
|
|
|
it("directly swap with incentive", async () => {
|
|
await ctx.mintToken(ctx.DODO, ctx.USDT, trader, decimalStr("1000"), mweiStr("0"));
|
|
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'));
|
|
|
|
var b_totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
var b_totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
console.log("Before Total Reward:" + fromWei(b_totalReward, 'ether') + "; Total distribution:" + fromWei(b_totalDistribution, 'ether'))
|
|
|
|
//approve DODO entry
|
|
await ctx.DODO.methods.approve(ctx.DODOApprove.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,
|
|
pairContract: ctx.DODO_USDT
|
|
}];
|
|
await logGas(await calcRoute(ctx, decimalStr('10'), 0.1, routes, pairs, false), ctx.sendParam(trader), "directly swap without incentive first")
|
|
await logGas(await calcRoute(ctx, decimalStr('10'), 0.1, routes, pairs, false), ctx.sendParam(trader), "directly swap without incentive second")
|
|
var a_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
var a_USDT = await ctx.USDT.methods.balanceOf(trader).call()
|
|
console.log("After No Incentive DODO:" + fromWei(a_DODO, 'ether') + "; USDT:" + fromWei(a_USDT, 'mwei'));
|
|
|
|
await logGas(await calcRoute(ctx, decimalStr('10'), 0.1, routes, pairs, true), ctx.sendParam(trader), "directly swap with incentive first")
|
|
await logGas(await calcRoute(ctx, decimalStr('10'), 0.1, routes, pairs, true), ctx.sendParam(trader), "directly swap with incentive second")
|
|
|
|
var a_totalReward = await ctx.DODOIncentive.methods.totalReward().call();
|
|
var a_totalDistribution = await ctx.DODOIncentive.methods.totalDistribution().call();
|
|
console.log("After Total Reward:" + fromWei(a_totalReward, 'ether') + "; Total distribution:" + fromWei(a_totalDistribution, 'ether'))
|
|
|
|
a_DODO = await ctx.DODO.methods.balanceOf(trader).call()
|
|
a_USDT = await ctx.USDT.methods.balanceOf(trader).call()
|
|
console.log("After Incentive DODO:" + fromWei(a_DODO, 'ether') + "; USDT:" + fromWei(a_USDT, 'mwei'));
|
|
assert(a_DODO, decimalStr("961.493"));
|
|
});
|
|
});
|
|
});
|