@ledgerhq/coin-aptos 2.0.0-nightly.5 → 2.0.0

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

@@ -1,12 +1,12 @@
 import prepareTransaction from "../../bridge/prepareTransaction";
-import { AptosAPI } from "../../api";
+import { AptosAPI } from "../../network";
 import { getEstimatedGas } from "../../bridge/getFeesForTransaction";
 import { getMaxSendBalance } from "../../bridge/logic";
 import BigNumber from "bignumber.js";
 import type { Account } from "@ledgerhq/types-live";
 import type { Transaction } from "../../types";
 
-jest.mock("../../api");
+jest.mock("../../network");
 jest.mock("../../bridge/getFeesForTransaction");
 jest.mock("../../bridge/logic");
 

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

@@ -1,16 +1,21 @@
 import BigNumber from "bignumber.js";
 import { Observable } from "rxjs";
-import { SignerContext } from "@ledgerhq/coin-framework/lib/signer";
+import { SignerContext } from "@ledgerhq/coin-framework/signer";
 import { createFixtureAccount, createFixtureTransaction } from "../../bridge/bridge.fixture";
 import buildSignOperation, { getAddress } from "../../bridge/signOperation";
 import { AptosSigner } from "../../types";
 import { signTransaction } from "../../network";
+import { TokenAccount } from "@ledgerhq/types-live";
+import { faker } from "@faker-js/faker";
+import { emptyHistoryCache } from "@ledgerhq/coin-framework/account/index";
 
-jest.mock("../../api", () => {
+const generateTransaction = jest.fn(() => "tx");
+
+jest.mock("../../network/client", () => {
   return {
     AptosAPI: function () {
       return {
-        generateTransaction: jest.fn(() => "tx"),
+        generateTransaction,
       };
     },
   };
@@ -38,6 +43,7 @@ describe("buildSignOperation", () => {
   beforeEach(() => {
     mockedSignTransaction = jest.mocked(signTransaction);
   });
+
   afterEach(() => jest.clearAllMocks());
 
   it("should thrown an error", async () => {
@@ -104,6 +110,7 @@ describe("buildSignOperation", () => {
             accountId: "js:2:aptos:0x000:",
             date,
             transactionSequenceNumber: 789,
+            subOperations: [],
           },
           signature: "7369676e65645478",
         },
@@ -125,7 +132,130 @@ describe("buildSignOperation", () => {
     jest.useFakeTimers().setSystemTime(date);
 
     const account = createFixtureAccount();
-    const transaction = createFixtureTransaction();
+    const transaction = createFixtureTransaction({ amount: BigNumber(5) });
+
+    account.balance = new BigNumber(40);
+    transaction.fees = new BigNumber(30);
+    transaction.useAllAmount = true;
+
+    account.id = "js:2:aptos:0x000:";
+    transaction.mode = "send";
+
+    const observable = await buildSignOperation({} as unknown as SignerContext<AptosSigner>)({
+      account,
+      deviceId: "1",
+      transaction,
+    });
+
+    expect(observable).toBeInstanceOf(Observable);
+
+    const expectedValues = [
+      { type: "device-signature-requested" },
+      { type: "device-signature-granted" },
+      {
+        type: "signed",
+        signedOperation: {
+          operation: {
+            id: "js:2:aptos:0x000:--OUT",
+            hash: "",
+            type: "OUT",
+            value: new BigNumber(35),
+            fee: transaction.fees,
+            extra: {},
+            blockHash: null,
+            blockHeight: null,
+            senders: [account.freshAddress],
+            recipients: [transaction.recipient],
+            accountId: "js:2:aptos:0x000:",
+            date,
+            transactionSequenceNumber: 789,
+            subOperations: [],
+          },
+          signature: "7369676e65645478",
+        },
+      },
+    ];
+
+    let i = 0;
+
+    observable.forEach(signOperationEvent => {
+      expect(signOperationEvent).toEqual(expectedValues[i]);
+      i++;
+    });
+  });
+
+  it("should have sub operations if it has subaccount", async () => {
+    mockedSignTransaction.mockReturnValue("signedTx");
+
+    const date = new Date("2020-01-01");
+    jest.useFakeTimers().setSystemTime(date);
+
+    const id = faker.string.uuid();
+    const account = createFixtureAccount({
+      id,
+      subAccounts: [
+        {
+          type: "TokenAccount",
+          id: "subAccountId",
+          parentId: id,
+          token: {
+            type: "TokenCurrency",
+            id: "aptos/coin/dstapt_0xd11107bdf0d6d7040c6c0bfbdecb6545191fdf13e8d8d259952f53e1713f61b5::staked_coin::stakedaptos",
+            contractAddress:
+              "0xd11107bdf0d6d7040c6c0bfbdecb6545191fdf13e8d8d259952f53e1713f61b5::staked_coin::StakedAptos",
+            parentCurrency: {
+              type: "CryptoCurrency",
+              id: "aptos",
+              coinType: 637,
+              name: "Aptos",
+              managerAppName: "Aptos",
+              ticker: "APT",
+              scheme: "aptos",
+              color: "#231F20",
+              family: "aptos",
+              units: [
+                {
+                  name: "APT",
+                  code: "APT",
+                  magnitude: 8,
+                },
+              ],
+              explorerViews: [
+                {
+                  address: "https://explorer.aptoslabs.com/account/$address?network=mainnet",
+                  tx: "https://explorer.aptoslabs.com/txn/$hash?network=mainnet",
+                },
+              ],
+            },
+            name: "dstAPT",
+            tokenType: "coin",
+            ticker: "dstAPT",
+            disableCountervalue: false,
+            delisted: false,
+            units: [
+              {
+                name: "dstAPT",
+                code: "dstAPT",
+                magnitude: 8,
+              },
+            ],
+          },
+          balance: BigNumber(100),
+          spendableBalance: BigNumber(100),
+          creationDate: date,
+          operationsCount: 0,
+          operations: [],
+          pendingOperations: [],
+          balanceHistoryCache: emptyHistoryCache,
+          swapHistory: [],
+        },
+      ] as TokenAccount[],
+    });
+
+    const transaction = createFixtureTransaction({
+      amount: BigNumber(5),
+      subAccountId: "subAccountId",
+    });
 
     account.balance = new BigNumber(40);
     transaction.fees = new BigNumber(30);
@@ -152,7 +282,7 @@ describe("buildSignOperation", () => {
             id: "js:2:aptos:0x000:--OUT",
             hash: "",
             type: "OUT",
-            value: new BigNumber(10),
+            value: new BigNumber(30),
             fee: transaction.fees,
             extra: {},
             blockHash: null,
@@ -162,6 +292,18 @@ describe("buildSignOperation", () => {
             accountId: "js:2:aptos:0x000:",
             date,
             transactionSequenceNumber: 789,
+            subOperations: [
+              {
+                accountId: "subAccountId",
+                date: new Date("2020-01-01"),
+                fee: BigNumber(30),
+                id: "subAccountId--OUT",
+                recipients: ["recipient"],
+                senders: ["address"],
+                type: "OUT",
+                value: BigNumber(5),
+              },
+            ],
           },
           signature: "7369676e65645478",
         },