Files
dodo-contractV2/test/DVM/AMMLikeCase.test.ts
2020-11-25 22:21:43 +08:00

137 lines
3.7 KiB
TypeScript

/*
Copyright 2020 DODO ZOO.
SPDX-License-Identifier: Apache-2.0
*/
// import * as assert from 'assert';
import { decimalStr } from '../utils/Converter';
import { DVMContext, getDVMContext } from '../utils/DVMContext';
import { assert } from 'chai';
let lp: string;
let trader: string;
async function init(ctx: DVMContext): Promise<void> {
lp = ctx.SpareAccounts[0];
trader = ctx.SpareAccounts[1];
await ctx.mintTestToken(lp, decimalStr("10"), decimalStr("1000"));
await ctx.mintTestToken(trader, decimalStr("10"), decimalStr("1000"));
await ctx.transferBaseToDVM(lp, decimalStr("10"))
await ctx.transferQuoteToDVM(lp, decimalStr("1000"))
await ctx.DVM.methods.buyShares(lp).send(ctx.sendParam(lp));
console.log("deposit")
}
describe("AMMLikeCase", () => {
let snapshotId: string;
let ctx: DVMContext;
before(async () => {
let AMMLikeDVMContextInitConfig = {
lpFeeRate: decimalStr("0.002"),
mtFeeRate: decimalStr("0.001"),
k: decimalStr("1"),
i: "1",
};
ctx = await getDVMContext(AMMLikeDVMContextInitConfig);
await init(ctx);
});
beforeEach(async () => {
snapshotId = await ctx.EVM.snapshot();
});
afterEach(async () => {
await ctx.EVM.reset(snapshotId);
});
describe("trade", () => {
it("basic state", async () => {
console.log(await ctx.DVM.methods.getMidPrice().call())
console.log(await ctx.DVM.methods.getBase0().call())
})
it("buy", async () => {
console.log("BASE0 before buy", await ctx.DVM.methods.getBase0().call())
// buy
await ctx.transferQuoteToDVM(trader, decimalStr("200"))
await ctx.DVM.methods.sellQuote(trader).send(ctx.sendParam(trader))
console.log("BASE0 after buy", await ctx.DVM.methods.getBase0().call())
// trader balances
assert.equal(
await ctx.BASE.methods.balanceOf(trader).call(),
"11661666666528194443"
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(trader).call(),
decimalStr("800")
);
// vault balances
assert.equal(
await ctx.BASE.methods.balanceOf(ctx.DVM.options.address).call(),
"8336666666805277779"
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(ctx.DVM.options.address).call(),
decimalStr("1200")
);
// maintainer balances
assert.equal(
await ctx.BASE.methods.balanceOf(ctx.Maintainer).call(),
"1666666666527778"
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(ctx.Maintainer).call(),
decimalStr("0")
);
});
it("sell", async () => {
console.log("BASE0 before sell", await ctx.DVM.methods.getBase0().call())
// sell
await ctx.transferBaseToDVM(trader, decimalStr("1"))
await ctx.DVM.methods.sellBase(trader).send(ctx.sendParam(trader))
console.log("BASE0 after sell", await ctx.DVM.methods.getBase0().call())
// trader balances
assert.equal(
await ctx.BASE.methods.balanceOf(trader).call(),
decimalStr("9")
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(trader).call(),
"1090636363645427272727"
);
// vault balances
assert.equal(
await ctx.BASE.methods.balanceOf(ctx.DVM.options.address).call(),
decimalStr("11")
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(ctx.DVM.options.address).call(),
"909272727263654545454"
);
// maintainer balances
assert.equal(
await ctx.BASE.methods.balanceOf(ctx.Maintainer).call(),
"0"
);
assert.equal(
await ctx.QUOTE.methods.balanceOf(ctx.Maintainer).call(),
"90909090918181819"
);
});
});
});