From 3c8caacc9270bda12df21287a4e77c97133140d0 Mon Sep 17 00:00:00 2001
From: Matias Alejo Garcia <ematiu@gmail.com>
Date: Fri, 24 Oct 2014 10:48:23 -0300
Subject: [PATCH] .

---
 test/Identity.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/Identity.js b/test/Identity.js
index 3169d1925..e999e5411 100644
--- a/test/Identity.js
+++ b/test/Identity.js
@@ -34,6 +34,7 @@ describe('Identity model', function() {
   beforeEach(function(done) {
     storage = sinon.stub();
     storage.getItem = sinon.stub();
+    storage.set = sinon.stub().yields(null);
     storage.savePassphrase = sinon.spy();
     storage.restorePassphrase = sinon.spy();
     storage.setPassword = sinon.spy();
@@ -45,8 +46,10 @@ describe('Identity model', function() {
 
 
     wallet = sinon.stub();
-    wallet.store = sinon.stub().yields(null);
+    wallet.on = sinon.stub().yields(null);
     wallet.netStart = sinon.stub();
+    wallet.toObj = sinon.stub();
+    wallet.getName = sinon.stub().returns('walletname');
     wallet.getId = sinon.stub().returns('wid:123');
     Identity._newWallet = sinon.stub().returns(wallet);