shontzu/WALL-865/Incorrect-total-asset-amount

packages/hooks/src/__tests__/useTotalAccountBalance.spec.tsx

@@ -17,23 +17,15 @@ describe('useTotalAccountBalance', () => {
 
     test('should return total balance correctly when user has multiple accounts', async () => {
         const mock = mockStore({
-            exchange_rates: {
-                data: {
-                    rates: {
-                        EUR: 2,
-                        AUD: 3,
-                    },
-                },
-            },
             client: {
                 active_accounts: [
                     {
-                        currency: 'AUD',
-                        balance: 300,
+                        currency: 'USD',
+                        balance: 10000,
                     },
                     {
                         currency: 'EUR',
-                        balance: 200,
+                        balance: 10000,
                     },
                 ],
             },
@@ -44,6 +36,6 @@ describe('useTotalAccountBalance', () => {
         );
         const { result } = renderHook(() => useTotalAccountBalance(mock.client.active_accounts), { wrapper });
 
-        expect(result.current.balance).toBe(200);
+        expect(result.current.balance).toBe(20000);
     });
 });

packages/hooks/src/useTotalAccountBalance.ts

@@ -1,22 +1,23 @@
 import useRealTotalAssetCurrency from './useTotalAssetCurrency';
 import useExchangeRate from './useExchangeRate';
-
+import { useStore } from '@deriv/stores';
 /**
  * we can use this hook to get the total balance of the given accounts list.
  * it loops through the accounts list and adds the balance of each account
  * to the total balance, it also converts the balance to the currency of the
  * first account in the list
  */
-
 const useTotalAccountBalance = (accounts: { balance?: number; currency?: string }[]) => {
+    const { traders_hub } = useStore();
+    const { is_demo } = traders_hub;
     const total_assets_real_currency = useRealTotalAssetCurrency();
     const { getRate } = useExchangeRate();
 
     if (!accounts.length) return { balance: 0, currency: total_assets_real_currency };
 
     const balance = accounts.reduce((total, account) => {
-        const base_rate = getRate(total_assets_real_currency || '');
-        const rate = getRate(account.currency || total_assets_real_currency || '');
+        const base_rate = is_demo ? getRate('USD') : getRate(total_assets_real_currency || '');
+        const rate = getRate(total_assets_real_currency || '');
         const exchange_rate = base_rate / rate;
 
         return total + (account.balance || 0) * exchange_rate;