Revert back to original formatBalance

1 files changed, 17 insertions, 5 deletions

diff --git a/test/unit/util_test.js b/test/unit/util_test.js
index 514a2a432..e2390c8d4 100644
--- a/test/unit/util_test.js
+++ b/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')
     })
 
   })