@ledgerhq/coin-aptos 2.0.0 → 2.1.0-next.0

package/src/__tests__/network/client.test.ts

@@ -4,32 +4,42 @@ import {
   Aptos,
   ChainId,
   Ed25519PublicKey,
+  Hex,
   InputEntryFunctionData,
   RawTransaction,
   Serializable,
-  post,
+  postAptosFullNode,
+  TransactionResponseType,
 } from "@aptos-labs/ts-sdk";
 import network from "@ledgerhq/live-network";
 import BigNumber from "bignumber.js";
 import { AptosAPI } from "../../network";
 import { AptosAsset, AptosExtra, AptosSender } from "../../types/assets";
-import { TransactionIntent } from "@ledgerhq/coin-framework/api/types";
+import { Pagination, TransactionIntent } from "@ledgerhq/coin-framework/api/types";
 import { TokenCurrency } from "@ledgerhq/types-cryptoassets";
+import { APTOS_ASSET_ID } from "../../constants";
+import { AptosTransaction } from "../../types";
 
 jest.mock("@aptos-labs/ts-sdk");
 jest.mock("@apollo/client");
 let mockedAptos: jest.Mocked<any>;
 let mockedApolloClient: jest.Mocked<any>;
 let mockedPost = jest.fn();
+let mockedHex: jest.Mocked<any>;
 
 jest.mock("@ledgerhq/live-network/network");
 const mockedNetwork = jest.mocked(network);
 
+jest.mock("@ledgerhq/cryptoassets/tokens", () => ({
+  findTokenByAddressInCurrency: jest.fn().mockReturnValue("token-name"),
+}));
+
 describe("Aptos API", () => {
   beforeEach(() => {
     mockedAptos = jest.mocked(Aptos);
+    mockedHex = jest.mocked(Hex);
     mockedApolloClient = jest.mocked(ApolloClient);
-    mockedPost = jest.mocked(post);
+    mockedPost = jest.mocked(postAptosFullNode);
   });
 
   afterEach(() => jest.clearAllMocks());
@@ -546,39 +556,48 @@ describe("Aptos API", () => {
       });
     });
   });
+
   describe("broadcast", () => {
     it("broadcasts the transaction", async () => {
       mockedPost.mockImplementation(async () => ({ data: { hash: "ok" } }));
-      const mockedPostSpy = jest.spyOn({ post: mockedPost }, "post");
+      const mockedPostSpy = jest.spyOn({ postAptosFullNode: mockedPost }, "postAptosFullNode");
 
+      // const txBytes = Hex.fromHexString(tx).toUint8Array();
       mockedAptos.mockImplementation(() => ({
         config: "config",
       }));
 
+      mockedHex.fromHexString = jest.fn().mockReturnValue({ toUint8Array: () => "123" });
+
       const api = new AptosAPI("aptos");
-      await api.broadcast("signature");
+
+      await api.broadcast("signed transaction");
 
       expect(mockedPostSpy).toHaveBeenCalledWith({
-        contentType: "application/x.aptos.signed_transaction+bcs",
         aptosConfig: "config",
-        body: Uint8Array.from(Buffer.from("signature", "hex")),
+        body: "123",
         path: "transactions",
-        type: "Fullnode",
         originMethod: "",
+        contentType: "application/x.aptos.signed_transaction+bcs",
       });
     });
   });
 
   describe("estimateFees", () => {
-    it("estimates the fees", async () => {
+    it("estimates the fees for native asset", async () => {
       const gasEstimation = { gas_estimate: 100 };
+      const buildSimple = jest.fn().mockResolvedValue({ rawTransaction: {} });
+
+      const time = new Date("2025-05-29");
+      jest.useFakeTimers().setSystemTime(time);
+
       mockedAptos.mockImplementation(() => ({
         getLedgerInfo: jest.fn().mockResolvedValue({
-          ledger_timestamp: Date.now(),
+          ledger_timestamp: time,
         }),
         transaction: {
           build: {
-            simple: jest.fn().mockResolvedValue({ rawTransaction: {} }),
+            simple: buildSimple,
           },
           simulate: {
             simple: jest.fn().mockResolvedValue([
@@ -612,7 +631,627 @@ describe("Aptos API", () => {
 
       const fees = await api.estimateFees(transactionIntent);
 
+      expect(buildSimple.mock.calls[0][0]).toEqual({
+        sender: "address1",
+        data: {
+          function: "0x1::aptos_account::transfer_coins",
+          typeArguments: ["0x1::aptos_coin::AptosCoin"],
+          functionArguments: ["address2", 100n],
+        },
+        options: {
+          maxGasAmount: 200,
+          gasUnitPrice: 100,
+          expireTimestamp: Number(Math.ceil(+time / 1_000_000 + 2 * 60)),
+        },
+      });
+
       expect(fees.value.toString()).toEqual("40");
     });
+
+    it("estimates the fees for token coin", async () => {
+      const gasEstimation = { gas_estimate: 100 };
+      const buildSimple = jest.fn().mockResolvedValue({ rawTransaction: {} });
+
+      const time = new Date("2025-05-29");
+      jest.useFakeTimers().setSystemTime(time);
+
+      mockedAptos.mockImplementation(() => ({
+        getLedgerInfo: jest.fn().mockResolvedValue({
+          ledger_timestamp: time,
+        }),
+        transaction: {
+          build: {
+            simple: buildSimple,
+          },
+          simulate: {
+            simple: jest.fn().mockResolvedValue([
+              {
+                gas_used: 10,
+                gas_unit_price: 2,
+              },
+            ]),
+          },
+        },
+        getGasPriceEstimation: jest.fn().mockReturnValue(gasEstimation),
+      }));
+
+      const amount = BigInt(100);
+      const sender: AptosSender = {
+        xpub: "xpub",
+        freshAddress: "address1",
+      };
+      const recipient = "address2";
+
+      const api = new AptosAPI("aptos");
+      const transactionIntent: TransactionIntent<AptosAsset, AptosExtra, AptosSender> = {
+        asset: {
+          type: "token",
+          standard: "coin",
+          contractAddress: "0x111",
+        },
+        type: "send",
+        sender,
+        amount,
+        recipient,
+      };
+
+      const fees = await api.estimateFees(transactionIntent);
+
+      expect(buildSimple.mock.calls[0][0]).toEqual({
+        sender: "address1",
+        data: {
+          function: "0x1::aptos_account::transfer_coins",
+          typeArguments: ["0x111"],
+          functionArguments: ["address2", 100n],
+        },
+        options: {
+          maxGasAmount: 200,
+          gasUnitPrice: 100,
+          expireTimestamp: Number(Math.ceil(+time / 1_000_000 + 2 * 60)),
+        },
+      });
+
+      expect(fees.value.toString()).toEqual("20");
+    });
+
+    it("estimates the fees for token FA", async () => {
+      const gasEstimation = { gas_estimate: 100 };
+      const buildSimple = jest.fn().mockResolvedValue({ rawTransaction: {} });
+      const time = new Date("2025-05-29");
+      jest.useFakeTimers().setSystemTime(time);
+
+      mockedAptos.mockImplementation(() => ({
+        getLedgerInfo: jest.fn().mockResolvedValue({
+          ledger_timestamp: time,
+        }),
+        transaction: {
+          build: {
+            simple: buildSimple,
+          },
+          simulate: {
+            simple: jest.fn().mockResolvedValue([
+              {
+                gas_used: 10,
+                gas_unit_price: 3,
+              },
+            ]),
+          },
+        },
+        getGasPriceEstimation: jest.fn().mockReturnValue(gasEstimation),
+      }));
+
+      const amount = BigInt(100);
+      const sender: AptosSender = {
+        xpub: "xpub",
+        freshAddress: "address1",
+      };
+      const recipient = "address2";
+
+      const api = new AptosAPI("aptos");
+      const transactionIntent: TransactionIntent<AptosAsset, AptosExtra, AptosSender> = {
+        asset: {
+          type: "token",
+          standard: "fungible_asset",
+          contractAddress: "0x111",
+        },
+        type: "send",
+        sender,
+        amount,
+        recipient,
+      };
+
+      const fees = await api.estimateFees(transactionIntent);
+
+      expect(buildSimple.mock.calls[0][0]).toEqual({
+        sender: "address1",
+        data: {
+          function: "0x1::primary_fungible_store::transfer",
+          typeArguments: ["0x1::fungible_asset::Metadata"],
+          functionArguments: ["0x111", "address2", 100n],
+        },
+        options: {
+          maxGasAmount: 200,
+          gasUnitPrice: 100,
+          expireTimestamp: Number(Math.ceil(+time / 1_000_000 + 2 * 60)),
+        },
+      });
+
+      expect(fees.value.toString()).toEqual("30");
+    });
+  });
+
+  describe("getBalances", () => {
+    it("returns an array of AptosBalances objects", async () => {
+      const assets = [{ asset_type: APTOS_ASSET_ID, amount: 200 }];
+      const mockGetCurrentFungibleAssetBalances = jest.fn().mockResolvedValue(assets);
+      mockedAptos.mockImplementation(() => ({
+        getCurrentFungibleAssetBalances: mockGetCurrentFungibleAssetBalances,
+      }));
+
+      const address = "0x42";
+
+      const api = new AptosAPI("aptos");
+      const balances = await api.getBalances(address);
+
+      expect(mockGetCurrentFungibleAssetBalances).toHaveBeenCalledWith({
+        options: {
+          offset: 0,
+          limit: 1000,
+          where: {
+            asset_type: { _eq: APTOS_ASSET_ID },
+            owner_address: { _eq: address },
+          },
+        },
+      });
+      expect(balances).toHaveLength(1);
+      expect(balances[0].contractAddress).toBe(assets[0].asset_type);
+      expect(balances[0].amount).toStrictEqual(BigNumber(assets[0].amount));
+    });
+  });
+
+  describe("listOperations", () => {
+    it("list of operations", async () => {
+      const api = new AptosAPI("aptos");
+      const address = "0x12345";
+      const pagination: Pagination = { minHeight: 0 };
+
+      const txs: AptosTransaction[] = [
+        {
+          version: "2532591427",
+          hash: "0x3f35",
+          state_change_hash: "0xb480",
+          event_root_hash: "0x3fa1",
+          state_checkpoint_hash: null,
+          gas_used: "12",
+          success: true,
+          vm_status: "Executed successfully",
+          accumulator_root_hash: "0x319f",
+          changes: [
+            {
+              address: "0x4e5e",
+              state_key_hash: "0x3c0c",
+              data: {
+                type: "0x1::coin::CoinStore<0xd111::staked_coin::StakedAptos>",
+                data: {
+                  coin: {
+                    value: "4000000",
+                  },
+                  deposit_events: {
+                    counter: "9",
+                    guid: {
+                      id: {
+                        addr: "0x4e5e",
+                        creation_num: "4",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "6",
+                    guid: {
+                      id: {
+                        addr: "0x4e5e",
+                        creation_num: "5",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x1709",
+              data: {
+                type: "0x1::coin::CoinStore<0x1::aptos_coin::AptosCoin>",
+                data: {
+                  coin: {
+                    value: "68254118",
+                  },
+                  deposit_events: {
+                    counter: "46",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "2",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "89",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "3",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x5520",
+              data: {
+                type: "0x1::coin::CoinStore<0xd111::staked_coin::StakedAptos>",
+                data: {
+                  coin: {
+                    value: "1000000",
+                  },
+                  deposit_events: {
+                    counter: "7",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "10",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "13",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "11",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x6f1e",
+              data: {
+                type: "0x1::account::Account",
+                data: {
+                  authentication_key: address,
+                  coin_register_events: {
+                    counter: "5",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "0",
+                      },
+                    },
+                  },
+                  guid_creation_num: "12",
+                  key_rotation_events: {
+                    counter: "0",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "1",
+                      },
+                    },
+                  },
+                  rotation_capability_offer: {
+                    for: {
+                      vec: [],
+                    },
+                  },
+                  sequence_number: "122",
+                  signer_capability_offer: {
+                    for: {
+                      vec: [],
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              state_key_hash: "0x6e4b",
+              handle: "0x1b85",
+              key: "0x0619",
+              value: "0x1ddaf8da3b1497010000000000000000",
+              type: "write_table_item",
+            },
+          ],
+          sender: address,
+          sequence_number: "121",
+          max_gas_amount: "12",
+          gas_unit_price: "100",
+          expiration_timestamp_secs: "1743177404",
+          payload: {
+            function: "0x1::aptos_account::transfer_coins",
+            type_arguments: ["0xd111::staked_coin::StakedAptos"],
+            arguments: ["0x4e5e", "1500000"],
+            type: "entry_function_payload",
+          },
+          events: [
+            {
+              guid: {
+                creation_number: "11",
+                account_address: address,
+              },
+              sequence_number: "12",
+              type: "0x1::coin::WithdrawEvent",
+              data: {
+                amount: "1500000",
+              },
+            },
+            {
+              guid: {
+                creation_number: "4",
+                account_address: "0x4e5e",
+              },
+              sequence_number: "8",
+              type: "0x1::coin::DepositEvent",
+              data: {
+                amount: "1500000",
+              },
+            },
+            {
+              guid: {
+                creation_number: "0",
+                account_address: "0x0",
+              },
+              sequence_number: "0",
+              type: "0x1::transaction_fee::FeeStatement",
+              data: {
+                execution_gas_units: "6",
+                io_gas_units: "6",
+                storage_fee_octas: "0",
+                storage_fee_refund_octas: "0",
+                total_charge_gas_units: "12",
+              },
+            },
+          ],
+          timestamp: "1743177360481259",
+          type: TransactionResponseType.User,
+          block: {
+            height: 311948147,
+            hash: "0x6d02",
+          },
+        },
+        {
+          version: "2532549325",
+          hash: "0x9a6b",
+          state_change_hash: "0xa424",
+          event_root_hash: "0x0321",
+          state_checkpoint_hash: null,
+          gas_used: "12",
+          success: true,
+          vm_status: "Executed successfully",
+          accumulator_root_hash: "0xede9",
+          changes: [
+            {
+              address: "0x4e5e",
+              state_key_hash: "0x3c0c",
+              data: {
+                type: "0x1::coin::CoinStore<0xd111::staked_coin::StakedAptos>",
+                data: {
+                  coin: {
+                    value: "2500000",
+                  },
+                  deposit_events: {
+                    counter: "8",
+                    guid: {
+                      id: {
+                        addr: "0x4e5e",
+                        creation_num: "4",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "6",
+                    guid: {
+                      id: {
+                        addr: "0x4e5e",
+                        creation_num: "5",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x1709",
+              data: {
+                type: "0x1::coin::CoinStore<0x1::aptos_coin::AptosCoin>",
+                data: {
+                  coin: {
+                    value: "68255318",
+                  },
+                  deposit_events: {
+                    counter: "46",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "2",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "89",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "3",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x5520",
+              data: {
+                type: "0x1::coin::CoinStore<0xd111::staked_coin::StakedAptos>",
+                data: {
+                  coin: {
+                    value: "2500000",
+                  },
+                  deposit_events: {
+                    counter: "7",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "10",
+                      },
+                    },
+                  },
+                  frozen: false,
+                  withdraw_events: {
+                    counter: "12",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "11",
+                      },
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              address: address,
+              state_key_hash: "0x6f1e",
+              data: {
+                type: "0x1::account::Account",
+                data: {
+                  authentication_key: address,
+                  coin_register_events: {
+                    counter: "5",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "0",
+                      },
+                    },
+                  },
+                  guid_creation_num: "12",
+                  key_rotation_events: {
+                    counter: "0",
+                    guid: {
+                      id: {
+                        addr: address,
+                        creation_num: "1",
+                      },
+                    },
+                  },
+                  rotation_capability_offer: {
+                    for: {
+                      vec: [],
+                    },
+                  },
+                  sequence_number: "121",
+                  signer_capability_offer: {
+                    for: {
+                      vec: [],
+                    },
+                  },
+                },
+              },
+              type: "write_resource",
+            },
+            {
+              state_key_hash: "0x6e4b",
+              handle: "0x1b85",
+              key: "0x0619",
+              value: "0xe86e0039581497010000000000000000",
+              type: "write_table_item",
+            },
+          ],
+          sender: address,
+          sequence_number: "120",
+          max_gas_amount: "12",
+          gas_unit_price: "100",
+          expiration_timestamp_secs: "1743176706",
+          payload: {
+            function: "0x1::aptos_account::transfer_coins",
+            type_arguments: ["0xd111::staked_coin::StakedAptos"],
+            arguments: ["0x4e5e", "2500000"],
+            type: "entry_function_payload",
+          },
+          events: [
+            {
+              guid: {
+                creation_number: "11",
+                account_address: address,
+              },
+              sequence_number: "11",
+              type: "0x1::coin::WithdrawEvent",
+              data: {
+                amount: "2500000",
+              },
+            },
+            {
+              guid: {
+                creation_number: "4",
+                account_address: "0x4e5e",
+              },
+              sequence_number: "7",
+              type: "0x1::coin::DepositEvent",
+              data: {
+                amount: "2500000",
+              },
+            },
+            {
+              guid: {
+                creation_number: "0",
+                account_address: "0x0",
+              },
+              sequence_number: "0",
+              type: "0x1::transaction_fee::FeeStatement",
+              data: {
+                execution_gas_units: "6",
+                io_gas_units: "6",
+                storage_fee_octas: "0",
+                storage_fee_refund_octas: "0",
+                total_charge_gas_units: "12",
+              },
+            },
+          ],
+          timestamp: "1743176594693251",
+          type: TransactionResponseType.User,
+          block: {
+            height: 311942427,
+            hash: "0x8655",
+          },
+        },
+      ];
+
+      const transactions: AptosTransaction[] = txs;
+
+      api.getAccountInfo = jest.fn().mockResolvedValue({ transactions });
+
+      const ops = await api.listOperations(address, pagination);
+
+      expect(ops[0]).toHaveLength(2);
+      expect(ops[1]).toBe("");
+    });
   });
 });