test: fix unit tests for synchronization and network

libs/coin-modules/coin-aptos/src/__tests__/bridge/synchronisation.test.ts

@@ -6,13 +6,29 @@ import { mergeOps } from "@ledgerhq/coin-framework/bridge/jsHelpers";
 import { AptosAPI } from "../../api";
 import { getAccountShape } from "../../bridge/synchronisation";
 
-jest.mock("@ledgerhq/coin-framework/account");
+jest.mock("@ledgerhq/coin-framework/account", () => {
+  const originalModule = jest.requireActual("@ledgerhq/coin-framework/account");
+  const partialMockedModule = Object.keys(originalModule).reduce(
+    (pre: { [key: string]: jest.Mock }, methodName) => {
+      pre[methodName] = jest.fn();
+      return pre;
+    },
+    {} as { [key: string]: jest.Mock },
+  );
+  return {
+    ...partialMockedModule,
+    // mock all except these
+    decodeAccountId: originalModule.decodeAccountId,
+  };
+});
+
 const mockedEncodeAccountId = jest.mocked(encodeAccountId);
 
 jest.mock("../../api");
 let mockedAptosAPI: jest.Mocked<any>;
 
 jest.mock("@ledgerhq/coin-framework/bridge/jsHelpers");
+jest.mock("invariant", () => jest.fn());
 
 describe("getAccountShape", () => {
   beforeEach(() => {
@@ -121,7 +137,7 @@ describe("getAccountShape", () => {
       {} as SyncConfig,
     );
 
-    expect(account.xpub).toEqual("");
+    expect(account.xpub).toEqual("1");
     expect(mockedEncodeAccountId).toHaveBeenCalledTimes(1);
     expect(mockedAptosAPI).toHaveBeenCalledTimes(1);
     expect(mockGetAccountSpy).toHaveBeenCalledWith("address", undefined);
@@ -217,7 +233,7 @@ describe("getAccountShape", () => {
       {} as SyncConfig,
     );
 
-    expect(account.xpub).toEqual("");
+    expect(account.xpub).toEqual("1");
     expect(mockedEncodeAccountId).toHaveBeenCalledTimes(1);
     expect(mockedAptosAPI).toHaveBeenCalledTimes(1);
     expect(mockGetAccountSpy).toHaveBeenCalledWith("address", undefined);
@@ -283,7 +299,74 @@ describe("getAccountShape", () => {
       {} as SyncConfig,
     );
 
-    expect(account.xpub).toEqual("");
+    expect(account.xpub).toEqual("1");
+    expect(mockedEncodeAccountId).toHaveBeenCalledTimes(1);
+    expect(mockedAptosAPI).toHaveBeenCalledTimes(1);
+    expect(mockGetAccountSpy).toHaveBeenCalledWith("address", 1);
+  });
+
+  it("get publicKey from rest", async () => {
+    const mockGetAccountInfo = jest.fn().mockImplementation(async () => ({
+      balance: BigInt(0),
+      transactions: [],
+      blockHeight: 0,
+    }));
+    mockedAptosAPI.mockImplementation(() => ({
+      getAccountInfo: mockGetAccountInfo,
+    }));
+    const mockGetAccountSpy = jest.spyOn({ getAccount: mockGetAccountInfo }, "getAccount");
+
+    const account = await getAccountShape(
+      {
+        id: "1",
+        address: "address",
+        currency: getCryptoCurrencyById("aptos"),
+        derivationMode: "",
+        index: 0,
+        xpub: "address",
+        derivationPath: "",
+        deviceId: "1",
+        initialAccount: {
+          id: "1:1:1:1:1",
+          seedIdentifier: "1",
+          derivationMode: "",
+          index: 0,
+          freshAddress: "address",
+          freshAddressPath: "",
+          used: true,
+          balance: BigInt(10),
+          spendableBalance: BigInt(10),
+          creationDate: new Date(),
+          blockHeight: 0,
+          currency: getCryptoCurrencyById("aptos"),
+          t: 1,
+          operations: [
+            {
+              id: "1",
+              hash: "hash",
+              type: "OUT",
+              value: BigInt(10),
+              fee: BigInt(0),
+              blockHeight: 0,
+              blockHash: "blockHash",
+              accountId: "1",
+              senders: ["sender"],
+              recipients: ["recipient"],
+              date: new Date(),
+              extra: { version: 1 },
+            },
+          ],
+          pendingOperations: [],
+          lastSyncDate: new Date(),
+          balanceHistoryCache: {},
+          swapHistory: [],
+        },
+        rest: { publicKey: "restPublicKey" },
+      } as unknown as AccountShapeInfo<Account>,
+      {} as SyncConfig,
+    );
+
+    expect(account.xpub).toEqual("restPublicKey");
     expect(mockedEncodeAccountId).toHaveBeenCalledTimes(1);
     expect(mockedAptosAPI).toHaveBeenCalledTimes(1);
     expect(mockGetAccountSpy).toHaveBeenCalledWith("address", 1);

libs/coin-modules/coin-aptos/src/__tests__/network/index.test.ts

@@ -10,19 +10,19 @@ import {
 } from "@aptos-labs/ts-sdk";
 
 jest.mock("@aptos-labs/ts-sdk", () => {
-  const originalAptosLabs = jest.requireActual("@aptos-labs/ts-sdk");
-  const partialMockedAptosLabs = Object.keys(originalAptosLabs).reduce(
+  const originalModule = jest.requireActual("@aptos-labs/ts-sdk");
+  const partialMockedModule = Object.keys(originalModule).reduce(
     (pre: { [key: string]: jest.Mock }, methodName) => {
       pre[methodName] = jest.fn();
       return pre;
     },
     {} as { [key: string]: jest.Mock },
   );
   return {
-    ...partialMockedAptosLabs,
+    ...partialMockedModule,
     // mock all except these
-    AccountAddress: originalAptosLabs.AccountAddress,
-    Hex: originalAptosLabs.Hex,
+    AccountAddress: originalModule.AccountAddress,
+    Hex: originalModule.Hex,
   };
 });
 let mockedGenerateSigningMessageForTransaction: jest.Mocked<any>;