feat(1708): mirror node getaccounttokenbalances response validation (#1731)

Signed-off-by: rozekmichal <michal.rozek@blockydevs.com>

Author: rozekmichal

src/core/services/mirrornode/__tests__/unit/mocks.ts

@@ -67,16 +67,14 @@ export const createMockGetAccountsAPIResponse = (
 export const createMockTokenBalancesResponse = (
   overrides: Partial<TokenBalancesResponse> = {},
 ): TokenBalancesResponse => ({
-  account: '0.0.1234',
-  balance: 0,
   tokens: [
     {
       token_id: '0.0.2000',
       balance: 100,
       decimals: 6,
     },
   ],
-  timestamp: '2024-01-01T12:00:00.000Z',
+  links: { next: null },
   ...overrides,
 });
 

src/core/services/mirrornode/hedera-mirrornode-service.ts

@@ -38,6 +38,7 @@ import { handleMirrorNodeErrorResponse } from '@/core/utils/handle-mirror-node-e
 import {
   AccountAPIResponseSchema,
   GetAccountsAPIResponseSchema,
+  TokenBalancesResponseSchema,
   TokenInfoSchema,
 } from './schemas';
 import { NetworkToBaseUrl } from './types';
@@ -135,8 +136,13 @@ export class HederaMirrornodeServiceDefaultImpl implements HederaMirrornodeServi
         );
       }
 
-      return (await response.json()) as TokenBalancesResponse;
+      return parseWithSchema(
+        TokenBalancesResponseSchema,
+        await response.json(),
+        `Mirror Node GET /accounts/${accountId}/tokens`,
+      );
     } catch (error) {
+      if (error instanceof CliError) throw error;
       if (error instanceof NotFoundError || error instanceof NetworkError)
         throw error;
       throw new NetworkError(

src/core/services/mirrornode/schemas.ts

@@ -6,6 +6,8 @@ import type {
   AccountListItemBalance,
   AccountListItemTokenBalance,
   GetAccountsAPIResponse,
+  TokenBalanceInfo,
+  TokenBalancesResponse,
   TokenInfo,
 } from './types';
 
@@ -101,3 +103,19 @@ export const GetAccountsAPIResponseSchema: z.ZodType<GetAccountsAPIResponse> =
       })
       .optional(),
   });
+
+export const TokenBalanceInfoSchema: z.ZodType<TokenBalanceInfo> = z.object({
+  token_id: z.string(),
+  balance: z.number(),
+  decimals: z.number().optional(),
+});
+
+export const TokenBalancesResponseSchema: z.ZodType<TokenBalancesResponse> =
+  z.object({
+    tokens: z.array(TokenBalanceInfoSchema),
+    links: z
+      .object({
+        next: z.string().nullable().optional(),
+      })
+      .optional(),
+  });

src/core/services/mirrornode/types.ts

@@ -49,18 +49,17 @@ export interface AccountResponse {
   keyAlgorithm: KeyAlgorithm;
 }
 
-// Token Balance Response
-export interface TokenBalancesResponse {
-  account: string;
-  balance: number;
-  tokens: TokenBalanceInfo[];
-  timestamp: string;
-}
-
 export interface TokenBalanceInfo {
   token_id: string;
   balance: number;
-  decimals: number;
+  decimals?: number;
+}
+
+export interface TokenBalancesResponse {
+  tokens: TokenBalanceInfo[];
+  links?: {
+    next?: string | null;
+  };
 }
 
 export type MirrorNodeTokenKey = {