Merge pull request #410 from MetaMask/cleanupEthbug

Revert back to original formatBalance

test/unit/util_test.js

@@ -142,23 +142,35 @@ describe('util', function() {
 
     it('when given nothing', function() {
       var result = util.formatBalance()
-      assert.equal(result.formatted, 'None', 'should return "None"')
+      assert.equal(result, 'None', 'should return "None"')
+    })
+
+    it('should return eth as string followed by ETH', function() {
+      var input = new ethUtil.BN(ethInWei, 10).toJSON()
+      var result = util.formatBalance(input, 4)
+      assert.equal(result, '1.0000 ETH')
+    })
+
+    it('should return eth as string followed by ETH', function() {
+      var input = new ethUtil.BN(ethInWei, 10).div(new ethUtil.BN('2', 10)).toJSON()
+      var result = util.formatBalance(input, 3)
+      assert.equal(result, '0.500 ETH')
     })
 
     it('should display specified decimal points', function() {
       var input = "0x128dfa6a90b28000"
       var result = util.formatBalance(input, 2)
-      assert.equal(result.formatted, '1.33 ETH')
+      assert.equal(result, '1.33 ETH')
     })
-    it('should default to 2 decimal points', function() {
+    it('should default to 3 decimal points', function() {
       var input = "0x128dfa6a90b28000"
       var result = util.formatBalance(input)
-      assert.equal(result.formatted, '1.33 ETH')
+      assert.equal(result, '1.337 ETH')
     })
     it('should show 2 significant digits for tiny balances', function() {
       var input = "0x1230fa6a90b28"
       var result = util.formatBalance(input)
-      assert.equal(result.formatted, '0.00032 ETH')
+      assert.equal(result, '0.00032 ETH')
     })
 
   })

ui/app/components/eth-balance.js

@@ -2,7 +2,6 @@ const Component = require('react').Component
 const h = require('react-hyperscript')
 const inherits = require('util').inherits
 const formatBalance = require('../util').formatBalance
-const Tooltip = require('./tooltip')
 
 module.exports = EthBalanceComponent
 
@@ -31,32 +30,28 @@ EthBalanceComponent.prototype.render = function () {
   )
 }
 EthBalanceComponent.prototype.renderBalance = function (value) {
+
   if (value === 'None') return value
 
-  var balance = value.formatted.split(' ')[0]
+  var balance = value.split(' ')[0]
-  var label = value.formatted.split(' ')[1]
+  var label = value.split(' ')[1]
 
   return (
-    h(Tooltip, {
+    h('.flex-column', {
-      title: value.balance,
+      style: {
-      position: 'bottom',
+        alignItems: 'flex-end',
+        lineHeight: '13px',
+        fontFamily: 'Montserrat Thin',
+        textRendering: 'geometricPrecision',
+      },
     }, [
-      h('.flex-column', {
+      h('div', balance),
+      h('div', {
         style: {
-          alignItems: 'flex-end',
+          color: ' #AEAEAE',
-          lineHeight: '13px',
+          fontSize: '12px',
-          fontFamily: 'Montserrat Light',
-          textRendering: 'geometricPrecision',
         },
-      }, [
+      }, label),
-        h('div', balance),
-        h('div', {
-          style: {
-            color: ' #AEAEAE',
-            fontSize: '12px',
-          },
-        }, label),
-      ]),
     ])
   )
 }

ui/app/util.js

@@ -99,29 +99,22 @@ function formatBalance (balance, decimalsToKeep) {
   var parsed = parseBalance(balance)
   var beforeDecimal = parsed[0]
   var afterDecimal = parsed[1]
-  var formatted, formattedBalance
+  var formatted = 'None'
-
+  if (decimalsToKeep === undefined) {
-  if (beforeDecimal === '0') {
+    if (beforeDecimal === '0') {
-    if (afterDecimal !== '0') {
+      if (afterDecimal !== '0') {
-      var sigFigs = afterDecimal.match(/^0*(.{2})/) // default: grabs 2 most significant digits
+        var sigFigs = afterDecimal.match(/^0*(.{2})/) // default: grabs 2 most significant digits
-      if (sigFigs) { afterDecimal = sigFigs[0] }
+        if (sigFigs) { afterDecimal = sigFigs[0] }
-      formattedBalance = afterDecimal.substr(0, 5) === '00000' ? '<0.00001' : `0.${afterDecimal.slice(0, 6)}`
+        formatted = '0.' + afterDecimal + ' ETH'
+      }
+    } else {
+      formatted = beforeDecimal + '.' + afterDecimal.slice(0, 3) + ' ETH'
     }
   } else {
-    formattedBalance = `${beforeDecimal}.${afterDecimal.slice(0, 2)}`
+    afterDecimal += Array(decimalsToKeep).join('0')
-  }
+    formatted = beforeDecimal + '.' + afterDecimal.slice(0, decimalsToKeep) + ' ETH'
-  if (decimalsToKeep) {
-    formattedBalance = `${beforeDecimal}.${afterDecimal.slice(0, decimalsToKeep)}`
-  }
-
-  formatted = `${formattedBalance} ETH`
-
-  if (formattedBalance === '0.0' || formattedBalance === undefined) {
-    formatted = 'None'
-    formattedBalance = 'None'
   }
-
+  return formatted
-  return {formattedBalance, balance: parsed.join('.'), formatted}
 }
 
 function dataSize (data) {