update feeratemodel & test case

test/DVM/FeeRateModelUpdate.test.ts

@@ -0,0 +1,90 @@
+/*
+
+    Copyright 2020 DODO ZOO.
+    SPDX-License-Identifier: Apache-2.0
+
+*/
+
+// import * as assert from 'assert';
+
+import { decimalStr, gweiStr } from '../utils/Converter';
+import { logGas } from '../utils/Log';
+import { DVMContext, getDVMContext } from '../utils/DVMContext';
+import { assert } from 'chai';
+import { EXTERNAL_VALUE_NAME, getContractWithAddress } from '../utils/Contracts';
+const truffleAssert = require('truffle-assertions');
+
+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.DVM.methods.buyShares(lp).send(ctx.sendParam(lp))
+}
+
+describe("FeeratemodelUpdate", () => {
+  let snapshotId: string;
+  let ctx: DVMContext;
+
+  before(async () => {
+    ctx = await getDVMContext();
+    await init(ctx);
+  });
+
+  beforeEach(async () => {
+    snapshotId = await ctx.EVM.snapshot();
+  });
+
+  afterEach(async () => {
+    await ctx.EVM.reset(snapshotId);
+  });
+
+  describe("Feeratemodel", () => {
+
+    it("feeRateUpdate", async () => {
+      var feeRate = await ctx.DVM.methods.getUserFeeRate(lp).call()
+      console.log(feeRate[1])//1000000000000000
+      assert.equal(
+        feeRate[1],
+        "1000000000000000"
+      );
+
+      console.log('~~~~~~~~~~~~~~~~~start set new feerate~~~~~~~~~~~~~~~~~')
+      var feerateLogicAddress = ctx.MtFeeRateModelLogic.options.address;
+
+      await ctx.mtFeeRateModel.methods.setFeeRate(decimalStr("0.003"),feerateLogicAddress).send(ctx.sendParam(ctx.Deployer))
+      var feeRateSet = await ctx.DVM.methods.getUserFeeRate(lp).call()
+      console.log(feeRateSet[1])
+      assert.equal(
+        feeRateSet[1],
+        "3000000000000000"
+      );
+      console.log('~~~~~~~~~~~~~~~~~start update feerateModel~~~~~~~~~~~~~~~~~')
+
+      var feerateLogicUpdateAddress = ctx.MtFeeRateModelLogicUpdate.options.address;
+      await ctx.mtFeeRateModel.methods.setFeeRate(decimalStr("0.001"),feerateLogicUpdateAddress).send(ctx.sendParam(ctx.Deployer))
+      var feeRateUpdate = await ctx.DVM.methods.getUserFeeRate(lp).call()
+      console.log(feeRateUpdate[1])
+      assert.equal(
+        feeRateUpdate[1],
+        "4000000000000000"
+      );
+      console.log('~~~~~~~~~~~~~~~~~set feeMapping[trader] ==0 ~~~~~~~~~~~~~~~~~')
+      await ctx.mtFeeRateModel.methods.setSpecificFeeRate(trader,decimalStr("0.001"),feerateLogicUpdateAddress).send(ctx.sendParam(ctx.Deployer))
+      var feeRateTrader = await ctx.DVM.methods.getUserFeeRate(trader).call()
+      console.log(feeRateTrader[1])
+      assert.equal(// if(feeMapping[trader] == 0) return _FEE_RATE_;
+        feeRateUpdate[1],
+        "4000000000000000"
+      );
+
+
+    })
+  });
+});

test/utils/Contracts.ts

@@ -37,6 +37,8 @@ export const PERMISSION_MANAGER_NAME = "PermissionManager"
 export const EXTERNAL_VALUE_NAME = "ExternalValue"
 export const DODO_PROXY_NAME = "DODOV2Proxy01"
 export const FEE_RATE_MODEL_NAME = "FeeRateModel"
+export const FEE_RATE_MODEL_LOGIC_NAME = "FeeRateModelLogic"
+export const FEE_RATE_MODEL_LOGIC_UPDATE_NAME = "FeeRateModelLogicUpdate"
 export const DPP_NAME = "DPP"
 export const DPP_FACTORY_NAME = "DPPFactory"
 export const SMART_APPROVE = "DODOApprove"

test/utils/DVMContext.ts

@@ -57,6 +57,11 @@ export class DVMContext {
   MtFeeRate: string;
   SpareAccounts: string[];
 
+  mtFeeRateModel: Contract;
+
+  MtFeeRateModelLogic: Contract;
+  MtFeeRateModelLogicUpdate: Contract;
+
   constructor() { }
 
   async init(config: DVMContextInitConfig) {
@@ -64,8 +69,9 @@ export class DVMContext {
     this.Web3 = getDefaultWeb3();
 
     this.DVM = await contracts.newContract(contracts.DVM_NAME)
-    var lpFeeRateModel = await contracts.newContract(contracts.CONST_FEE_RATE_MODEL_NAME)
-    var mtFeeRateModel = await contracts.newContract(contracts.CONST_FEE_RATE_MODEL_NAME)
+    var lpFeeRateModel = await contracts.newContract(contracts.FEE_RATE_MODEL_NAME)
+    var mtFeeRateModel = await contracts.newContract(contracts.FEE_RATE_MODEL_NAME)
+    this.mtFeeRateModel = mtFeeRateModel;
     this.MtFeeRate = mtFeeRateModel.options.address
     var permissionManager = await contracts.newContract(contracts.PERMISSION_MANAGER_NAME)
     var gasPriceSource = await contracts.newContract(contracts.EXTERNAL_VALUE_NAME)
@@ -103,6 +109,10 @@ export class DVMContext {
     await lpFeeRateModel.methods.init(this.Deployer, config.lpFeeRate).send(this.sendParam(this.Deployer))
     await mtFeeRateModel.methods.init(this.Deployer, config.mtFeeRate).send(this.sendParam(this.Deployer))
 
+
+    this.MtFeeRateModelLogic = await contracts.newContract(contracts.FEE_RATE_MODEL_LOGIC_NAME)
+    this.MtFeeRateModelLogicUpdate = await contracts.newContract(contracts.FEE_RATE_MODEL_LOGIC_UPDATE_NAME)
+
     console.log(log.blueText("[Init DVM context]"));
   }
 

test/utils/ProxyContextV2.ts

@@ -39,6 +39,8 @@ export class ProxyContext {
 
   //Functions
   DODOIncentive: Contract;
+  MtFeeRateModelLogic: Contract;
+  MtFeeRateModelLogicUpdate: Contract;
 
   Deployer: string;
   Maintainer: string;
@@ -78,8 +80,8 @@ export class ProxyContext {
     var dppAdminTemplate = await contracts.newContract(contracts.DPP_ADMIN_NAME)
     var permissionManagerTemplate = await contracts.newContract(contracts.PERMISSION_MANAGER_NAME)
     var mtFeeRateModelTemplate = await contracts.newContract(contracts.FEE_RATE_MODEL_NAME)
-    // await mtFeeRateModelTemplate.methods.init(this.Deployer,decimalStr("0.01")).send(this.sendParam(this.Deployer));
-    await mtFeeRateModelTemplate.methods.init(this.Deployer,decimalStr("0")).send(this.sendParam(this.Deployer));
+    await mtFeeRateModelTemplate.methods.init(this.Deployer,decimalStr("0.01")).send(this.sendParam(this.Deployer));
+    // await mtFeeRateModelTemplate.methods.init(this.Deployer,decimalStr("0")).send(this.sendParam(this.Deployer));
 
     this.DVMFactory = await contracts.newContract(contracts.DVM_FACTORY_NAME,
       [
@@ -150,6 +152,11 @@ export class ProxyContext {
       [this.WETH.options.address]
     )
 
+
+    this.MtFeeRateModelLogic = await contracts.newContract(contracts.FEE_RATE_MODEL_LOGIC_NAME)
+    this.MtFeeRateModelLogicUpdate = await contracts.newContract(contracts.FEE_RATE_MODEL_LOGIC_UPDATE_NAME)
+
+
     console.log(log.blueText("[Init DVM context]"));
   }