dodo-contractV2/test/utils/VDODOContext.ts

/*

    Copyright 2021 DODO ZOO.
    SPDX-License-Identifier: Apache-2.0

*/

import BigNumber from 'bignumber.js';
import Web3 from 'web3';
import { Contract } from 'web3-eth-contract';

import * as contracts from './Contracts';
import { decimalStr, MAX_UINT256 } from './Converter';
import { EVM, getDefaultWeb3 } from './EVM';
import * as log from './Log';

BigNumber.config({
  EXPONENTIAL_AT: 1000,
  DECIMAL_PLACES: 80,
});

export class VDODOContext {
  EVM: EVM;
  Web3: Web3;
  Deployer: string;
  Maintainer: string;
  SpareAccounts: string[];

  //token
  DODO: Contract;
  VDODO: Contract;

  DODOApprove: Contract;
  DODOApproveProxy: Contract;

  DODOCirculationHelper: Contract;
  // Governance: Contract;

  lastRewardBlock: number;
  alpha: number;



  constructor() { }

  async init() {
    this.EVM = new EVM();
    this.Web3 = getDefaultWeb3();

    const allAccounts = await this.Web3.eth.getAccounts();
    this.Deployer = allAccounts[0];
    this.Maintainer = allAccounts[1];
    this.SpareAccounts = allAccounts.slice(2, 10);

    this.DODO = await contracts.newContract(
      contracts.MINTABLE_ERC20_CONTRACT_NAME,
      ["DODO Token", "DODO", 18]
    );

    this.DODOApprove = await contracts.newContract(
      contracts.SMART_APPROVE
    );

    this.DODOApproveProxy = await contracts.newContract(
      contracts.SMART_APPROVE_PROXY,
      [this.DODOApprove.options.address]
    )

    // this.Governance = await contracts.newContract(
    //   contracts.DODO_GOVERNANCE,
    //   [
    //     this.DODO.options.address
    //   ]
    // )

    this.VDODO = await contracts.newContract(
      contracts.VDODO_NAME,
      [
        "0x0000000000000000000000000000000000000000",
        this.DODO.options.address,
        this.DODOApproveProxy.options.address,
        this.Deployer
      ]
    )

    this.DODOCirculationHelper = await contracts.newContract(
      contracts.DODO_CULATION_HELPER,
      [
        this.VDODO.options.address,
        this.DODO.options.address
      ]
    );

    // await this.Governance.methods.initOwner(
    //   this.Deployer
    // ).send(this.sendParam(this.Deployer))

    // await this.Governance.methods.setVDODOAddress(
    //   this.VDODO.options.address
    // ).send(this.sendParam(this.Deployer))

    await this.DODOApprove.methods.init(this.Deployer, this.DODOApproveProxy.options.address).send(this.sendParam(this.Deployer));
    await this.DODOApproveProxy.methods.init(this.Deployer, [this.VDODO.options.address]).send(this.sendParam(this.Deployer));


    await this.VDODO.methods.initOwner(
      this.Deployer
    ).send(this.sendParam(this.Deployer))


    await this.VDODO.methods.updateDODOCirculationHelper(this.DODOCirculationHelper.options.address).send(this.sendParam(this.Deployer));
    await this.mintTestToken(allAccounts[8], decimalStr("10000"));
    await this.approveProxy(allAccounts[8]);

    await this.VDODO.methods.preDepositedBlockReward(decimalStr("10000")).send(this.sendParam(allAccounts[8]));

    var lastRewardBlock = await this.VDODO.methods._LAST_REWARD_BLOCK_().call();
    var curBlock = await this.Web3.eth.getBlockNumber();
    console.log("init-block:" + lastRewardBlock + " blockNumber:" + curBlock)

    await this.VDODO.methods.changePerReward(decimalStr("1")).send(this.sendParam(this.Deployer));
    console.log(log.blueText("[Init VDODO context]"));
  }

  sendParam(sender, value = "0") {
    return {
      from: sender,
      gas: process.env["COVERAGE"] ? 10000000000 : 7000000,
      gasPrice: process.env.GAS_PRICE,
      value: decimalStr(value),
    };
  }

  async mintTestToken(to: string, amount: string) {
    await this.DODO.methods.mint(to, amount).send(this.sendParam(this.Deployer));
  }

  async approveProxy(account: string) {
    await this.DODO.methods
      .approve(this.DODOApprove.options.address, MAX_UINT256)
      .send(this.sendParam(account));
  }
}

export async function getVDODOContext(): Promise<VDODOContext> {
  var context = new VDODOContext();
  await context.init();
  return context;
}