dvm init commit

2020-10-23 17:11:50 +08:00
parent a798c553ba
commit bd75b611b4
18 changed files with 179 additions and 63 deletions
--- a/test/utils/DVMContext.ts
+++ b/test/utils/DVMContext.ts
@@ -0,0 +1,182 @@
+/*
+
+    Copyright 2020 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, gweiStr, MAX_UINT256 } from './Converter';
+import { EVM, getDefaultWeb3 } from './EVM';
+import * as log from './Log';
+
+BigNumber.config({
+  EXPONENTIAL_AT: 1000,
+  DECIMAL_PLACES: 80,
+});
+
+export interface DVMContextInitConfig {
+  lpFeeRate: string;
+  mtFeeRate: string;
+  k: string;
+  i: string;
+  gasPriceLimit: string;
+}
+
+/*
+  price curve when k=0.1
+  +──────────────────────+───────────────+
+  | purchase percentage  | avg slippage  |
+  +──────────────────────+───────────────+
+  | 1%                   | 0.1%         |
+  | 5%                   | 0.5%         |
+  | 10%                  | 1.1%         |
+  | 20%                  | 2.5%         |
+  | 50%                  | 10%          |
+  | 70%                  | 23.3%        |
+  +──────────────────────+───────────────+
+*/
+export let DefaultDODOContextInitConfig = {
+  lpFeeRate: decimalStr("0.002"),
+  mtFeeRate: decimalStr("0.001"),
+  k: decimalStr("0.1"),
+  i: decimalStr("100"),
+  gasPriceLimit: gweiStr("100"),
+};
+
+export class DODOContext {
+  EVM: EVM;
+  Web3: Web3;
+  DODO: Contract;
+  DODOZoo: Contract;
+  BASE: Contract;
+  BaseCapital: Contract;
+  QUOTE: Contract;
+  QuoteCapital: Contract;
+  ORACLE: Contract;
+  Deployer: string;
+  Supervisor: string;
+  Maintainer: string;
+  spareAccounts: string[];
+
+  constructor() { }
+
+  async init(config: DODOContextInitConfig) {
+    this.EVM = new EVM();
+    this.Web3 = getDefaultWeb3();
+    var cloneFactory = await contracts.newContract(
+      contracts.CLONE_FACTORY_CONTRACT_NAME
+    );
+
+    this.BASE = await contracts.newContract(
+      contracts.TEST_ERC20_CONTRACT_NAME,
+      ["TestBase", 18]
+    );
+    this.QUOTE = await contracts.newContract(
+      contracts.TEST_ERC20_CONTRACT_NAME,
+      ["TestQuote", 18]
+    );
+    this.ORACLE = await contracts.newContract(
+      contracts.NAIVE_ORACLE_CONTRACT_NAME
+    );
+
+    const allAccounts = await this.Web3.eth.getAccounts();
+    this.Deployer = allAccounts[0];
+    this.Supervisor = allAccounts[1];
+    this.Maintainer = allAccounts[2];
+    this.spareAccounts = allAccounts.slice(3, 10);
+
+    var DODOTemplate = await contracts.newContract(
+      contracts.DODO_CONTRACT_NAME
+    );
+    this.DODOZoo = await contracts.newContract(
+      contracts.DODO_ZOO_CONTRACT_NAME,
+      [
+        DODOTemplate.options.address,
+        cloneFactory.options.address,
+        this.Supervisor,
+      ]
+    );
+
+    await this.DODOZoo.methods
+      .breedDODO(
+        this.Maintainer,
+        this.BASE.options.address,
+        this.QUOTE.options.address,
+        this.ORACLE.options.address,
+        config.lpFeeRate,
+        config.mtFeeRate,
+        config.k,
+        config.gasPriceLimit
+      )
+      .send(this.sendParam(this.Deployer));
+
+    this.DODO = contracts.getContractWithAddress(
+      contracts.DODO_CONTRACT_NAME,
+      await this.DODOZoo.methods
+        .getDODO(this.BASE.options.address, this.QUOTE.options.address)
+        .call()
+    );
+    await this.DODO.methods
+      .enableBaseDeposit()
+      .send(this.sendParam(this.Deployer));
+    await this.DODO.methods
+      .enableQuoteDeposit()
+      .send(this.sendParam(this.Deployer));
+    await this.DODO.methods.enableTrading().send(this.sendParam(this.Deployer));
+
+    this.BaseCapital = contracts.getContractWithAddress(
+      contracts.DODO_LP_TOKEN_CONTRACT_NAME,
+      await this.DODO.methods._BASE_CAPITAL_TOKEN_().call()
+    );
+    this.QuoteCapital = contracts.getContractWithAddress(
+      contracts.DODO_LP_TOKEN_CONTRACT_NAME,
+      await this.DODO.methods._QUOTE_CAPITAL_TOKEN_().call()
+    );
+
+    console.log(log.blueText("[Init dodo context]"));
+  }
+
+  sendParam(sender, value = "0") {
+    return {
+      from: sender,
+      gas: process.env["COVERAGE"] ? 10000000000 : 7000000,
+      gasPrice: process.env.GAS_PRICE,
+      value: decimalStr(value),
+    };
+  }
+
+  async setOraclePrice(price: string) {
+    await this.ORACLE.methods
+      .setPrice(price)
+      .send(this.sendParam(this.Deployer));
+  }
+
+  async mintTestToken(to: string, base: string, quote: string) {
+    await this.BASE.methods.mint(to, base).send(this.sendParam(this.Deployer));
+    await this.QUOTE.methods
+      .mint(to, quote)
+      .send(this.sendParam(this.Deployer));
+  }
+
+  async approveDODO(account: string) {
+    await this.BASE.methods
+      .approve(this.DODO.options.address, MAX_UINT256)
+      .send(this.sendParam(account));
+    await this.QUOTE.methods
+      .approve(this.DODO.options.address, MAX_UINT256)
+      .send(this.sendParam(account));
+  }
+}
+
+export async function getDODOContext(
+  config: DODOContextInitConfig = DefaultDODOContextInitConfig
+): Promise<DODOContext> {
+  var context = new DODOContext();
+  await context.init(config);
+  return context;
+}