Fix tests

// FREEBIE

libtextsecure/test/in_memory_signal_protocol_store.js

@@ -3,6 +3,7 @@ function SignalProtocolStore() {
 }
 
 SignalProtocolStore.prototype = {
+  Direction: { SENDING: 1, RECEIVING: 2},
 	getIdentityKeyPair: function() {
 		return Promise.resolve(this.get('identityKey'));
 	},
@@ -53,7 +54,13 @@ SignalProtocolStore.prototype = {
 		if (identifier === null || identifier === undefined)
 			throw new Error("Tried to put identity key for undefined/null key");
 		return new Promise(function(resolve) {
-			resolve(this.put('identityKey' + identifier, identityKey));
+      var existing = this.get('identityKey' + identifier);
+			this.put('identityKey' + identifier, identityKey);
+      if (existing && existing !== identityKey) {
+        resolve(true);
+      } else {
+        resolve(false);
+      }
 		}.bind(this));
 	},
 

libtextsecure/test/protocol_wrapper_test.js

@@ -14,7 +14,9 @@ describe('Protocol Wrapper', function() {
         localStorage.clear();
         libsignal.KeyHelper.generateIdentityKeyPair().then(function(identityKey) {
             return textsecure.storage.protocol.saveIdentity(identifier, identityKey);
-        }).then(done);
+            }).then(function() {
+              done();
+            });
     });
     describe('processPreKey', function() {
         it('rejects if the identity key changes', function(done) {