@dhedge/v2-sdk 1.9.2 → 1.9.4

package/src/test/utils/testingHelper.ts

@@ -1,10 +1,11 @@
 import { ethers } from "ethers";
 import { Network } from "../../types";
-import { getWallet } from "../wallet";
+import { getWalletData } from "../wallet";
 
 export type TestingRunParams = {
   network: Network;
   wallet: ethers.Wallet;
+  provider: ethers.providers.JsonRpcProvider;
 };
 
 type TestHelperParams = {
@@ -15,6 +16,27 @@ export const testingHelper = ({
   network,
   testingRun
 }: TestHelperParams): void => {
-  const wallet = getWallet(network);
-  testingRun({ network, wallet });
+  const { wallet, provider } = getWalletData(network);
+  testingRun({ network, wallet, provider });
+};
+
+export const beforeAfterReset = ({
+  beforeAll,
+  afterAll,
+  provider
+}: {
+  beforeAll: jest.Lifecycle;
+  afterAll: jest.Lifecycle;
+  provider: ethers.providers.JsonRpcProvider;
+}): void => {
+  let snapshot = "";
+  beforeAll(async () => {
+    snapshot = (await provider.send("evm_snapshot", [])) as string;
+    await provider.send("evm_mine", []);
+  });
+
+  afterAll(async () => {
+    await provider.send("evm_revert", [snapshot]);
+    await provider.send("evm_mine", []);
+  });
 };

package/src/test/utils.test.ts

@@ -1,61 +1,82 @@
 //import { ethers } from "ethers";
-import { Dhedge } from "..";
+import { Dhedge, Pool } from "..";
 import { Network } from "../types";
+import { CONTRACT_ADDRESS, TEST_POOL } from "./constants";
+import {
+  TestingRunParams,
+  beforeAfterReset,
+  testingHelper
+} from "./utils/testingHelper";
 
-import { wallet } from "./wallet";
-
-const myPool = "0xe3528a438b94e64669def9b875c381c46ef713bf";
+// const myPool = "0xe3528a438b94e64669def9b875c381c46ef713bf";
 
 // const usdt = "0xc2132D05D31c914a87C6611C10748AEb04B58e8F";
-const usdc = "0x2791bca1f2de4661ed88a30c99a7a9449aa84174";
-const weth = "0x7ceB23fD6bC0adD59E62ac25578270cFf1b9f619";
-const wbtc = "0x1BFD67037B42Cf73acF2047067bd4F2C47D9BfD6";
+
 // const lpUsdcWeth = "0x34965ba0ac2451A34a0471F04CCa3F990b8dea27";
 
-let dhedge: Dhedge;
+const testUtils = ({ wallet, network, provider }: TestingRunParams) => {
+  let dhedge: Dhedge;
+  let pool: Pool;
+  const usdc = CONTRACT_ADDRESS[network].USDC;
+  const weth = CONTRACT_ADDRESS[network].WETH;
+  const wbtc = CONTRACT_ADDRESS[network].WBTC;
+  // const usdt = CONTRACT_ADDRESS[network].USDT;
 
-jest.setTimeout(100000);
+  jest.setTimeout(100000);
 
-describe("utils", () => {
-  beforeAll(() => {
-    dhedge = new Dhedge(wallet, Network.POLYGON);
-  });
+  describe(`utils on ${network}`, () => {
+    beforeAll(async () => {
+      dhedge = new Dhedge(wallet, network);
+      pool = await dhedge.loadPool(TEST_POOL[network]);
+    });
+    beforeAfterReset({ beforeAll, afterAll, provider });
+
+    // it("gets lp ratio of the USDT/USDC pool", async () => {
+    //   const result = await dhedge.utils.getLpReserves(Dapp.SUSHISWAP, usdc, usdt);
+    //   expect(Number(result.assetA) / Number(result.assetB)).toBeGreaterThan(0.9);
+    // });
+
+    // it("gets pool id of sushi LP pool for USDC/WETH", async () => {
+    //   const result = await dhedge.utils.getLpPoolId(Dapp.SUSHISWAP, lpUsdcWeth);
+    //   expect(result).toBe(1);
+    // });
 
-  // it("gets lp ratio of the USDT/USDC pool", async () => {
-  //   const result = await dhedge.utils.getLpReserves(Dapp.SUSHISWAP, usdc, usdt);
-  //   expect(Number(result.assetA) / Number(result.assetB)).toBeGreaterThan(0.9);
-  // });
-
-  // it("gets pool id of sushi LP pool for USDC/WETH", async () => {
-  //   const result = await dhedge.utils.getLpPoolId(Dapp.SUSHISWAP, lpUsdcWeth);
-  //   expect(result).toBe(1);
-  // });
-
-  // it("gets USDC balance of a pool", async () => {
-  //   const result = await dhedge.utils.getBalance(usdc, myPool);
-  //   expect(result.gt(0));
-  // });
-  // it("gets minumum amount out of WETH for 1 USDC", async () => {
-  //   const result = await dhedge.utils.getMinAmountOut(
-  //     Dapp.SUSHISWAP,
-  //     usdc,
-  //     weth,
-  //     "1000000",
-  //     0.5
-  //   );
-  //   expect(result.gt(0));
-  // });
-
-  it("gets Balancer pool tx data", async () => {
-    const pool = await dhedge.loadPool(myPool);
-    const assets = [wbtc, usdc, weth];
-    const amounts = ["0", "1000000", "0"];
-    const result = await dhedge.utils.getBalancerJoinPoolTx(
-      pool,
-      "0x03cd191f589d12b0582a99808cf19851e468e6b500010000000000000000000a",
-      assets,
-      amounts
-    );
-    expect(result);
+    // it("gets USDC balance of a pool", async () => {
+    //   const result = await dhedge.utils.getBalance(usdc, myPool);
+    //   expect(result.gt(0));
+    // });
+    // it("gets minumum amount out of WETH for 1 USDC", async () => {
+    //   const result = await dhedge.utils.getMinAmountOut(
+    //     Dapp.SUSHISWAP,
+    //     usdc,
+    //     weth,
+    //     "1000000",
+    //     0.5
+    //   );
+    //   expect(result.gt(0));
+    // });
+
+    it("gets Balancer pool tx data", async () => {
+      // pool = await dhedge.loadPool(pool);
+      const assets = [wbtc, usdc, weth];
+      const amounts = ["0", "1000000", "0"];
+      const result = await dhedge.utils.getBalancerJoinPoolTx(
+        pool,
+        "0x03cd191f589d12b0582a99808cf19851e468e6b500010000000000000000000a",
+        assets,
+        amounts
+      );
+      expect(result);
+    });
   });
+};
+
+testingHelper({
+  network: Network.POLYGON,
+  testingRun: testUtils
+});
+
+testingHelper({
+  network: Network.OPTIMISM,
+  testingRun: testUtils
 });

package/src/test/wallet.ts

@@ -32,9 +32,17 @@ export const networkPortMap = {
   [Network.ARBITRUM]: 8540
 };
 
-export const getWallet = (network: Network): ethers.Wallet => {
+export const getWalletData = (
+  network: Network
+): {
+  wallet: ethers.Wallet;
+  provider: ethers.providers.JsonRpcProvider;
+} => {
   const provider = new ethers.providers.JsonRpcProvider(
     `http://127.0.0.1:${networkPortMap[network]}/`
   );
-  return new ethers.Wallet(process.env.PRIVATE_KEY as string, provider);
+  return {
+    wallet: new ethers.Wallet(process.env.PRIVATE_KEY as string, provider),
+    provider
+  };
 };

package/src/types.ts

@@ -4,7 +4,7 @@ import { BigNumber } from "ethers";
 export enum Network {
   POLYGON = "polygon",
   OPTIMISM = "optimism",
-  ARBITRUM = "arbitrum"
+  ARBITRUM = "arbitrum",
 }
 
 export enum Dapp {
@@ -21,7 +21,8 @@ export enum Dapp {
   VELODROME = "velodrome",
   VELODROMEV2 = "velodromeV2",
   LYRA = "lyra",
-  ZEROEX = "0x"
+  ZEROEX = "0x",
+  RAMSES = "ramses",
 }
 
 export enum Transaction {
@@ -43,7 +44,7 @@ export enum Transaction {
   BURN = "burn",
   SWAP_SYNTHS = "exchangeWithTracking",
   ADD_LIQUIDITY_STAKE = "addLiquidityAndStake",
-  REMOVE_LIQUIDITY_UNSTAKE = "removeLiquidityAndUnstake"
+  REMOVE_LIQUIDITY_UNSTAKE = "removeLiquidityAndUnstake",
 }
 
 export type AddressNetworkMap = Readonly<Record<Network, string>>;