d-bis/dodo-contractV2
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DODO Mine reward vault & reader

Browse Source
This commit is contained in:
mingda
2020-10-04 10:47:47 +08:00
parent c3c69cc0d9
commit 5f10f065e4
7 changed files with 104 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/Mining.test.ts
View File

@@ -8,7 +8,7 @@
import { DODOContext, getDODOContext } from './utils/Context';
import { decimalStr, MAX_UINT256 } from './utils/Converter';
// import * as assert from "assert"
import { newContract, DODO_TOKEN_CONTRACT_NAME, DODO_MINE_NAME, TEST_ERC20_CONTRACT_NAME, getContractWithAddress } from './utils/Contracts';
import { newContract, DODO_TOKEN_CONTRACT_NAME, DODO_MINE_NAME, TEST_ERC20_CONTRACT_NAME, getContractWithAddress, DODO_MINE_READER_NAME } from './utils/Contracts';
import { Contract } from 'web3-eth-contract';
import { assert } from 'chai';
import { logGas } from './utils/Log';
@@ -17,6 +17,7 @@ let BaseDLP: Contract
let QuoteDLP: Contract
let DODOToken: Contract
let DODOMine: Contract
let DODOMineReader: Contract
let lp1: string;
let lp2: string;
@@ -38,6 +39,7 @@ async function init(ctx: DODOContext): Promise<void> {
DODOToken = await newContract(DODO_TOKEN_CONTRACT_NAME)
DODOMine = await newContract(DODO_MINE_NAME, [DODOToken.options.address, (await ctx.Web3.eth.getBlockNumber()).toString()])
DODOMineReader = await newContract(DODO_MINE_READER_NAME)
BaseDLP = await getContractWithAddress(TEST_ERC20_CONTRACT_NAME, await ctx.DODO.methods._BASE_CAPITAL_TOKEN_().call())
QuoteDLP = await getContractWithAddress(TEST_ERC20_CONTRACT_NAME, await ctx.DODO.methods._QUOTE_CAPITAL_TOKEN_().call())
@@ -48,10 +50,12 @@ async function init(ctx: DODOContext): Promise<void> {
await BaseDLP.methods.approve(DODOMine.options.address, MAX_UINT256).send(ctx.sendParam(lp2))
await QuoteDLP.methods.approve(DODOMine.options.address, MAX_UINT256).send(ctx.sendParam(lp2))
await DODOMine.methods.setReward(decimalStr("100")).send(ctx.sendParam(ctx.Deployer))
await DODOMine.methods.setReward(decimalStr("100"), true).send(ctx.sendParam(ctx.Deployer))
await DODOMine.methods.addLpToken(BaseDLP.options.address, "1", true).send(ctx.sendParam(ctx.Deployer))
await DODOMine.methods.addLpToken(QuoteDLP.options.address, "2", true).send(ctx.sendParam(ctx.Deployer))
await DODOToken.methods.transfer(DODOMine.options.address, decimalStr("100000000")).send(ctx.sendParam(ctx.Deployer))
const rewardVault = await DODOMine.methods.dodoRewardVault().call()
await DODOToken.methods.transfer(rewardVault, decimalStr("100000000")).send(ctx.sendParam(ctx.Deployer))
}
describe("Lock DODO Token", () => {
@@ -73,7 +77,7 @@ describe("Lock DODO Token", () => {
});
describe("Lp Deposit", () => {
it.only("single lp deposit", async () => {
it("single lp deposit", async () => {
await logGas(DODOMine.methods.deposit(BaseDLP.options.address, decimalStr("100")), ctx.sendParam(lp1), "deposit")
await ctx.EVM.fastMove(100)
assert.equal(await DODOMine.methods.getPendingReward(BaseDLP.options.address, lp1).call(), "3333333333333333333300")
@@ -99,7 +103,7 @@ describe("Lock DODO Token", () => {
assert.equal(await DODOMine.methods.getAllPendingReward(lp2).call(), "8366666666666666666600")
})
it("lp multi deposit and withdraw", async () => {
it.only("lp multi deposit and withdraw", async () => {
await DODOMine.methods.deposit(BaseDLP.options.address, decimalStr("100")).send(ctx.sendParam(lp2))
await DODOMine.methods.deposit(BaseDLP.options.address, decimalStr("100")).send(ctx.sendParam(lp1))
await ctx.EVM.fastMove(100)
@@ -112,6 +116,10 @@ describe("Lock DODO Token", () => {
assert.equal(await DODOMine.methods.getAllPendingReward(lp1).call(), "0")
assert.equal(await DODOToken.methods.balanceOf(lp1).call(), "2805555555555555555500")
assert.equal(await DODOMine.methods.getRealizedReward(lp1).call(), "2805555555555555555500")
var balance = await DODOMineReader.methods.getUserStakedBalance(DODOMine.options.address, ctx.DODO.options.address, lp1).call()
assert.equal(balance.baseBalance, decimalStr("100"))
assert.equal(balance.quoteBalance, decimalStr("0"))
})
it("lp claim", async () => {

3
test/utils/Contracts.ts
View File

@@ -22,6 +22,7 @@ const Uniswap = require(`${jsonPath}UniswapV2Pair.json`)
const UniswapArbitrageur = require(`${jsonPath}UniswapArbitrageur.json`)
const DODOToken = require(`${jsonPath}DODOToken.json`)
const DODOMine = require(`${jsonPath}DODOMine.json`)
const DODOMineReader = require(`${jsonPath}DODOMineReader.json`)
const LockedTokenVault = require(`${jsonPath}LockedTokenVault.json`)
import { getDefaultWeb3 } from './EVM';
@@ -41,6 +42,7 @@ export const UNISWAP_ARBITRAGEUR_CONTRACT_NAME = "UniswapArbitrageur"
export const DODO_TOKEN_CONTRACT_NAME = "DODOToken"
export const LOCKED_TOKEN_VAULT_CONTRACT_NAME = "LockedTokenVault"
export const DODO_MINE_NAME = "DODOMine"
export const DODO_MINE_READER_NAME = "DODOMineReader"
var contractMap: { [name: string]: any } = {}
contractMap[CLONE_FACTORY_CONTRACT_NAME] = CloneFactory
@@ -56,6 +58,7 @@ contractMap[UNISWAP_ARBITRAGEUR_CONTRACT_NAME] = UniswapArbitrageur
contractMap[DODO_TOKEN_CONTRACT_NAME] = DODOToken
contractMap[LOCKED_TOKEN_VAULT_CONTRACT_NAME] = LockedTokenVault
contractMap[DODO_MINE_NAME] = DODOMine
contractMap[DODO_MINE_READER_NAME] = DODOMineReader
interface ContractJson {
abi: any;