fix getBalance

coinbase · Jun 3, 2024 · 7a351cc · 7a351cc
1 parent e599637
commit 7a351cc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/coinbase/address.ts b/src/coinbase/address.ts
@@ -120,10 +120,21 @@ export class Address {
    * @returns {Decimal} The balance of the asset.
    */
   async getBalance(assetId: string): Promise<Decimal> {
+    const normalizedAssetId = ((): string => {
+      switch (assetId) {
+        case Coinbase.assets.Gwei:
+          return Coinbase.assets.Eth;
+        case Coinbase.assets.Wei:
+          return Coinbase.assets.Eth;
+        default:
+          return assetId;
+      }
+    })();
+
     const response = await Coinbase.apiClients.address!.getAddressBalance(
       this.model.wallet_id,
       this.model.address_id,
-      assetId,
+      normalizedAssetId,
     );
 
     if (!response.data) {

diff --git a/src/coinbase/tests/address_test.ts b/src/coinbase/tests/address_test.ts
@@ -101,7 +101,7 @@ describe("Address", () => {
     expect(Coinbase.apiClients.address!.getAddressBalance).toHaveBeenCalledWith(
       address.getWalletId(),
       address.getId(),
-      assetId,
+      Coinbase.assets.Eth,
     );
     expect(Coinbase.apiClients.address!.getAddressBalance).toHaveBeenCalledTimes(1);
   });
@@ -114,7 +114,7 @@ describe("Address", () => {
     expect(Coinbase.apiClients.address!.getAddressBalance).toHaveBeenCalledWith(
       address.getWalletId(),
       address.getId(),
-      assetId,
+      Coinbase.assets.Eth,
     );
     expect(Coinbase.apiClients.address!.getAddressBalance).toHaveBeenCalledTimes(1);
   });