From 3e38d84d0b93cb0c19b7ebf55b9a2aaacc6cc96e Mon Sep 17 00:00:00 2001
From: Patrick Nagurny <patrick@bitpay.com>
Date: Fri, 2 Oct 2015 17:10:12 -0400
Subject: [PATCH] emit address/balance events for mempool transactions

---
 lib/services/address/index.js       | 1 +
 test/services/address/index.unit.js | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/services/address/index.js b/lib/services/address/index.js
index 75878d70..df2a593c 100644
--- a/lib/services/address/index.js
+++ b/lib/services/address/index.js
@@ -160,6 +160,7 @@ AddressService.prototype.transactionHandler = function(txInfo) {
 
   for (var key in messages) {
     this.transactionEventHandler(messages[key]);
+    this.balanceEventHandler(null, messages[key].addressInfo);
   }
 };
 
diff --git a/test/services/address/index.unit.js b/test/services/address/index.unit.js
index 14cc1eda..17655b49 100644
--- a/test/services/address/index.unit.js
+++ b/test/services/address/index.unit.js
@@ -92,7 +92,7 @@ describe('Address Service', function() {
   });
 
   describe('#transactionHandler', function() {
-    it('will pass outputs to transactionOutputHandler and call transactionEventHandler', function() {
+    it('will pass outputs to transactionOutputHandler and call transactionEventHandler and balanceEventHandler', function() {
       var txBuf = new Buffer('01000000010000000000000000000000000000000000000000000000000000000000000000ffffffff0704ffff001d0104ffffffff0100f2052a0100000043410496b538e853519c726a2c91e61ec11600ae1390813a627c66fb8be7947be63c52da7589379515d4e0a604f8141781e62294721166bf621e73a82cbf2342c858eeac00000000', 'hex');
       var am = new AddressService({node: mocknode});
       var address = '12c6DSiU4Rq3P4ZxziKxzrL5LmMBrzjrJX';
@@ -101,10 +101,12 @@ describe('Address Service', function() {
         messages[address] = message;
       };
       am.transactionEventHandler = sinon.spy();
+      am.balanceEventHandler = sinon.spy();
       am.transactionHandler({
         buffer: txBuf
       });
       am.transactionEventHandler.callCount.should.equal(1);
+      am.balanceEventHandler.callCount.should.equal(1);
     });
   });