From bb726bac8b9ee845add38a07915261ceb5c675a4 Mon Sep 17 00:00:00 2001
From: Braydon Fuller <courier@braydon.com>
Date: Fri, 15 Apr 2016 16:35:31 -0400
Subject: [PATCH] test: bitcoind getaddressbalance unit test

---
 test/services/bitcoind.unit.js | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/test/services/bitcoind.unit.js b/test/services/bitcoind.unit.js
index a994b9d9..b638d1ff 100644
--- a/test/services/bitcoind.unit.js
+++ b/test/services/bitcoind.unit.js
@@ -882,6 +882,43 @@ describe('Bitcoin Service', function() {
   });
 
   describe('#getAddressBalance', function() {
+    it('will give rpc error', function(done) {
+      var bitcoind = new BitcoinService(baseConfig);
+      bitcoind.nodes.push({
+        client: {
+          getAddressBalance: sinon.stub().callsArgWith(1, {code: -1, message: 'Test error'})
+        }
+      });
+      var address = '1Cj4UZWnGWAJH1CweTMgPLQMn26WRMfXmo';
+      var options = {};
+      bitcoind.getAddressBalance(address, options, function(err) {
+        err.should.be.instanceof(Error);
+        done();
+      });
+    });
+    it('will give balance', function(done) {
+      var bitcoind = new BitcoinService(baseConfig);
+      bitcoind.nodes.push({
+        client: {
+          getAddressBalance: sinon.stub().callsArgWith(1, null, {
+            result: {
+              received: 100000,
+              balance: 10000
+            }
+          })
+        }
+      });
+      var address = '1Cj4UZWnGWAJH1CweTMgPLQMn26WRMfXmo';
+      var options = {};
+      bitcoind.getAddressBalance(address, options, function(err, data) {
+        if (err) {
+          return done(err);
+        }
+        data.balance.should.equal(10000);
+        data.received.should.equal(100000);
+        done();
+      });
+    });
   });
 
   describe('#getAddressUnspentOutputs', function() {