From cdd1e9ea58c0807cf15252b4cce5b588e3c1d1cd Mon Sep 17 00:00:00 2001
From: Ivan Socolsky <jungans@gmail.com>
Date: Wed, 1 Apr 2015 17:11:39 -0300
Subject: [PATCH] test wallet recreation

---
 test/integration/server.js | 51 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 50 insertions(+), 1 deletion(-)

diff --git a/test/integration/server.js b/test/integration/server.js
index e477989..03aa772 100644
--- a/test/integration/server.js
+++ b/test/integration/server.js
@@ -2481,7 +2481,13 @@ describe('Wallet service', function() {
   });
 
   describe('#scan', function() {
-    WalletService.scanConfig.SCAN_WINDOW = 2;
+    var scanWindowOld = WalletService.scanConfig.SCAN_WINDOW;
+    beforeEach(function() {
+      WalletService.scanConfig.SCAN_WINDOW = 2;
+    });
+    afterEach(function() {
+      WalletService.scanConfig.SCAN_WINDOW = scanWindowOld;
+    });
 
     it('should scan main addresses', function(done) {
       helpers.stubAddressActivity(['3K2VWMXheGZ4qG35DyGjA2dLeKfaSr534A']);
@@ -2539,6 +2545,49 @@ describe('Wallet service', function() {
         });
       });
     });
+    it('should restore wallet balance', function(done) {
+      async.waterfall([
+
+        function(next) {
+          helpers.createAndJoinWallet(1, 2, function(server, wallet) {
+            helpers.stubUtxos(server, wallet, [1, 2, 3], function(utxos) {
+              should.exist(utxos);
+              helpers.stubAddressActivity(_.pluck(utxos, 'address'));
+              server.getBalance({}, function(err, balance) {
+                balance.totalAmount.should.equal(helpers.toSatoshi(6));
+                next(null, server, wallet);
+              });
+            });
+          });
+        },
+        function(server, wallet, next) {
+          server.removeWallet({}, function(err) {
+            next(err);
+          });
+        },
+        function(next) {
+          // NOTE: this works because it creates the exact same wallet!
+          helpers.createAndJoinWallet(1, 2, function(server, wallet) {
+            server.getBalance({}, function(err, balance) {
+              balance.totalAmount.should.equal(0);
+              next(null, server, wallet);
+            });
+          });
+        },
+        function(server, wallet, next) {
+          server.scan({}, function(err) {
+            should.not.exist(err);
+            server.getBalance(wallet.id, function(err, balance) {
+              balance.totalAmount.should.equal(helpers.toSatoshi(6));
+              next();
+            })
+          });
+        },
+      ], function(err) {
+        should.not.exist(err);
+        done();
+      });
+    });
   });
 });