@ledgerhq/coin-aptos 1.9.5 → 1.10.0-nightly.0

package/lib-es/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,KAAK,EACV,OAAO,EACP,SAAS,EACT,iBAAiB,EACjB,oBAAoB,EACpB,uBAAuB,EACvB,0BAA0B,EAC3B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,cAAc,UAAU,CAAC;AACzB,cAAc,UAAU,CAAC;AAEzB,MAAM,MAAM,gBAAgB,GAAG,uBAAuB,GAAG;IACvD,KAAK,EAAE;QACL,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG,SAAS,CAAC;AAEvC,MAAM,MAAM,YAAY,GAAG,OAAO,CAAC;AAEnC,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,CAAC;AAExD,MAAM,MAAM,oBAAoB,GAAG,0BAA0B,CAAC;AAE9D,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,IAAI;IAC/D,IAAI,EAAE,CAAC,CAAC;IACR,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,MAAM,iBAAiB,GAAG;IAC9B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG;IAC5C,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,CAAC,EAAE,SAAS,GAAG,IAAI,CAAC;IACxB,OAAO,EAAE,kBAAkB,CAAC;IAC5B,MAAM,CAAC,EAAE,iBAAiB,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG,oBAAoB,GAAG;IAClD,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,CAAC,GAAG,EAAE,MAAM,GAAG;QAAE,IAAI,EAAE;YAAE,EAAE,EAAE;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,YAAY,EAAE,MAAM,CAAA;aAAE,CAAA;SAAE,CAAA;KAAE,CAAC;CACzE,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,KAAK,EACV,OAAO,EACP,SAAS,EACT,iBAAiB,EACjB,oBAAoB,EACpB,uBAAuB,EACvB,0BAA0B,EAC3B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,cAAc,UAAU,CAAC;AACzB,cAAc,UAAU,CAAC;AAEzB,MAAM,MAAM,gBAAgB,GAAG,uBAAuB,GAAG;IACvD,KAAK,EAAE;QACL,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG,SAAS,CAAC;AAEvC,MAAM,MAAM,YAAY,GAAG,OAAO,CAAC;AAEnC,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,CAAC;AAExD,MAAM,MAAM,oBAAoB,GAAG,0BAA0B,CAAC;AAE9D,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,IAAI;IAC/D,IAAI,EAAE,CAAC,CAAC;IACR,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,MAAM,iBAAiB,GAAG;IAC9B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG;IAC5C,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,CAAC,EAAE,SAAS,GAAG,IAAI,CAAC;IACxB,OAAO,EAAE,kBAAkB,CAAC;IAC5B,MAAM,CAAC,EAAE,iBAAiB,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG,oBAAoB,GAAG;IAClD,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG;IAC3C,OAAO,EAAE,SAAS,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;CAC7B,CAAC;AAEF,MAAM,MAAM,oCAAoC,GAAG;IACjD,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG;IAClC,IAAI,EAAE;QAAE,EAAE,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,YAAY,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,CAAC,GAAG,EAAE,MAAM,GAAG,qBAAqB,CAAC;CACtC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ledgerhq/coin-aptos",
-  "version": "1.9.5",
+  "version": "1.10.0-nightly.0",
   "description": "Ledger Aptos Coin integration",
   "keywords": [
     "Ledger",
@@ -90,14 +90,15 @@
     "expect": "^27.4.6",
     "lodash": "^4.17.21",
     "rxjs": "^7.8.1",
-    "@ledgerhq/coin-framework": "^5.0.1",
-    "@ledgerhq/cryptoassets": "^13.16.0",
-    "@ledgerhq/devices": "8.4.4",
-    "@ledgerhq/errors": "^6.19.1",
-    "@ledgerhq/live-env": "^2.8.0",
-    "@ledgerhq/live-network": "^2.0.7",
+    "@ledgerhq/coin-framework": "^5.1.0-nightly.0",
+    "@ledgerhq/cryptoassets": "^13.17.0-nightly.1",
+    "@ledgerhq/types-cryptoassets": "^7.23.0",
+    "@ledgerhq/devices": "8.4.5-nightly.1",
+    "@ledgerhq/errors": "^6.20.0-nightly.1",
+    "@ledgerhq/live-env": "^2.9.0-nightly.0",
+    "@ledgerhq/live-network": "^2.0.8-nightly.0",
     "@ledgerhq/logs": "^6.12.0",
-    "@ledgerhq/types-live": "^6.70.0"
+    "@ledgerhq/types-live": "^6.71.0-nightly.0"
   },
   "devDependencies": {
     "@faker-js/faker": "^9.4.0",

package/src/__tests__/api/index.test.ts

@@ -12,6 +12,7 @@ import {
 import network from "@ledgerhq/live-network";
 import BigNumber from "bignumber.js";
 import { AptosAPI } from "../../api";
+import { TokenCurrency } from "@ledgerhq/types-cryptoassets";
 
 jest.mock("@aptos-labs/ts-sdk");
 jest.mock("@apollo/client");
@@ -83,8 +84,77 @@ describe("Aptos API", () => {
     });
   });
 
+  describe("getBalance", () => {
+    let token: TokenCurrency;
+
+    beforeEach(() => {
+      token = {
+        type: "TokenCurrency",
+        id: "aptos_token",
+        name: "Aptos Token",
+        ticker: "APT",
+        units: [{ name: "APT", code: "APT", magnitude: 6 }],
+        contractAddress: "APTOS_1_ADDRESS",
+        tokenType: "fungible_asset",
+        parentCurrency: {
+          type: "CryptoCurrency",
+          id: "aptos",
+          name: "Aptos",
+          ticker: "APT",
+          units: [{ name: "APT", code: "APT", magnitude: 6 }],
+          color: "#000000",
+          family: "aptos",
+          scheme: "aptos",
+          explorerViews: [],
+          managerAppName: "Aptos",
+          coinType: 637,
+        },
+      };
+    });
+
+    it("get coin balance", async () => {
+      mockedAptos.mockImplementation(() => ({
+        view: jest.fn().mockReturnValue(["1234"]),
+      }));
+
+      token.tokenType = "coin";
+      const api = new AptosAPI("aptos");
+      const balance = await api.getBalance("address", token);
+
+      expect(balance).toEqual(new BigNumber(1234));
+    });
+
+    it("get fungible assets balance", async () => {
+      mockedAptos.mockImplementation(() => ({
+        view: jest.fn().mockReturnValue(["12345"]),
+      }));
+
+      token.tokenType = "fungible_asset";
+
+      const api = new AptosAPI("aptos");
+      const balance = await api.getBalance("address", token);
+
+      expect(balance).toEqual(new BigNumber(12345));
+    });
+
+    it("return 0 balace if could not retrieve proper balance of fungible assets", async () => {
+      mockedAptos.mockImplementation(() => ({
+        view: jest.fn().mockImplementation(() => {
+          throw new Error("error");
+        }),
+      }));
+
+      token.tokenType = "fungible_asset";
+
+      const api = new AptosAPI("aptos");
+      const balance = await api.getBalance("address", token);
+
+      expect(balance).toEqual(new BigNumber(0));
+    });
+  });
+
   describe("getAccountInfo", () => {
-    it("calls getBalance, fetchTransactions and getHeight", async () => {
+    it("calls getCoinBalance, fetchTransactions and getHeight", async () => {
       mockedAptos.mockImplementation(() => ({
         view: jest.fn().mockReturnValue(["123"]),
         getTransactionByVersion: jest.fn().mockReturnValue({
@@ -124,7 +194,7 @@ describe("Aptos API", () => {
       mockedApolloClient.mockImplementation(() => ({
         query: async () => ({
           data: {
-            address_version_from_move_resources: [{ transaction_version: "v1" }],
+            account_transactions: [{ transaction_version: 1 }],
           },
           loading: false,
           networkStatus: 7,
@@ -190,7 +260,7 @@ describe("Aptos API", () => {
       mockedApolloClient.mockImplementation(() => ({
         query: async () => ({
           data: {
-            address_version_from_move_resources: [{ transaction_version: "v1" }],
+            account_transactions: [{ transaction_version: 1 }],
           },
           loading: false,
           networkStatus: 7,
@@ -254,7 +324,7 @@ describe("Aptos API", () => {
       mockedApolloClient.mockImplementation(() => ({
         query: async () => ({
           data: {
-            address_version_from_move_resources: [{ transaction_version: "v1" }],
+            account_transactions: [{ transaction_version: 1 }],
           },
           loading: false,
           networkStatus: 7,
@@ -308,7 +378,7 @@ describe("Aptos API", () => {
       mockedApolloClient.mockImplementation(() => ({
         query: async () => ({
           data: {
-            address_version_from_move_resources: [{ transaction_version: "v1" }],
+            account_transactions: [{ transaction_version: 1 }],
           },
           loading: false,
           networkStatus: 7,

package/src/__tests__/bridge/buildTransaction.test.ts

@@ -1,4 +1,9 @@
-import { createFixtureAccount, createFixtureTransaction } from "../../bridge/bridge.fixture";
+import {
+  createFixtureAccount,
+  createFixtureAccountWithSubAccount,
+  createFixtureTransaction,
+  createFixtureTransactionWithSubAccount,
+} from "../../bridge/bridge.fixture";
 import buildTransaction from "../../bridge/buildTransaction";
 import { AptosAPI } from "../../api";
 import { normalizeTransactionOptions } from "../../bridge/logic";
@@ -26,8 +31,12 @@ jest.mock("../../api", () => {
   };
 });
 
+afterEach(() => {
+  jest.clearAllMocks();
+});
+
 describe("buildTransaction Test", () => {
-  it("should return tx", async () => {
+  it("should return transaction for main account", async () => {
     const account = createFixtureAccount();
     const transaction = createFixtureTransaction();
     const aptosClient = new AptosAPI(account.currency.id);
@@ -54,8 +63,81 @@ describe("buildTransaction Test", () => {
     expect(generateTransactionArgs[1]).toEqual({
       function: "0x1::aptos_account::transfer_coins",
       typeArguments: ["0x1::aptos_coin::AptosCoin"],
-      functionArguments: ["", "0"],
+      functionArguments: ["recipient", "0"],
+    });
+    expect(generateTransactionArgs[2]).toEqual({ maxGasAmount: "100", gasUnitPrice: "200" });
+  });
+
+  it("should return transaction for token account of type fungible asset", async () => {
+    const account = createFixtureAccountWithSubAccount("fungible_asset");
+    const transaction = createFixtureTransactionWithSubAccount();
+    const aptosClient = new AptosAPI(account.currency.id);
+    const result = await buildTransaction(account, transaction, aptosClient);
+
+    const expected = "tx";
+
+    expect(result).toBe(expected);
+
+    const mockedNormalizeTransactionOptions = jest.mocked(normalizeTransactionOptions);
+
+    expect(mockedNormalizeTransactionOptions).toHaveBeenCalledTimes(1);
+    expect(generateTransaction).toHaveBeenCalledTimes(1);
+
+    const generateTransactionArgs: [string, InputEntryFunctionData, TransactionOptions][] =
+      generateTransaction.mock.calls[0];
+
+    expect(mockedNormalizeTransactionOptions.mock.calls[0][0]).toEqual({
+      maxGasAmount: "0",
+      gasUnitPrice: "0",
+    });
+
+    expect(generateTransactionArgs[0]).toBe("address");
+    expect(generateTransactionArgs[1]).toEqual({
+      function: "0x1::primary_fungible_store::transfer",
+      typeArguments: ["0x1::fungible_asset::Metadata"],
+      functionArguments: ["contract_address", "recipient", "0"],
+    });
+    expect(generateTransactionArgs[2]).toEqual({ maxGasAmount: "100", gasUnitPrice: "200" });
+  });
+
+  it("should return transaction for token account of type coin", async () => {
+    const account = createFixtureAccountWithSubAccount("coin");
+    const transaction = createFixtureTransactionWithSubAccount();
+    const aptosClient = new AptosAPI(account.currency.id);
+    const result = await buildTransaction(account, transaction, aptosClient);
+
+    const expected = "tx";
+
+    expect(result).toBe(expected);
+
+    const mockedNormalizeTransactionOptions = jest.mocked(normalizeTransactionOptions);
+
+    expect(mockedNormalizeTransactionOptions).toHaveBeenCalledTimes(1);
+    expect(generateTransaction).toHaveBeenCalledTimes(1);
+
+    const generateTransactionArgs: [string, InputEntryFunctionData, TransactionOptions][] =
+      generateTransaction.mock.calls[0];
+
+    expect(mockedNormalizeTransactionOptions.mock.calls[0][0]).toEqual({
+      maxGasAmount: "0",
+      gasUnitPrice: "0",
+    });
+
+    expect(generateTransactionArgs[0]).toBe("address");
+    expect(generateTransactionArgs[1]).toEqual({
+      function: "0x1::aptos_account::transfer_coins",
+      typeArguments: ["contract_address"],
+      functionArguments: ["recipient", "0"],
     });
     expect(generateTransactionArgs[2]).toEqual({ maxGasAmount: "100", gasUnitPrice: "200" });
   });
+
+  it("should throw error if token is not supported", async () => {
+    const account = createFixtureAccountWithSubAccount("not_supported_token_type");
+    const transaction = createFixtureTransactionWithSubAccount();
+    const aptosClient = new AptosAPI(account.currency.id);
+    expect(async () => await buildTransaction(account, transaction, aptosClient)).rejects.toThrow(
+      "Token type not_supported_token_type not supported",
+    );
+  });
 });

package/src/__tests__/bridge/getFeesForTransaction.test.ts

@@ -1,5 +1,10 @@
 import BigNumber from "bignumber.js";
-import { createFixtureAccount, createFixtureTransaction } from "../../bridge/bridge.fixture";
+import {
+  createFixtureAccount,
+  createFixtureAccountWithSubAccount,
+  createFixtureTransaction,
+  createFixtureTransactionWithSubAccount,
+} from "../../bridge/bridge.fixture";
 import * as getFeesForTransaction from "../../bridge/getFeesForTransaction";
 import { AptosAPI } from "../../api";
 
@@ -24,16 +29,18 @@ jest.mock("@aptos-labs/ts-sdk", () => {
   };
 });
 
-jest.mock("../../bridge/logic", () => {
-  return {
-    DEFAULT_GAS: 201,
-    DEFAULT_GAS_PRICE: 101,
-    ESTIMATE_GAS_MUL: 1,
-    normalizeTransactionOptions: jest.fn(),
-  };
-});
+const mockedGetTokenAccount = jest.fn();
 
 describe("getFeesForTransaction Test", () => {
+  beforeEach(() => {
+    jest.mock("../../bridge/logic", () => ({
+      DEFAULT_GAS: 201,
+      DEFAULT_GAS_PRICE: 101,
+      ESTIMATE_GAS_MUL: 1,
+      normalizeTransactionOptions: jest.fn(),
+      getTokenAccount: mockedGetTokenAccount,
+    }));
+  });
   describe("when using getFee", () => {
     describe("with vm_status as INSUFFICIENT_BALANCE", () => {
       it("should return a fee estimation object", async () => {
@@ -46,6 +53,7 @@ describe("getFeesForTransaction Test", () => {
             gas_unit_price: "102",
           },
         ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
 
         const account = createFixtureAccount();
         const transaction = createFixtureTransaction();
@@ -68,6 +76,40 @@ describe("getFeesForTransaction Test", () => {
 
         expect(result).toEqual(expected);
       });
+
+      it("should return a fee estimation object for the token transaction", async () => {
+        simulateTransaction = jest.fn(() => [
+          {
+            success: true,
+            vm_status: [],
+            expiration_timestamp_secs: 5,
+            gas_used: "202",
+            gas_unit_price: "102",
+          },
+        ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
+
+        const account = createFixtureAccountWithSubAccount("coin");
+        const transaction = createFixtureTransactionWithSubAccount();
+        const aptosClient = new AptosAPI(account.currency.id);
+
+        transaction.amount = new BigNumber(1);
+        account.xpub = "xpub";
+        account.spendableBalance = new BigNumber(100000000);
+
+        const result = await getFeesForTransaction.getFee(account, transaction, aptosClient);
+
+        const expected = {
+          fees: new BigNumber(20604),
+          estimate: {
+            maxGasAmount: "202",
+            gasUnitPrice: "102",
+          },
+          errors: {},
+        };
+
+        expect(result).toEqual(expected);
+      });
     });
 
     describe("with vm_status as DUMMY_STATE", () => {
@@ -81,6 +123,7 @@ describe("getFeesForTransaction Test", () => {
             gas_unit_price: "100",
           },
         ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
 
         const account = createFixtureAccount();
         const transaction = createFixtureTransaction();
@@ -95,6 +138,44 @@ describe("getFeesForTransaction Test", () => {
         }).rejects.toThrow("Simulation failed with following error: DUMMY_STATE");
       });
     });
+
+    describe("with vm_status as MAX_GAS_UNITS_BELOW_MIN_TRANSACTION_GAS_UNITS", () => {
+      it("should return a fee estimation object with GasInsuficeinetBalance error", async () => {
+        simulateTransaction = jest.fn(() => [
+          {
+            success: false,
+            vm_status: ["MAX_GAS_UNITS_BELOW_MIN_TRANSACTION_GAS_UNITS"],
+            expiration_timestamp_secs: 5,
+            gas_used: "0",
+            gas_unit_price: "100",
+          },
+        ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
+
+        const account = createFixtureAccountWithSubAccount("coin");
+        const transaction = createFixtureTransactionWithSubAccount();
+        const aptosClient = new AptosAPI(account.currency.id);
+
+        transaction.amount = new BigNumber(1);
+        account.xpub = "xpub";
+        account.spendableBalance = new BigNumber(100000000);
+
+        const result = await getFeesForTransaction.getFee(account, transaction, aptosClient);
+
+        const expected = {
+          fees: new BigNumber(0),
+          estimate: {
+            maxGasAmount: "0",
+            gasUnitPrice: "100",
+          },
+          errors: {
+            maxGasAmount: "GasInsufficientBalance",
+          },
+        };
+
+        expect(result).toEqual(expected);
+      });
+    });
   });
 
   describe("when using getEstimatedGas", () => {
@@ -109,6 +190,7 @@ describe("getFeesForTransaction Test", () => {
             gas_unit_price: "102",
           },
         ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
 
         const account = createFixtureAccount();
         const transaction = createFixtureTransaction();
@@ -152,6 +234,8 @@ describe("getFeesForTransaction Test", () => {
             gas_unit_price: "102",
           },
         ]);
+        mockedGetTokenAccount.mockReturnValue(undefined);
+
         const account = createFixtureAccount();
         account.xpub = "xpub";
         const transaction = createFixtureTransaction();
@@ -186,4 +270,55 @@ describe("getFeesForTransaction Test", () => {
       });
     });
   });
+
+  describe("when key is in cache from a token account", () => {
+    beforeEach(() => {
+      jest.clearAllMocks();
+    });
+
+    it("should return cached fee", async () => {
+      simulateTransaction = jest.fn(() => [
+        {
+          success: true,
+          vm_status: [],
+          expiration_timestamp_secs: 5,
+          gas_used: "202",
+          gas_unit_price: "102",
+        },
+      ]);
+      mockedGetTokenAccount.mockReturnValue(undefined);
+
+      const account = createFixtureAccountWithSubAccount("coin");
+      account.xpub = "xpub";
+      const transaction = createFixtureTransactionWithSubAccount();
+      const aptosClient = new AptosAPI(account.currency.id);
+
+      transaction.amount = new BigNumber(10);
+
+      const result1 = await getFeesForTransaction.getEstimatedGas(
+        account,
+        transaction,
+        aptosClient,
+      );
+      const result2 = await getFeesForTransaction.getEstimatedGas(
+        account,
+        transaction,
+        aptosClient,
+      );
+
+      expect(simulateTransaction.mock.calls).toHaveLength(1);
+
+      const expected = {
+        errors: {},
+        estimate: {
+          gasUnitPrice: "102",
+          maxGasAmount: "202",
+        },
+        fees: new BigNumber("20604"),
+      };
+
+      expect(result1).toEqual(expected);
+      expect(result2).toEqual(expected);
+    });
+  });
 });