Add WETH balance to ETH balance PM-666 (#397)

gnosis · Jun 7, 2018 · 95382d3 · 95382d3
1 parent 66b62d2
commit 95382d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/routes/MarketDetails/components/ExpandableViews/MarketBuySharesForm/index.js b/src/routes/MarketDetails/components/ExpandableViews/MarketBuySharesForm/index.js
@@ -77,8 +77,10 @@ class MarketBuySharesForm extends Component {
 
     let balance = currentBalance
 
-    if (collateralTokenSymbol !== 'WETH') {
+    if (!/^(WETH|ETH|W-ETH)$/.test(collateralTokenSymbol)) {
       balance = collateralTokenBalance.div(1e18)
+    } else {
+      balance = Decimal(balance).add(collateralTokenBalance.div(1e18))
     }
 
     if (decimalValue.gt(balance)) {