Add buying, selling and capital limit controls

These advanced controls are disabled by default.

Also disable depositing and trading when creating a new DODO.

contracts/dodo.sol

@@ -16,6 +16,7 @@ import {LiquidityProvider} from "./impl/LiquidityProvider.sol";
 import {Admin} from "./impl/Admin.sol";
 import {DODOLpToken} from "./impl/DODOLpToken.sol";
 
+
 /**
  * @title DODO
  * @author DODO Breeder
@@ -48,11 +49,18 @@ contract DODO is Admin, Trader, LiquidityProvider {
         _QUOTE_TOKEN_ = quoteToken;
         _ORACLE_ = oracle;
 
-        _DEPOSIT_BASE_ALLOWED_ = true;
+        _DEPOSIT_BASE_ALLOWED_ = false;
-        _DEPOSIT_QUOTE_ALLOWED_ = true;
+        _DEPOSIT_QUOTE_ALLOWED_ = false;
-        _TRADE_ALLOWED_ = true;
+        _TRADE_ALLOWED_ = false;
         _GAS_PRICE_LIMIT_ = gasPriceLimit;
 
+        // Advanced controls are disabled by default
+        _BUYING_ALLOWED_ = true;
+        _SELLING_ALLOWED_ = true;
+        uint256 MAX_INT = 0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff;
+        _BASE_BALANCE_LIMIT_ = MAX_INT;
+        _QUOTE_BALANCE_LIMIT_ = MAX_INT;
+
         _LP_FEE_RATE_ = lpFeeRate;
         _MT_FEE_RATE_ = mtFeeRate;
         _K_ = k;

contracts/impl/Admin.sol

@@ -10,6 +10,7 @@ pragma experimental ABIEncoderV2;
 
 import {Storage} from "./Storage.sol";
 
+
 /**
  * @title Admin
  * @author DODO Breeder
@@ -92,4 +93,30 @@ contract Admin is Storage {
     function enableBaseDeposit() external onlyOwner notClosed {
         _DEPOSIT_BASE_ALLOWED_ = true;
     }
+
+    // ============ Advanced Control Functions ============
+
+    function disableBuying() external onlySupervisorOrOwner {
+        _BUYING_ALLOWED_ = false;
+    }
+
+    function enableBuying() external onlyOwner notClosed {
+        _BUYING_ALLOWED_ = true;
+    }
+
+    function disableSelling() external onlySupervisorOrOwner {
+        _SELLING_ALLOWED_ = false;
+    }
+
+    function enableSelling() external onlyOwner notClosed {
+        _SELLING_ALLOWED_ = true;
+    }
+
+    function setBaseBalanceLimit(uint256 newBaseBalanceLimit) external onlyOwner notClosed {
+        _BASE_BALANCE_LIMIT_ = newBaseBalanceLimit;
+    }
+
+    function setQuoteBalanceLimit(uint256 newQuoteBalanceLimit) external onlyOwner notClosed {
+        _QUOTE_BALANCE_LIMIT_ = newQuoteBalanceLimit;
+    }
 }

contracts/impl/Settlement.sol

@@ -15,6 +15,7 @@ import {Types} from "../lib/Types.sol";
 import {IERC20} from "../intf/IERC20.sol";
 import {Storage} from "./Storage.sol";
 
+
 /**
  * @title Settlement
  * @author DODO Breeder
@@ -34,11 +35,16 @@ contract Settlement is Storage {
     // ============ Assets IN/OUT Functions ============
 
     function _baseTokenTransferIn(address from, uint256 amount) internal {
+        require(_BASE_BALANCE_.add(amount) <= _BASE_BALANCE_LIMIT_, "BASE_BALANCE_LIMIT_EXCEEDED");
         IERC20(_BASE_TOKEN_).safeTransferFrom(from, address(this), amount);
         _BASE_BALANCE_ = _BASE_BALANCE_.add(amount);
     }
 
     function _quoteTokenTransferIn(address from, uint256 amount) internal {
+        require(
+            _QUOTE_BALANCE_.add(amount) <= _QUOTE_BALANCE_LIMIT_,
+            "QUOTE_BALANCE_LIMIT_EXCEEDED"
+        );
         IERC20(_QUOTE_TOKEN_).safeTransferFrom(from, address(this), amount);
         _QUOTE_BALANCE_ = _QUOTE_BALANCE_.add(amount);
     }

contracts/impl/Storage.sol

@@ -16,6 +16,7 @@ import {IOracle} from "../intf/IOracle.sol";
 import {IDODOLpToken} from "../intf/IDODOLpToken.sol";
 import {Types} from "../lib/Types.sol";
 
+
 /**
  * @title Storage
  * @author DODO Breeder
@@ -34,6 +35,12 @@ contract Storage is InitializableOwnable, ReentrancyGuard {
     bool public _TRADE_ALLOWED_;
     uint256 public _GAS_PRICE_LIMIT_;
 
+    // ============ Advanced Controls ============
+    bool public _BUYING_ALLOWED_;
+    bool public _SELLING_ALLOWED_;
+    uint256 public _BASE_BALANCE_LIMIT_;
+    uint256 public _QUOTE_BALANCE_LIMIT_;
+
     // ============ Core Address ============
 
     address public _SUPERVISOR_; // could freeze system in emergency
@@ -106,6 +113,6 @@ contract Storage is InitializableOwnable, ReentrancyGuard {
 
     // ============ Version Control ============
     function version() external pure returns (uint256) {
-        return 100; // 1.0.0
+        return 101; // 1.0.1
     }
 }

contracts/impl/Trader.sol

@@ -16,6 +16,7 @@ import {Storage} from "./Storage.sol";
 import {Pricing} from "./Pricing.sol";
 import {Settlement} from "./Settlement.sol";
 
+
 /**
  * @title Trader
  * @author DODO Breeder
@@ -40,6 +41,16 @@ contract Trader is Storage, Pricing, Settlement {
         _;
     }
 
+    modifier buyingAllowed() {
+        require(_BUYING_ALLOWED_, "BUYING_NOT_ALLOWED");
+        _;
+    }
+
+    modifier sellingAllowed() {
+        require(_SELLING_ALLOWED_, "SELLING_NOT_ALLOWED");
+        _;
+    }
+
     modifier gasPriceLimit() {
         require(tx.gasprice <= _GAS_PRICE_LIMIT_, "GAS_PRICE_EXCEED");
         _;
@@ -51,7 +62,7 @@ contract Trader is Storage, Pricing, Settlement {
         uint256 amount,
         uint256 minReceiveQuote,
         bytes calldata data
-    ) external tradeAllowed gasPriceLimit preventReentrant returns (uint256) {
+    ) external tradeAllowed sellingAllowed gasPriceLimit preventReentrant returns (uint256) {
         // query price
         (
             uint256 receiveQuote,
@@ -95,7 +106,7 @@ contract Trader is Storage, Pricing, Settlement {
         uint256 amount,
         uint256 maxPayQuote,
         bytes calldata data
-    ) external tradeAllowed gasPriceLimit preventReentrant returns (uint256) {
+    ) external tradeAllowed buyingAllowed gasPriceLimit preventReentrant returns (uint256) {
         // query price
         (
             uint256 payQuote,

test/Admin.test.ts

@@ -5,336 +5,606 @@
 
 */
 
+import * as assert from 'assert';
+
 import { DODOContext, getDODOContext } from './utils/Context';
 import { decimalStr } from './utils/Converter';
-// import BigNumber from "bignumber.js";
-import * as assert from "assert"
 
-let lp1: string
+let lp1: string;
-let lp2: string
+let lp2: string;
-let trader: string
+let trader: string;
-let tempAccount: string
+let tempAccount: string;
 
 async function init(ctx: DODOContext): Promise<void> {
-  await ctx.setOraclePrice(decimalStr("100"))
+  await ctx.setOraclePrice(decimalStr("100"));
-  tempAccount = ctx.spareAccounts[5]
+  tempAccount = ctx.spareAccounts[5];
-  lp1 = ctx.spareAccounts[0]
+  lp1 = ctx.spareAccounts[0];
-  lp2 = ctx.spareAccounts[1]
+  lp2 = ctx.spareAccounts[1];
-  trader = ctx.spareAccounts[2]
+  trader = ctx.spareAccounts[2];
-  await ctx.mintTestToken(lp1, decimalStr("100"), decimalStr("10000"))
+  await ctx.mintTestToken(lp1, decimalStr("100"), decimalStr("10000"));
-  await ctx.mintTestToken(lp2, decimalStr("100"), decimalStr("10000"))
+  await ctx.mintTestToken(lp2, decimalStr("100"), decimalStr("10000"));
-  await ctx.mintTestToken(trader, decimalStr("100"), decimalStr("10000"))
+  await ctx.mintTestToken(trader, decimalStr("100"), decimalStr("10000"));
-  await ctx.approveDODO(lp1)
+  await ctx.approveDODO(lp1);
-  await ctx.approveDODO(lp2)
+  await ctx.approveDODO(lp2);
-  await ctx.approveDODO(trader)
+  await ctx.approveDODO(trader);
 }
 
 describe("Admin", () => {
-
+  let snapshotId: string;
-  let snapshotId: string
+  let ctx: DODOContext;
-  let ctx: DODOContext
 
   before(async () => {
-    ctx = await getDODOContext()
+    ctx = await getDODOContext();
     await init(ctx);
-  })
+  });
 
   beforeEach(async () => {
     snapshotId = await ctx.EVM.snapshot();
   });
 
   afterEach(async () => {
-    await ctx.EVM.reset(snapshotId)
+    await ctx.EVM.reset(snapshotId);
   });
 
   describe("Settings", () => {
     it("set oracle", async () => {
-      await ctx.DODO.methods.setOracle(tempAccount).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._ORACLE_().call(), tempAccount)
+        .setOracle(tempAccount)
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._ORACLE_().call(), tempAccount);
+    });
 
     it("set suprevisor", async () => {
-      await ctx.DODO.methods.setSupervisor(tempAccount).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._SUPERVISOR_().call(), tempAccount)
+        .setSupervisor(tempAccount)
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._SUPERVISOR_().call(), tempAccount);
+    });
 
     it("set maintainer", async () => {
-      await ctx.DODO.methods.setMaintainer(tempAccount).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._MAINTAINER_().call(), tempAccount)
+        .setMaintainer(tempAccount)
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._MAINTAINER_().call(), tempAccount);
+    });
 
     it("set liquidity provider fee rate", async () => {
-      await ctx.DODO.methods.setLiquidityProviderFeeRate(decimalStr("0.01")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._LP_FEE_RATE_().call(), decimalStr("0.01"))
+        .setLiquidityProviderFeeRate(decimalStr("0.01"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(
+        await ctx.DODO.methods._LP_FEE_RATE_().call(),
+        decimalStr("0.01")
+      );
+    });
 
     it("set maintainer fee rate", async () => {
-      await ctx.DODO.methods.setMaintainerFeeRate(decimalStr("0.01")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._MT_FEE_RATE_().call(), decimalStr("0.01"))
+        .setMaintainerFeeRate(decimalStr("0.01"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(
+        await ctx.DODO.methods._MT_FEE_RATE_().call(),
+        decimalStr("0.01")
+      );
+    });
 
     it("set k", async () => {
-      await ctx.DODO.methods.setK(decimalStr("0.2")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._K_().call(), decimalStr("0.2"))
+        .setK(decimalStr("0.2"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._K_().call(), decimalStr("0.2"));
+    });
 
     it("set gas price limit", async () => {
-      await ctx.DODO.methods.setGasPriceLimit(decimalStr("100")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.DODO.methods._GAS_PRICE_LIMIT_().call(), decimalStr("100"))
+        .setGasPriceLimit(decimalStr("100"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
-  })
+      assert.equal(
+        await ctx.DODO.methods._GAS_PRICE_LIMIT_().call(),
+        decimalStr("100")
+      );
+    });
+  });
 
   describe("Controls", () => {
     it("control flow", async () => {
-      await ctx.DODO.methods.disableBaseDeposit().send(ctx.sendParam(ctx.Supervisor))
+      await ctx.DODO.methods
+        .disableBaseDeposit()
+        .send(ctx.sendParam(ctx.Supervisor));
       await assert.rejects(
         ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1)),
         /DEPOSIT_BASE_NOT_ALLOWED/
-      )
+      );
 
-      await ctx.DODO.methods.enableBaseDeposit().send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+        .enableBaseDeposit()
-      assert.equal(await ctx.DODO.methods._TARGET_BASE_TOKEN_AMOUNT_().call(), decimalStr("10"))
+        .send(ctx.sendParam(ctx.Deployer));
+      await ctx.DODO.methods
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      assert.equal(
+        await ctx.DODO.methods._TARGET_BASE_TOKEN_AMOUNT_().call(),
+        decimalStr("10")
+      );
 
-      await ctx.DODO.methods.disableQuoteDeposit().send(ctx.sendParam(ctx.Supervisor))
+      await ctx.DODO.methods
+        .disableQuoteDeposit()
+        .send(ctx.sendParam(ctx.Supervisor));
       await assert.rejects(
-        ctx.DODO.methods.depositQuote(decimalStr("1000")).send(ctx.sendParam(lp1)),
+        ctx.DODO.methods
+          .depositQuote(decimalStr("1000"))
+          .send(ctx.sendParam(lp1)),
         /DEPOSIT_QUOTE_NOT_ALLOWED/
-      )
+      );
 
-      await ctx.DODO.methods.enableQuoteDeposit().send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositQuote(decimalStr("10")).send(ctx.sendParam(lp1))
+        .enableQuoteDeposit()
-      assert.equal(await ctx.DODO.methods._TARGET_QUOTE_TOKEN_AMOUNT_().call(), decimalStr("10"))
+        .send(ctx.sendParam(ctx.Deployer));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      assert.equal(
+        await ctx.DODO.methods._TARGET_QUOTE_TOKEN_AMOUNT_().call(),
+        decimalStr("10")
+      );
 
-      await ctx.DODO.methods.disableTrading().send(ctx.sendParam(ctx.Supervisor))
+      await ctx.DODO.methods
+        .disableTrading()
+        .send(ctx.sendParam(ctx.Supervisor));
       await assert.rejects(
-        ctx.DODO.methods.buyBaseToken(decimalStr("1"), decimalStr("200"), "0x").send(ctx.sendParam(trader)),
+        ctx.DODO.methods
+          .buyBaseToken(decimalStr("1"), decimalStr("200"), "0x")
+          .send(ctx.sendParam(trader)),
         /TRADE_NOT_ALLOWED/
-      )
+      );
 
-      await ctx.DODO.methods.enableTrading().send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods.enableTrading().send(ctx.sendParam(ctx.Deployer));
-      await ctx.DODO.methods.buyBaseToken(decimalStr("1"), decimalStr("200"), "0x").send(ctx.sendParam(trader))
+      await ctx.DODO.methods
-      assert.equal(await ctx.BASE.methods.balanceOf(trader).call(), decimalStr("101"))
+        .buyBaseToken(decimalStr("1"), decimalStr("200"), "0x")
-    })
+        .send(ctx.sendParam(trader));
+      assert.equal(
+        await ctx.BASE.methods.balanceOf(trader).call(),
+        decimalStr("101")
+      );
+    });
 
     it("control flow premission", async () => {
       await assert.rejects(
         ctx.DODO.methods.setGasPriceLimit("1").send(ctx.sendParam(trader)),
         /NOT_SUPERVISOR_OR_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.disableTrading().send(ctx.sendParam(trader)),
         /NOT_SUPERVISOR_OR_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.disableQuoteDeposit().send(ctx.sendParam(trader)),
         /NOT_SUPERVISOR_OR_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.disableBaseDeposit().send(ctx.sendParam(trader)),
         /NOT_SUPERVISOR_OR_OWNER/
-      )
+      );
+      await assert.rejects(
+        ctx.DODO.methods.disableBuying().send(ctx.sendParam(trader)),
+        /NOT_SUPERVISOR_OR_OWNER/
+      );
+      await assert.rejects(
+        ctx.DODO.methods.disableSelling().send(ctx.sendParam(trader)),
+        /NOT_SUPERVISOR_OR_OWNER/
+      );
 
       await assert.rejects(
         ctx.DODO.methods.setOracle(trader).send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.setSupervisor(trader).send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.setMaintainer(trader).send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
-        ctx.DODO.methods.setLiquidityProviderFeeRate(decimalStr("0.1")).send(ctx.sendParam(trader)),
+        ctx.DODO.methods
+          .setLiquidityProviderFeeRate(decimalStr("0.1"))
+          .send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
-        ctx.DODO.methods.setMaintainerFeeRate(decimalStr("0.1")).send(ctx.sendParam(trader)),
+        ctx.DODO.methods
+          .setMaintainerFeeRate(decimalStr("0.1"))
+          .send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.setK(decimalStr("0.1")).send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
 
       await assert.rejects(
         ctx.DODO.methods.enableTrading().send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.enableQuoteDeposit().send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.enableBaseDeposit().send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
-    })
+      await assert.rejects(
-  })
+        ctx.DODO.methods.enableBuying().send(ctx.sendParam(trader)),
+        /NOT_OWNER/
+      );
+      await assert.rejects(
+        ctx.DODO.methods.enableSelling().send(ctx.sendParam(trader)),
+        /NOT_OWNER/
+      );
+      await assert.rejects(
+        ctx.DODO.methods
+          .setBaseBalanceLimit(decimalStr("0"))
+          .send(ctx.sendParam(trader)),
+        /NOT_OWNER/
+      );
+      await assert.rejects(
+        ctx.DODO.methods
+          .setQuoteBalanceLimit(decimalStr("0"))
+          .send(ctx.sendParam(trader)),
+        /NOT_OWNER/
+      );
+      await assert.rejects(
+        ctx.DODO.methods.enableTrading().send(ctx.sendParam(trader)),
+        /NOT_OWNER/
+      );
+    });
+
+    it("advanced controls", async () => {
+      await ctx.DODO.methods
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+
+      await ctx.DODO.methods
+        .disableBuying()
+        .send(ctx.sendParam(ctx.Supervisor));
+      await assert.rejects(
+        ctx.DODO.methods
+          .buyBaseToken(decimalStr("1"), decimalStr("200"), "0x")
+          .send(ctx.sendParam(trader)),
+        /BUYING_NOT_ALLOWED/
+      );
+      await ctx.DODO.methods.enableBuying().send(ctx.sendParam(ctx.Deployer));
+
+      await ctx.DODO.methods
+        .disableSelling()
+        .send(ctx.sendParam(ctx.Supervisor));
+      await assert.rejects(
+        ctx.DODO.methods
+          .sellBaseToken(decimalStr("1"), decimalStr("200"), "0x")
+          .send(ctx.sendParam(trader)),
+        /SELLING_NOT_ALLOWED/
+      );
+      await ctx.DODO.methods.enableSelling().send(ctx.sendParam(ctx.Deployer));
+
+      await ctx.DODO.methods
+        .setBaseBalanceLimit(decimalStr("0"))
+        .send(ctx.sendParam(ctx.Deployer));
+      await assert.rejects(
+        ctx.DODO.methods
+          .depositBase(decimalStr("1000"))
+          .send(ctx.sendParam(lp1)),
+        /BASE_BALANCE_LIMIT_EXCEEDED/
+      );
+
+      await ctx.DODO.methods
+        .setQuoteBalanceLimit(decimalStr("0"))
+        .send(ctx.sendParam(ctx.Deployer));
+      await assert.rejects(
+        ctx.DODO.methods
+          .depositQuote(decimalStr("1000"))
+          .send(ctx.sendParam(lp1)),
+        /QUOTE_BALANCE_LIMIT_EXCEEDED/
+      );
+    });
+  });
 
   describe("Final settlement", () => {
     it("final settlement when R is ONE", async () => {
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositQuote(decimalStr("1000")).send(ctx.sendParam(lp1))
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("1000"))
+        .send(ctx.sendParam(lp1));
 
-      await ctx.DODO.methods.finalSettlement().send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
+        .finalSettlement()
+        .send(ctx.sendParam(ctx.Deployer));
 
-      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1));
-      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1));
-      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1));
 
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), decimalStr("100"))
+      assert.equal(
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), decimalStr("10000"))
+        await ctx.BASE.methods.balanceOf(lp1).call(),
-    })
+        decimalStr("100")
+      );
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
+        decimalStr("10000")
+      );
+    });
 
     it("final settlement when R is ABOVE ONE", async () => {
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositQuote(decimalStr("1000")).send(ctx.sendParam(lp1))
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("1000"))
+        .send(ctx.sendParam(lp1));
 
-      await ctx.DODO.methods.buyBaseToken(decimalStr("5"), decimalStr("1000"), "0x").send(ctx.sendParam(trader))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.finalSettlement().send(ctx.sendParam(ctx.Deployer))
+        .buyBaseToken(decimalStr("5"), decimalStr("1000"), "0x")
-      assert.equal(await ctx.DODO.methods._R_STATUS_().call(), "0")
+        .send(ctx.sendParam(trader));
+      await ctx.DODO.methods
+        .finalSettlement()
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._R_STATUS_().call(), "0");
 
-      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1));
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), decimalStr("90"))
+      assert.equal(
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), "9551951805416248746110")
+        await ctx.BASE.methods.balanceOf(lp1).call(),
-      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1))
+        decimalStr("90")
-      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1))
+      );
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), decimalStr("94.995"))
+      assert.equal(
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), "10551951805416248746110")
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
-    })
+        "9551951805416248746110"
+      );
+      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1));
+      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1));
+      assert.equal(
+        await ctx.BASE.methods.balanceOf(lp1).call(),
+        decimalStr("94.995")
+      );
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
+        "10551951805416248746110"
+      );
+    });
 
     it("final settlement when R is BELOW ONE", async () => {
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositQuote(decimalStr("1000")).send(ctx.sendParam(lp1))
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("1000"))
+        .send(ctx.sendParam(lp1));
 
-      await ctx.DODO.methods.sellBaseToken(decimalStr("5"), decimalStr("100"), "0x").send(ctx.sendParam(trader))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.finalSettlement().send(ctx.sendParam(ctx.Deployer))
+        .sellBaseToken(decimalStr("5"), decimalStr("100"), "0x")
-      assert.equal(await ctx.DODO.methods._R_STATUS_().call(), "0")
+        .send(ctx.sendParam(trader));
+      await ctx.DODO.methods
+        .finalSettlement()
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(await ctx.DODO.methods._R_STATUS_().call(), "0");
 
-      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1));
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), decimalStr("95"))
+      assert.equal(
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), decimalStr("9000"))
+        await ctx.BASE.methods.balanceOf(lp1).call(),
-      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1))
+        decimalStr("95")
-      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1))
+      );
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), decimalStr("105"))
+      assert.equal(
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), "9540265973590798352835")
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
-    })
+        decimalStr("9000")
+      );
+      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1));
+      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1));
+      assert.equal(
+        await ctx.BASE.methods.balanceOf(lp1).call(),
+        decimalStr("105")
+      );
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
+        "9540265973590798352835"
+      );
+    });
 
     it("final settlement revert cases", async () => {
       await assert.rejects(
         ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp1)),
         /DODO_NOT_CLOSED/
-      )
+      );
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositQuote(decimalStr("500")).send(ctx.sendParam(lp2))
+        .depositBase(decimalStr("10"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("500"))
+        .send(ctx.sendParam(lp2));
 
-      await ctx.DODO.methods.buyBaseToken(decimalStr("5"), decimalStr("1000"), "0x").send(ctx.sendParam(trader))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.finalSettlement().send(ctx.sendParam(ctx.Deployer))
+        .buyBaseToken(decimalStr("5"), decimalStr("1000"), "0x")
+        .send(ctx.sendParam(trader));
+      await ctx.DODO.methods
+        .finalSettlement()
+        .send(ctx.sendParam(ctx.Deployer));
       await assert.rejects(
         ctx.DODO.methods.finalSettlement().send(ctx.sendParam(ctx.Deployer)),
         / DODO_CLOSED/
-      )
+      );
 
-      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp2))
+      await ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp2));
       await assert.rejects(
         ctx.DODO.methods.claimAssets().send(ctx.sendParam(lp2)),
         /ALREADY_CLAIMED/
-      )
+      );
 
       await assert.rejects(
         ctx.DODO.methods.enableQuoteDeposit().send(ctx.sendParam(ctx.Deployer)),
         /DODO_CLOSED/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.enableBaseDeposit().send(ctx.sendParam(ctx.Deployer)),
         /DODO_CLOSED/
-      )
+      );
       await assert.rejects(
         ctx.DODO.methods.enableTrading().send(ctx.sendParam(ctx.Deployer)),
         /DODO_CLOSED/
-      )
+      );
-    })
+      await assert.rejects(
-  })
+        ctx.DODO.methods.enableBuying().send(ctx.sendParam(ctx.Deployer)),
+        /DODO_CLOSED/
+      );
+      await assert.rejects(
+        ctx.DODO.methods.enableSelling().send(ctx.sendParam(ctx.Deployer)),
+        /DODO_CLOSED/
+      );
+    });
+  });
 
   describe("donate", () => {
     it("donate quote & base token", async () => {
-      await ctx.DODO.methods.depositBase(decimalStr("10")).send(ctx.sendParam(lp1))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.depositBase(decimalStr("20")).send(ctx.sendParam(lp2))
+        .depositBase(decimalStr("10"))
-      await ctx.DODO.methods.depositQuote(decimalStr("1000")).send(ctx.sendParam(lp1))
+        .send(ctx.sendParam(lp1));
-      await ctx.DODO.methods.depositQuote(decimalStr("2000")).send(ctx.sendParam(lp2))
+      await ctx.DODO.methods
+        .depositBase(decimalStr("20"))
+        .send(ctx.sendParam(lp2));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("1000"))
+        .send(ctx.sendParam(lp1));
+      await ctx.DODO.methods
+        .depositQuote(decimalStr("2000"))
+        .send(ctx.sendParam(lp2));
 
-      await ctx.DODO.methods.donateBaseToken(decimalStr("2")).send(ctx.sendParam(trader))
+      await ctx.DODO.methods
-      await ctx.DODO.methods.donateQuoteToken(decimalStr("500")).send(ctx.sendParam(trader))
+        .donateBaseToken(decimalStr("2"))
+        .send(ctx.sendParam(trader));
+      await ctx.DODO.methods
+        .donateQuoteToken(decimalStr("500"))
+        .send(ctx.sendParam(trader));
 
-      assert.equal(await ctx.DODO.methods.getLpBaseBalance(lp1).call(), "10666666666666666666")
+      assert.equal(
-      assert.equal(await ctx.DODO.methods.getLpQuoteBalance(lp1).call(), "1166666666666666666666")
+        await ctx.DODO.methods.getLpBaseBalance(lp1).call(),
-      assert.equal(await ctx.DODO.methods.getLpBaseBalance(lp2).call(), "21333333333333333333")
+        "10666666666666666666"
-      assert.equal(await ctx.DODO.methods.getLpQuoteBalance(lp2).call(), "2333333333333333333333")
+      );
+      assert.equal(
+        await ctx.DODO.methods.getLpQuoteBalance(lp1).call(),
+        "1166666666666666666666"
+      );
+      assert.equal(
+        await ctx.DODO.methods.getLpBaseBalance(lp2).call(),
+        "21333333333333333333"
+      );
+      assert.equal(
+        await ctx.DODO.methods.getLpQuoteBalance(lp2).call(),
+        "2333333333333333333333"
+      );
 
-      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp1));
-      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp2))
+      await ctx.DODO.methods.withdrawAllBase().send(ctx.sendParam(lp2));
 
-      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1))
+      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp1));
-      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp2))
+      await ctx.DODO.methods.withdrawAllQuote().send(ctx.sendParam(lp2));
 
-      assert.equal(await ctx.BASE.methods.balanceOf(lp1).call(), "100666666666666666666")
+      assert.equal(
-      assert.equal(await ctx.BASE.methods.balanceOf(lp2).call(), "101333333333333333334")
+        await ctx.BASE.methods.balanceOf(lp1).call(),
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp1).call(), "10166666666666666666666")
+        "100666666666666666666"
-      assert.equal(await ctx.QUOTE.methods.balanceOf(lp2).call(), "10333333333333333333334")
+      );
-    })
+      assert.equal(
-  })
+        await ctx.BASE.methods.balanceOf(lp2).call(),
+        "101333333333333333334"
+      );
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(lp1).call(),
+        "10166666666666666666666"
+      );
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(lp2).call(),
+        "10333333333333333333334"
+      );
+    });
+  });
 
   describe("retrieve", () => {
     it("retrieve base token", async () => {
-      await ctx.BASE.methods.transfer(ctx.DODO.options.address, decimalStr("1")).send(ctx.sendParam(trader))
+      await ctx.BASE.methods
+        .transfer(ctx.DODO.options.address, decimalStr("1"))
+        .send(ctx.sendParam(trader));
       await assert.rejects(
-        ctx.DODO.methods.retrieve(ctx.BASE.options.address, decimalStr("1")).send(ctx.sendParam(trader)),
+        ctx.DODO.methods
+          .retrieve(ctx.BASE.options.address, decimalStr("1"))
+          .send(ctx.sendParam(trader)),
         /NOT_OWNER/
-      )
+      );
       await assert.rejects(
-        ctx.DODO.methods.retrieve(ctx.BASE.options.address, decimalStr("2")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .retrieve(ctx.BASE.options.address, decimalStr("2"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /DODO_BASE_BALANCE_NOT_ENOUGH/
-      )
+      );
-      await ctx.DODO.methods.retrieve(ctx.BASE.options.address, decimalStr("1")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.BASE.methods.balanceOf(ctx.Deployer).call(), decimalStr("1"))
+        .retrieve(ctx.BASE.options.address, decimalStr("1"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
+      assert.equal(
+        await ctx.BASE.methods.balanceOf(ctx.Deployer).call(),
+        decimalStr("1")
+      );
+    });
 
     it("retrieve quote token", async () => {
-      await ctx.QUOTE.methods.transfer(ctx.DODO.options.address, decimalStr("1")).send(ctx.sendParam(trader))
+      await ctx.QUOTE.methods
+        .transfer(ctx.DODO.options.address, decimalStr("1"))
+        .send(ctx.sendParam(trader));
       await assert.rejects(
-        ctx.DODO.methods.retrieve(ctx.QUOTE.options.address, decimalStr("2")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .retrieve(ctx.QUOTE.options.address, decimalStr("2"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /DODO_QUOTE_BALANCE_NOT_ENOUGH/
-      )
+      );
-      await ctx.DODO.methods.retrieve(ctx.QUOTE.options.address, decimalStr("1")).send(ctx.sendParam(ctx.Deployer))
+      await ctx.DODO.methods
-      assert.equal(await ctx.QUOTE.methods.balanceOf(ctx.Deployer).call(), decimalStr("1"))
+        .retrieve(ctx.QUOTE.options.address, decimalStr("1"))
-    })
+        .send(ctx.sendParam(ctx.Deployer));
-  })
+      assert.equal(
+        await ctx.QUOTE.methods.balanceOf(ctx.Deployer).call(),
+        decimalStr("1")
+      );
+    });
+  });
 
   describe("revert cases", () => {
     it("k revert cases", async () => {
       await assert.rejects(
-        ctx.DODO.methods.setK(decimalStr("1")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .setK(decimalStr("1"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /K>=1/
-      )
+      );
       await assert.rejects(
-        ctx.DODO.methods.setK(decimalStr("0")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .setK(decimalStr("0"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /K=0/
-      )
+      );
-    })
+    });
 
     it("fee revert cases", async () => {
       await assert.rejects(
-        ctx.DODO.methods.setLiquidityProviderFeeRate(decimalStr("0.999")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .setLiquidityProviderFeeRate(decimalStr("0.999"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /FEE_RATE>=1/
-      )
+      );
       await assert.rejects(
-        ctx.DODO.methods.setMaintainerFeeRate(decimalStr("0.998")).send(ctx.sendParam(ctx.Deployer)),
+        ctx.DODO.methods
+          .setMaintainerFeeRate(decimalStr("0.998"))
+          .send(ctx.sendParam(ctx.Deployer)),
         /FEE_RATE>=1/
-      )
+      );
-    })
+    });
-  })
+  });
-})
+});

test/DODOEthProxyAsBase.test.ts

@@ -44,6 +44,9 @@ async function init(ctx: DODOContext): Promise<void> {
       .getDODO(WETH.options.address, ctx.QUOTE.options.address)
       .call()
   );
+  await ctx.DODO.methods.enableBaseDeposit().send(ctx.sendParam(ctx.Deployer));
+  await ctx.DODO.methods.enableQuoteDeposit().send(ctx.sendParam(ctx.Deployer));
+  await ctx.DODO.methods.enableTrading().send(ctx.sendParam(ctx.Deployer));
 
   ctx.BASE = WETH;
 
@@ -102,12 +105,11 @@ describe("DODO ETH PROXY", () => {
     it("buy", async () => {
       const buyAmount = "1";
       await logGas(
-        DODOEthProxy.methods
+        DODOEthProxy.methods.buyEthWithToken(
-          .buyEthWithToken(
+          ctx.QUOTE.options.address,
-            ctx.QUOTE.options.address,
+          decimalStr(buyAmount),
-            decimalStr(buyAmount),
+          decimalStr("200")
-            decimalStr("200")
+        ),
-          ),
         ctx.sendParam(trader),
         "buy ETH with token directly"
       );
@@ -123,12 +125,11 @@ describe("DODO ETH PROXY", () => {
     it("sell", async () => {
       const sellAmount = "1";
       await logGas(
-        DODOEthProxy.methods
+        DODOEthProxy.methods.sellEthToToken(
-          .sellEthToToken(
+          ctx.QUOTE.options.address,
-            ctx.QUOTE.options.address,
+          decimalStr(sellAmount),
-            decimalStr(sellAmount),
+          decimalStr("50")
-            decimalStr("50")
+        ),
-          ),
         ctx.sendParam(trader, sellAmount),
         "sell ETH to token directly"
       );

test/DODOEthProxyAsQuote.test.ts

@@ -44,6 +44,9 @@ async function init(ctx: DODOContext): Promise<void> {
       .getDODO(ctx.BASE.options.address, WETH.options.address)
       .call()
   );
+  await ctx.DODO.methods.enableBaseDeposit().send(ctx.sendParam(ctx.Deployer));
+  await ctx.DODO.methods.enableQuoteDeposit().send(ctx.sendParam(ctx.Deployer));
+  await ctx.DODO.methods.enableTrading().send(ctx.sendParam(ctx.Deployer));
 
   ctx.QUOTE = WETH;
 
@@ -103,12 +106,15 @@ describe("DODO ETH PROXY", () => {
       const maxPayEthAmount = "2.1";
       const ethInPoolBefore = decimalStr("10");
       const traderEthBalanceBefore = await ctx.Web3.eth.getBalance(trader);
-      const txReceipt: TransactionReceipt = await logGas(DODOEthProxy.methods
+      const txReceipt: TransactionReceipt = await logGas(
-        .buyTokenWithEth(
+        DODOEthProxy.methods.buyTokenWithEth(
           ctx.BASE.options.address,
           decimalStr("200"),
           decimalStr(maxPayEthAmount)
-        ), ctx.sendParam(trader, maxPayEthAmount), "buy token with ETH directly");
+        ),
+        ctx.sendParam(trader, maxPayEthAmount),
+        "buy token with ETH directly"
+      );
       const ethInPoolAfter = "12056338203652739553";
       assert.strictEqual(
         await ctx.DODO.methods._QUOTE_BALANCE_().call(),
@@ -133,12 +139,11 @@ describe("DODO ETH PROXY", () => {
     it("sell", async () => {
       const minReceiveEthAmount = "0.45";
       await logGas(
-        DODOEthProxy.methods
+        DODOEthProxy.methods.sellTokenToEth(
-          .sellTokenToEth(
+          ctx.BASE.options.address,
-            ctx.BASE.options.address,
+          decimalStr("50"),
-            decimalStr("50"),
+          decimalStr(minReceiveEthAmount)
-            decimalStr(minReceiveEthAmount)
+        ),
-          ),
         ctx.sendParam(trader),
         "sell token to ETH directly"
       );

test/utils/Context.ts

@@ -5,13 +5,14 @@
 
 */
 
-import { EVM, getDefaultWeb3 } from "./EVM";
+import BigNumber from 'bignumber.js';
-import Web3 from "web3";
+import Web3 from 'web3';
-import { Contract } from "web3-eth-contract";
+import { Contract } from 'web3-eth-contract';
-import BigNumber from "bignumber.js";
+
-import * as contracts from "./Contracts";
+import * as contracts from './Contracts';
-import { decimalStr, gweiStr, MAX_UINT256 } from "./Converter";
+import { decimalStr, gweiStr, MAX_UINT256 } from './Converter';
-import * as log from "./Log";
+import { EVM, getDefaultWeb3 } from './EVM';
+import * as log from './Log';
 
 BigNumber.config({
   EXPONENTIAL_AT: 1000,
@@ -19,10 +20,10 @@ BigNumber.config({
 });
 
 export interface DODOContextInitConfig {
-  lpFeeRate: string,
+  lpFeeRate: string;
-  mtFeeRate: string,
+  mtFeeRate: string;
-  k: string,
+  k: string;
-  gasPriceLimit: string,
+  gasPriceLimit: string;
 }
 
 /*
@@ -43,60 +44,99 @@ export let DefaultDODOContextInitConfig = {
   mtFeeRate: decimalStr("0.001"),
   k: decimalStr("0.1"),
   gasPriceLimit: gweiStr("100"),
-}
+};
 
 export class DODOContext {
-  EVM: EVM
+  EVM: EVM;
-  Web3: Web3
+  Web3: Web3;
-  DODO: Contract
+  DODO: Contract;
-  DODOZoo: Contract
+  DODOZoo: Contract;
-  BASE: Contract
+  BASE: Contract;
-  BaseCapital: Contract
+  BaseCapital: Contract;
-  QUOTE: Contract
+  QUOTE: Contract;
-  QuoteCapital: Contract
+  QuoteCapital: Contract;
-  ORACLE: Contract
+  ORACLE: Contract;
-  Deployer: string
+  Deployer: string;
-  Supervisor: string
+  Supervisor: string;
-  Maintainer: string
+  Maintainer: string;
-  spareAccounts: string[]
+  spareAccounts: string[];
 
-  constructor() { }
+  constructor() {}
 
   async init(config: DODOContextInitConfig) {
-    this.EVM = new EVM
+    this.EVM = new EVM();
-    this.Web3 = getDefaultWeb3()
+    this.Web3 = getDefaultWeb3();
-    var cloneFactory = await contracts.newContract(contracts.CLONE_FACTORY_CONTRACT_NAME)
+    var cloneFactory = await contracts.newContract(
+      contracts.CLONE_FACTORY_CONTRACT_NAME
+    );
 
-    this.BASE = await contracts.newContract(contracts.TEST_ERC20_CONTRACT_NAME, ["TestBase", 18])
+    this.BASE = await contracts.newContract(
-    this.QUOTE = await contracts.newContract(contracts.TEST_ERC20_CONTRACT_NAME, ["TestQuote", 18])
+      contracts.TEST_ERC20_CONTRACT_NAME,
-    this.ORACLE = await contracts.newContract(contracts.NAIVE_ORACLE_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.Deployer = allAccounts[0];
-    this.Supervisor = allAccounts[1]
+    this.Supervisor = allAccounts[1];
-    this.Maintainer = allAccounts[2]
+    this.Maintainer = allAccounts[2];
-    this.spareAccounts = allAccounts.slice(3, 10)
+    this.spareAccounts = allAccounts.slice(3, 10);
 
-    var DODOTemplate = await contracts.newContract(contracts.DODO_CONTRACT_NAME)
+    var DODOTemplate = await contracts.newContract(
-    this.DODOZoo = await contracts.newContract(contracts.DODO_ZOO_CONTRACT_NAME, [DODOTemplate.options.address, cloneFactory.options.address, this.Supervisor])
+      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(
+    await this.DODOZoo.methods
-      this.Maintainer,
+      .breedDODO(
-      this.BASE.options.address,
+        this.Maintainer,
-      this.QUOTE.options.address,
+        this.BASE.options.address,
-      this.ORACLE.options.address,
+        this.QUOTE.options.address,
-      config.lpFeeRate,
+        this.ORACLE.options.address,
-      config.mtFeeRate,
+        config.lpFeeRate,
-      config.k,
+        config.mtFeeRate,
-      config.gasPriceLimit
+        config.k,
-    ).send(this.sendParam(this.Deployer))
+        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())
+    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.BaseCapital = contracts.getContractWithAddress(
-    this.QuoteCapital = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, await this.DODO.methods._QUOTE_CAPITAL_TOKEN_().call())
+      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]"))
+    console.log(log.blueText("[Init dodo context]"));
   }
 
   sendParam(sender, value = "0") {
@@ -104,27 +144,37 @@ export class DODOContext {
       from: sender,
       gas: process.env["COVERAGE"] ? 10000000000 : 7000000,
       gasPrice: process.env.GAS_PRICE,
-      value: decimalStr(value)
+      value: decimalStr(value),
-    }
+    };
   }
 
   async setOraclePrice(price: string) {
-    await this.ORACLE.methods.setPrice(price).send(this.sendParam(this.Deployer))
+    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.BASE.methods.mint(to, base).send(this.sendParam(this.Deployer));
-    await this.QUOTE.methods.mint(to, quote).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.BASE.methods
-    await this.QUOTE.methods.approve(this.DODO.options.address, MAX_UINT256).send(this.sendParam(account))
+      .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> {
+export async function getDODOContext(
-  var context = new DODOContext()
+  config: DODOContextInitConfig = DefaultDODOContextInitConfig
-  await context.init(config)
+): Promise<DODOContext> {
-  return context
+  var context = new DODOContext();
-}
+  await context.init(config);
+  return context;
+}