feat(balances): fix lb balance selector in total balance

blockchain · Mar 30, 2020 · 6a147ad · 6a147ad
1 parent f9ea32c
commit 6a147ad
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/packages/blockchain-wallet-v4-frontend/src/components/Balances/total/selectors.js b/packages/blockchain-wallet-v4-frontend/src/components/Balances/total/selectors.js
@@ -23,11 +23,13 @@ export const getBtcBalance = createDeepEqualSelector(
           )
         )
   ],
-  (walletBalancesR, lockboxBalancesR) => {
-    const modulesToBalance = (walletBalance, lockboxBalance) => {
-      return lockboxBalance.concat(walletBalance)
+  (walletBalance, lockboxBalancesR) => {
+    const modulesToBalance = (walletBalance, lockboxBalances) => {
+      const lbBalances = lockboxBalances.map(b => b.getOrElse(0))
+      return lbBalances.concat(walletBalance)
     }
-    const balancesR = lift(modulesToBalance)(walletBalancesR, lockboxBalancesR)
+    const balancesR = lift(modulesToBalance)(walletBalance, lockboxBalancesR)
+
     return balancesR.map(reduce(add, 0))
   }
 )
@@ -45,8 +47,9 @@ export const getBchBalance = createDeepEqualSelector(
         )
   ],
   (walletBalancesR, lockboxBalancesR) => {
-    const modulesToBalance = (walletBalance, lockboxBalance) => {
-      return lockboxBalance.concat(walletBalance)
+    const modulesToBalance = (walletBalance, lockboxBalances) => {
+      const lbBalances = lockboxBalances.map(b => b.getOrElse(0))
+      return lbBalances.concat(walletBalance)
     }
     const balancesR = lift(modulesToBalance)(walletBalancesR, lockboxBalancesR)
     return balancesR.map(reduce(add, 0))
@@ -169,8 +172,8 @@ export const getTotalBalance = createDeepEqualSelector(
     selectors.router.getPathname
   ],
   (
-    btcBalanceInfoR,
     bchBalanceInfoR,
+    btcBalanceInfoR,
     ethBalanceInfoR,
     paxBalanceInfoR,
     xlmBalanceInfoR,