Updating Coinbase Class tests

coinbase · May 20, 2024 · 59a9e35 · 59a9e35
1 parent 65880bb
commit 59a9e35
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/coinbase/tests/coinbase_test.ts b/src/coinbase/tests/coinbase_test.ts
@@ -67,20 +67,21 @@ describe("Coinbase tests", () => {
       const faucetTransaction = await wallet?.faucet();
       expect(faucetTransaction.getTransactionHash()).toBe("0xdeadbeef");
       expect(addressesApiMock.requestFaucetFunds).toHaveBeenCalledWith(
-        wallet.getId(),
-        wallet.defaultAddress()?.getId(),
+        defaultAddress.getWalletId(),
+        defaultAddress?.getId(),
       );
       expect(addressesApiMock.requestFaucetFunds).toHaveBeenCalledTimes(1);
     });
   });
 
   it("should raise an error if the user is not found", async () => {
-    jest.mock("../../client/api", () => ({
-      UsersApiFactory: jest.fn().mockReturnValue({
-        me: jest.fn().mockRejectedValue(new APIError("")),
-      }),
-    }));
+    Coinbase.apiClients.user = {
+      ...usersApiMock,
+      getCurrentUser: jest.fn().mockRejectedValue(new APIError("User not found")),
+    };
     const cbInstance = Coinbase.configureFromJson(`${PATH_PREFIX}/coinbase_cloud_api_key.json`);
     await expect(cbInstance.getDefaultUser()).rejects.toThrow(APIError);
+    expect(usersApiMock.getCurrentUser).toHaveBeenCalledWith();
+    expect(usersApiMock.getCurrentUser).toHaveBeenCalledTimes(1);
   });
 });