@dhedge/v2-sdk 1.9.5 → 1.9.6

package/src/test/oneInch.test.ts

@@ -7,7 +7,8 @@ import { CONTRACT_ADDRESS, MAX_AMOUNT, TEST_POOL } from "./constants";
 import {
   TestingRunParams,
   setUSDCAmount,
-  testingHelper
+  testingHelper,
+  wait
 } from "./utils/testingHelper";
 import { allowanceDelta, balanceDelta } from "./utils/token";
 import { getTxOptions } from "./txOptions";
@@ -32,7 +33,7 @@ const testOneInch = ({ wallet, network, provider }: TestingRunParams) => {
       await provider.send("evm_mine", []);
       // top up USDC
       await setUSDCAmount({
-        amount: new BigNumber(100).times(1e18).toFixed(0),
+        amount: new BigNumber(2).times(1e6).toFixed(0),
         userAddress: pool.address,
         network,
         provider
@@ -50,7 +51,38 @@ const testOneInch = ({ wallet, network, provider }: TestingRunParams) => {
       await expect(usdcAllowanceDelta.gt(0));
     });
 
+    it("gets gas estimation for 2 USDC into WETH on 1Inch", async () => {
+      const gasEstimate = await pool.trade(
+        Dapp.ONEINCH,
+        USDC,
+        WETH,
+        "2000000",
+        0.5,
+        await getTxOptions(network),
+        true
+      );
+      expect(gasEstimate.gt(0));
+    });
+
+    it("gets error on gas estimation for 200 USDC into WETH on 1Inch", async () => {
+      await wait(1);
+      let gasEstimate = null;
+      try {
+        gasEstimate = await pool.trade(
+          Dapp.ONEINCH,
+          USDC,
+          WETH,
+          "200000000",
+          0.5,
+          await getTxOptions(network),
+          true
+        );
+      } catch (err) {}
+      expect(gasEstimate).toBeNull();
+    });
+
     it("trades 2 USDC into WETH on 1Inch", async () => {
+      await wait(1);
       await pool.trade(
         Dapp.ONEINCH,
         USDC,

package/src/test/utils/testingHelper.ts

@@ -90,3 +90,7 @@ export const setUSDCAmount = async ({
     slot: USDC_BALANCEOF_SLOT[network]
   });
 };
+
+export const wait = (seconds: number): Promise<void> => {
+  return new Promise(resolve => setTimeout(resolve, seconds * 1000));
+};

package/src/utils/contract.ts

@@ -3,7 +3,7 @@
 import set from "lodash/set";
 import { Interface } from "@ethersproject/abi";
 import { multiCallAddress } from "../config";
-import { ethers, Network } from "..";
+import { ethers, Network, Pool } from "..";
 
 export async function call(
   provider: ethers.Signer,
@@ -93,3 +93,19 @@ export class Multicaller {
     return obj;
   }
 }
+
+export const getPoolTxOrGasEstimate = async (
+  pool: Pool,
+  args: any[],
+  estimateGas: boolean
+): Promise<any> => {
+  if (estimateGas) {
+    return await pool.poolLogic.estimateGas.execTransaction(
+      args[0],
+      args[1],
+      args[2]
+    );
+  } else {
+    return await pool.poolLogic.execTransaction(args[0], args[1], args[2]);
+  }
+};