347 lines
15 KiB
TypeScript
347 lines
15 KiB
TypeScript
/*
|
||
|
||
Copyright 2021 DODO ZOO.
|
||
SPDX-License-Identifier: Apache-2.0
|
||
|
||
*/
|
||
|
||
import { decimalStr, fromWei } from '../utils/Converter';
|
||
import { logGas } from '../utils/Log';
|
||
import { VDODOContext, getVDODOContext } from '../utils/VDODOContext';
|
||
import { assert } from 'chai';
|
||
import BigNumber from 'bignumber.js';
|
||
const truffleAssert = require('truffle-assertions');
|
||
|
||
let account0: string;
|
||
let account1: string;
|
||
let account2: string;
|
||
let account3: string;
|
||
let defaultSuperAddress: string;
|
||
let owner: string;
|
||
|
||
async function init(ctx: VDODOContext): Promise<void> {
|
||
account0 = ctx.SpareAccounts[0];
|
||
account1 = ctx.SpareAccounts[1];
|
||
account2 = ctx.SpareAccounts[2];
|
||
account3 = ctx.SpareAccounts[3];
|
||
defaultSuperAddress = ctx.Maintainer
|
||
owner = ctx.Deployer
|
||
|
||
await ctx.mintTestToken(account0, decimalStr("1000"));
|
||
await ctx.mintTestToken(account2, decimalStr("1000"));
|
||
|
||
await ctx.approveProxy(account0);
|
||
await ctx.approveProxy(account1);
|
||
await ctx.approveProxy(account2);
|
||
await ctx.approveProxy(account3);
|
||
|
||
await ctx.VDODO.methods.setCantransfer(true).send(ctx.sendParam(owner))
|
||
}
|
||
|
||
async function getGlobalState(ctx: VDODOContext, logInfo?: string) {
|
||
var alpha = await ctx.VDODO.methods.getLatestAlpha().call();
|
||
var lastRewardBlock = await ctx.VDODO.methods.lastRewardBlock().call();
|
||
var totalSuppy = await ctx.VDODO.methods.totalSupply().call();
|
||
// console.log(logInfo + " alpha:" + fromWei(alpha, 'ether') + " lastRewardBlock:" + lastRewardBlock + " totalSuppy:" + fromWei(totalSuppy, 'ether'));
|
||
return [alpha, lastRewardBlock,totalSuppy]
|
||
}
|
||
|
||
async function dodoBalance(ctx: VDODOContext, user: string, logInfo?: string) {
|
||
var dodo_contract = await ctx.DODO.methods.balanceOf(ctx.VDODO.options.address).call();
|
||
var dodo_account = await ctx.DODO.methods.balanceOf(user).call();
|
||
|
||
// console.log(logInfo + " DODO:" + fromWei(dodo_contract, 'ether') + " account:" + fromWei(dodo_account, 'ether'));
|
||
return [dodo_contract, dodo_account]
|
||
}
|
||
|
||
async function getUserInfo(ctx: VDODOContext, user: string, logInfo?: string) {
|
||
var info = await ctx.VDODO.methods.userInfo(user).call();
|
||
var res = {
|
||
"VDODOAmount": info.VDODOAmount,
|
||
"superiorVDODO": info.superiorVDODO,
|
||
"superior": info.superior,
|
||
"credit": info.credit
|
||
}
|
||
// console.log(logInfo + " VDODOAmount:" + fromWei(info.VDODOAmount, 'ether') + " superiorVDODO:" + fromWei(info.superiorVDODO, 'ether') + " superior:" + info.superior + " credit:" + fromWei(info.credit, 'ether'));
|
||
return res
|
||
}
|
||
|
||
async function mint(ctx: VDODOContext, user: string, mintAmount: string, superior: string) {
|
||
await ctx.VDODO.methods.mint(
|
||
mintAmount,
|
||
superior
|
||
).send(ctx.sendParam(user));
|
||
}
|
||
|
||
describe("vDODO-erc20", () => {
|
||
let snapshotId: string;
|
||
let ctx: VDODOContext;
|
||
|
||
before(async () => {
|
||
ctx = await getVDODOContext();
|
||
//打开transfer开关
|
||
await init(ctx);
|
||
});
|
||
|
||
beforeEach(async () => {
|
||
snapshotId = await ctx.EVM.snapshot();
|
||
});
|
||
|
||
afterEach(async () => {
|
||
await ctx.EVM.reset(snapshotId);
|
||
});
|
||
|
||
describe("vdodo-erc20", () => {
|
||
|
||
it("transfer-vdodo", async () => {
|
||
//检查四个人 【包括from, to 以及各自的上级】，info变化
|
||
//alpha lastRewardBlock
|
||
//各自dodo余额变化
|
||
|
||
let [,lastRewardBlockStart,] = await getGlobalState(ctx, "before");
|
||
await ctx.VDODO.methods.mint(decimalStr("10"),account1).send(ctx.sendParam(account0))
|
||
await ctx.VDODO.methods.mint(decimalStr("10"),account3).send(ctx.sendParam(account2))
|
||
|
||
//增加一个区块
|
||
await ctx.mintTestToken(account0, decimalStr("0"));
|
||
let [alpha,lastRewardBlock,] = await getGlobalState(ctx, "after");
|
||
|
||
assert.equal(lastRewardBlock,Number(lastRewardBlockStart)+11);
|
||
|
||
assert.equal(alpha, "113833992094861660108");
|
||
var totalSuppy = await ctx.VDODO.methods.totalSupply().call();
|
||
assert.equal(
|
||
totalSuppy
|
||
, decimalStr("0.210833333333333332"));
|
||
|
||
|
||
|
||
let userInfo0 = await getUserInfo(ctx, account0, "User0 ");
|
||
assert.equal(userInfo0.VDODOAmount, decimalStr("0.1"));
|
||
assert.equal(userInfo0.superiorVDODO, decimalStr("0.01"));
|
||
assert.equal(userInfo0.credit, "0");
|
||
let userInfo1 = await getUserInfo(ctx, account1, "User0 Superior ")
|
||
assert.equal(userInfo1.VDODOAmount, decimalStr("0.01"));
|
||
assert.equal(userInfo1.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo1.credit, decimalStr("1"));
|
||
|
||
let userInfo2 = await getUserInfo(ctx, account2, "User2 ");
|
||
assert.equal(userInfo2.VDODOAmount, decimalStr("0.091666666666666666"));
|
||
assert.equal(userInfo2.superiorVDODO, decimalStr("0.009166666666666666"));
|
||
assert.equal(userInfo2.credit, decimalStr("0"));
|
||
let userInfo3 = await getUserInfo(ctx, account3, "User2 Superior");
|
||
assert.equal(userInfo3.VDODOAmount, decimalStr("0.009166666666666666"));
|
||
assert.equal(userInfo3.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo3.credit, decimalStr("0.999999999999999928"));
|
||
|
||
|
||
let [, dodo_u0] = await dodoBalance(ctx, account0, "start")
|
||
assert.equal(dodo_u0, "990000000000000000000");
|
||
let [, dodo_u1] = await dodoBalance(ctx, account1, "start")
|
||
assert.equal(dodo_u1, "0");
|
||
let [, dodo_u2] = await dodoBalance(ctx, account2, "start")
|
||
assert.equal(dodo_u2, "990000000000000000000");
|
||
let [, dodo_u3] = await dodoBalance(ctx, account3, "start")
|
||
assert.equal(dodo_u3, "0");
|
||
|
||
await logGas(await ctx.VDODO.methods.transfer(
|
||
account2,
|
||
decimalStr("0.1")
|
||
), ctx.sendParam(account0), "transfer");
|
||
|
||
|
||
// await ctx.VDODO.methods.transfer(account2,decimalStr("0.1")).send(ctx.sendParam(account0))
|
||
|
||
let userInfo0_after = await getUserInfo(ctx, account0, "userInfo0_after");
|
||
let userInfo1_after = await getUserInfo(ctx, account1, "userInfo1_after");
|
||
let userInfo2_after = await getUserInfo(ctx, account2, "userInfo2_after");
|
||
let userInfo3_after = await getUserInfo(ctx, account3, "userInfo3_after");
|
||
|
||
|
||
assert.equal(userInfo0_after.VDODOAmount, "0");
|
||
assert.equal(userInfo0_after.superiorVDODO, "0");
|
||
assert.equal(userInfo0_after.credit, "0");
|
||
|
||
assert.equal(userInfo1_after.VDODOAmount, decimalStr("0.001566666666666667"));
|
||
assert.equal(userInfo1_after.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo1_after.credit, "0");
|
||
|
||
assert.equal(userInfo2_after.VDODOAmount, decimalStr("0.191666666666666666"));
|
||
assert.equal(userInfo2_after.superiorVDODO, decimalStr("0.019166666666666666"));
|
||
assert.equal(userInfo2_after.credit, "0");
|
||
|
||
assert.equal(userInfo3_after.VDODOAmount, decimalStr("0.019166666666666666"));
|
||
assert.equal(userInfo3_after.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo3_after.credit, decimalStr("2.185770750988142222"));
|
||
|
||
|
||
|
||
let [alphaEnd,lastRewardBlockEnd,totalSuppyEnd] = await getGlobalState(ctx, "end");
|
||
assert.equal(alphaEnd, decimalStr("118.577075098814229308"));
|
||
assert.equal(totalSuppyEnd, decimalStr("0.212399999999999999"));
|
||
assert.equal(lastRewardBlockEnd,Number(lastRewardBlock)+2);
|
||
|
||
|
||
let [, dodo_u0_end] = await dodoBalance(ctx, account0, "end")
|
||
assert.equal(dodo_u0_end, "990000000000000000000");
|
||
let [, dodo_u1_end] = await dodoBalance(ctx, account1, "end")
|
||
assert.equal(dodo_u1_end, "0");
|
||
let [, dodo_u2_end] = await dodoBalance(ctx, account2, "end")
|
||
assert.equal(dodo_u2_end, "990000000000000000000");
|
||
let [, dodo_u3_end] = await dodoBalance(ctx, account3, "end")
|
||
assert.equal(dodo_u3_end, "0");
|
||
|
||
});
|
||
|
||
it("transferFrom-vdodo", async () => {
|
||
//检查四个人 【包括from, to 以及各自的上级】，info变化
|
||
//alpha lastRewardBlock
|
||
//各自dodo余额变化
|
||
//approve 状态变化
|
||
|
||
let [,lastRewardBlockStart,] = await getGlobalState(ctx, "before");
|
||
await ctx.VDODO.methods.mint(decimalStr("10"),account1).send(ctx.sendParam(account0))
|
||
await ctx.VDODO.methods.mint(decimalStr("10"),account3).send(ctx.sendParam(account2))
|
||
|
||
//增加一个区块
|
||
await ctx.mintTestToken(account0, decimalStr("0"));
|
||
let [alpha,lastRewardBlock,] = await getGlobalState(ctx, "after");
|
||
|
||
assert.equal(lastRewardBlock,Number(lastRewardBlockStart)+11);
|
||
|
||
assert.equal(alpha, "113833992094861660108");
|
||
var totalSuppy = await ctx.VDODO.methods.totalSupply().call();
|
||
assert.equal(
|
||
totalSuppy
|
||
, decimalStr("0.210833333333333332"));
|
||
|
||
|
||
|
||
let userInfo0 = await getUserInfo(ctx, account0, "User0 ");
|
||
assert.equal(userInfo0.VDODOAmount, decimalStr("0.1"));
|
||
assert.equal(userInfo0.superiorVDODO, decimalStr("0.01"));
|
||
assert.equal(userInfo0.credit, "0");
|
||
let userInfo1 = await getUserInfo(ctx, account1, "User0 Superior ")
|
||
assert.equal(userInfo1.VDODOAmount, decimalStr("0.01"));
|
||
assert.equal(userInfo1.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo1.credit, decimalStr("1"));
|
||
|
||
let userInfo2 = await getUserInfo(ctx, account2, "User2 ");
|
||
assert.equal(userInfo2.VDODOAmount, decimalStr("0.091666666666666666"));
|
||
assert.equal(userInfo2.superiorVDODO, decimalStr("0.009166666666666666"));
|
||
assert.equal(userInfo2.credit, decimalStr("0"));
|
||
let userInfo3 = await getUserInfo(ctx, account3, "User2 Superior");
|
||
assert.equal(userInfo3.VDODOAmount, decimalStr("0.009166666666666666"));
|
||
assert.equal(userInfo3.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo3.credit, decimalStr("0.999999999999999928"));
|
||
|
||
|
||
let [, dodo_u0] = await dodoBalance(ctx, account0, "start")
|
||
assert.equal(dodo_u0, "990000000000000000000");
|
||
let [, dodo_u1] = await dodoBalance(ctx, account1, "start")
|
||
assert.equal(dodo_u1, "0");
|
||
let [, dodo_u2] = await dodoBalance(ctx, account2, "start")
|
||
assert.equal(dodo_u2, "990000000000000000000");
|
||
let [, dodo_u3] = await dodoBalance(ctx, account3, "start")
|
||
assert.equal(dodo_u3, "0");
|
||
|
||
|
||
await logGas(await ctx.VDODO.methods.approve(
|
||
account3,
|
||
decimalStr("0.1")
|
||
), ctx.sendParam(account0), "approve");
|
||
|
||
await logGas(await ctx.VDODO.methods.transferFrom(
|
||
account0,
|
||
account2,
|
||
decimalStr("0.1")
|
||
), ctx.sendParam(account3), "transferFrom");
|
||
|
||
let userInfo0_after = await getUserInfo(ctx, account0, "userInfo0_after");
|
||
let userInfo1_after = await getUserInfo(ctx, account1, "userInfo1_after");
|
||
let userInfo2_after = await getUserInfo(ctx, account2, "userInfo2_after");
|
||
let userInfo3_after = await getUserInfo(ctx, account3, "userInfo3_after");
|
||
|
||
|
||
assert.equal(userInfo0_after.VDODOAmount, "0");
|
||
assert.equal(userInfo0_after.superiorVDODO, "0");
|
||
assert.equal(userInfo0_after.credit, "0");
|
||
|
||
assert.equal(userInfo1_after.VDODOAmount, decimalStr("0.001891025641025642"));
|
||
assert.equal(userInfo1_after.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo1_after.credit, "0");
|
||
|
||
assert.equal(userInfo2_after.VDODOAmount, decimalStr("0.191666666666666666"));
|
||
assert.equal(userInfo2_after.superiorVDODO, decimalStr("0.019166666666666666"));
|
||
assert.equal(userInfo2_after.credit, "0");
|
||
|
||
assert.equal(userInfo3_after.VDODOAmount, decimalStr("0.019166666666666666"));
|
||
assert.equal(userInfo3_after.superiorVDODO, decimalStr("0"));
|
||
assert.equal(userInfo3_after.credit, decimalStr("2.233201581027667914"));
|
||
|
||
|
||
|
||
let [alphaEnd,lastRewardBlockEnd,totalSuppyEnd] = await getGlobalState(ctx, "end");
|
||
assert.equal(alphaEnd, decimalStr("123.320158102766798508"));
|
||
assert.equal(totalSuppyEnd, decimalStr("0.212724358974358974"));
|
||
assert.equal(lastRewardBlockEnd,Number(lastRewardBlock)+3);
|
||
|
||
|
||
let [, dodo_u0_end] = await dodoBalance(ctx, account0, "end")
|
||
assert.equal(dodo_u0_end, "990000000000000000000");
|
||
let [, dodo_u1_end] = await dodoBalance(ctx, account1, "end")
|
||
assert.equal(dodo_u1_end, "0");
|
||
let [, dodo_u2_end] = await dodoBalance(ctx, account2, "end")
|
||
assert.equal(dodo_u2_end, "990000000000000000000");
|
||
let [, dodo_u3_end] = await dodoBalance(ctx, account3, "end")
|
||
assert.equal(dodo_u3_end, "0");
|
||
|
||
|
||
//再次transferFrom 预期revert
|
||
//预期revert
|
||
await truffleAssert.reverts(
|
||
ctx.VDODO.methods.transferFrom(account0,account2,decimalStr("0.1")).send(ctx.sendParam(account3)),
|
||
"ALLOWANCE_NOT_ENOUGH"
|
||
)
|
||
});
|
||
|
||
it("transfer - close", async () => {
|
||
|
||
await ctx.VDODO.methods.setCantransfer(false).send(ctx.sendParam(owner))
|
||
|
||
await ctx.VDODO.methods.mint(decimalStr("10"),defaultSuperAddress).send(ctx.sendParam(account0))
|
||
assert.equal(
|
||
await ctx.DODO.methods.balanceOf(account0).call(),
|
||
decimalStr("990")
|
||
);
|
||
assert.equal(
|
||
await ctx.DODO.methods.balanceOf(ctx.VDODO.options.address).call(),
|
||
decimalStr("100010")
|
||
);
|
||
assert.equal(
|
||
await ctx.VDODO.methods.balanceOf(account0).call(),
|
||
decimalStr("0.1")
|
||
);
|
||
|
||
assert.equal(
|
||
await ctx.VDODO.methods.balanceOf(account1).call(),
|
||
decimalStr("0")
|
||
);
|
||
//预期revert
|
||
await truffleAssert.reverts(
|
||
ctx.VDODO.methods.transfer(account1,decimalStr("0.1")).send(ctx.sendParam(account0)),
|
||
"vDODOToken: not allowed transfer"
|
||
)
|
||
assert.equal(
|
||
await ctx.VDODO.methods.balanceOf(account0).call(),
|
||
decimalStr("0.1")
|
||
);
|
||
assert.equal(
|
||
await ctx.VDODO.methods.balanceOf(account1).call(),
|
||
decimalStr("0")
|
||
);
|
||
|
||
});
|
||
})
|
||
});
|