From 2ba20aee9194f28ce07f8499083b173a58d0f084 Mon Sep 17 00:00:00 2001
From: thisyahlen <104053934+thisyahlen-deriv@users.noreply.github.com>
Date: Thu, 21 Mar 2024 14:45:05 +0800
Subject: [PATCH] test: unit test for usePlatformAssets (#14289)

---
 .../hooks/__tests__/usePlatformAssets.spec.ts | 132 ++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 packages/tradershub/src/hooks/__tests__/usePlatformAssets.spec.ts

diff --git a/packages/tradershub/src/hooks/__tests__/usePlatformAssets.spec.ts b/packages/tradershub/src/hooks/__tests__/usePlatformAssets.spec.ts
new file mode 100644
index 000000000000..ab58943aa604
--- /dev/null
+++ b/packages/tradershub/src/hooks/__tests__/usePlatformAssets.spec.ts
@@ -0,0 +1,132 @@
+import { Regulation } from '@/constants';
+import { useExchangeRates, useTradingAccountsList } from '@deriv/api-v2';
+import { renderHook } from '@testing-library/react-hooks';
+import usePlatformAssets from '../usePlatformAssets';
+
+jest.mock('@deriv/api-v2', () => ({
+    useExchangeRates: jest.fn(),
+    useTradingAccountsList: jest.fn(),
+}));
+
+describe('usePlatformAssets', () => {
+    beforeEach(() => {
+        (useTradingAccountsList as jest.Mock).mockReturnValue({
+            data: [
+                { broker: 'MF', currency: 'USD', is_virtual: false, balance: 100 },
+                { broker: 'CR', currency: 'EUR', is_virtual: false, balance: 200 },
+                { broker: 'MF', currency: 'GBP', is_virtual: true, balance: 300 },
+            ],
+            fiat_account: 'USD',
+            isSuccess: true,
+        });
+
+        (useExchangeRates as jest.Mock).mockReturnValue({
+            data: {},
+            getExchangeRate: () => 1,
+            subscribe: jest.fn(),
+            unsubscribe: jest.fn(),
+        });
+    });
+
+    it('should return correct balances and currency for EU regulation', () => {
+        const { result } = renderHook(() => usePlatformAssets(Regulation.EU));
+
+        expect(result.current.demoAccountBalance).toBe(300);
+        expect(result.current.fiatCurrency).toBe('USD');
+        expect(result.current.isSuccess).toBe(true);
+        expect(result.current.realAccounts).toHaveLength(2);
+        expect(result.current.totalRealPlatformBalance).toBe(100);
+    });
+
+    it('should return correct balances and currency for non-EU regulation', () => {
+        const { result } = renderHook(() => usePlatformAssets(Regulation.NonEU));
+
+        expect(result.current.fiatCurrency).toBe('USD');
+        expect(result.current.isSuccess).toBe(true);
+        expect(result.current.realAccounts).toHaveLength(2);
+        expect(result.current.totalRealPlatformBalance).toBe(200);
+    });
+
+    it('should return 0 for demoAccountBalance when demo account has no balance', () => {
+        (useTradingAccountsList as jest.Mock).mockReturnValue({
+            data: [
+                { broker: 'MF', currency: 'USD', is_virtual: false, balance: 100 },
+                { broker: 'CR', currency: 'EUR', is_virtual: false, balance: 200 },
+                { broker: 'MF', currency: 'GBP', is_virtual: true, balance: undefined },
+            ],
+            fiat_account: 'USD',
+            isSuccess: true,
+        });
+
+        const { result } = renderHook(() => usePlatformAssets(Regulation.EU));
+
+        expect(result.current.demoAccountBalance).toBe(0);
+    });
+
+    it('should use fallback values in totalRealPlatformBalance calculation', () => {
+        (useTradingAccountsList as jest.Mock).mockReturnValue({
+            data: [
+                { broker: 'MF', currency: undefined, is_virtual: false, balance: 100 },
+                { broker: 'CR', currency: undefined, is_virtual: false, balance: 200 },
+                { broker: 'MF', currency: 'GBP', is_virtual: true, balance: 300 },
+            ],
+            fiat_account: undefined,
+            isSuccess: true,
+        });
+
+        const getExchangeRateMock = jest.fn().mockReturnValue(1);
+        (useExchangeRates as jest.Mock).mockReturnValue({
+            getExchangeRate: getExchangeRateMock,
+        });
+
+        const { result } = renderHook(() => usePlatformAssets(Regulation.EU));
+
+        expect(getExchangeRateMock).toHaveBeenCalledWith('USD', 'USD');
+
+        expect(result.current.totalRealPlatformBalance).toBe(100);
+    });
+
+    it('should call multiSubscribe with an empty target_currencies array when there are no real accounts', () => {
+        (useTradingAccountsList as jest.Mock).mockReturnValue({
+            data: undefined, // no real accounts
+            fiat_account: 'USD',
+            isSuccess: true,
+        });
+
+        const multiSubscribeMock = jest.fn();
+        (useExchangeRates as jest.Mock).mockReturnValue({
+            subscribe: multiSubscribeMock,
+        });
+
+        renderHook(() => usePlatformAssets(Regulation.EU));
+
+        if (multiSubscribeMock.mock.calls.length > 0) {
+            expect(multiSubscribeMock.mock.calls[0][0]).toEqual({
+                base_currency: 'USD',
+                target_currencies: [],
+            });
+        }
+    });
+
+    it('should use fallback value for account balance in totalRealPlatformBalance calculation', () => {
+        (useTradingAccountsList as jest.Mock).mockReturnValue({
+            data: [
+                { broker: 'MF', currency: 'USD', is_virtual: false, balance: undefined }, // real account with no balance
+                { broker: 'CR', currency: 'EUR', is_virtual: false, balance: 200 },
+                { broker: 'MF', currency: 'GBP', is_virtual: true, balance: 300 },
+            ],
+            fiat_account: 'USD',
+            isSuccess: true,
+        });
+
+        const getExchangeRateMock = jest.fn().mockReturnValue(1);
+        (useExchangeRates as jest.Mock).mockReturnValue({
+            getExchangeRate: getExchangeRateMock,
+            subscribe: jest.fn(),
+        });
+
+        const { result } = renderHook(() => usePlatformAssets(Regulation.EU));
+
+        expect(result.current.totalRealPlatformBalance).toBe(0);
+    });
+});