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) {

test/keychange_listener_test.js

@@ -42,7 +42,7 @@ describe('KeyChangeListener', function() {
           done();
         });
       });
-      return store.isTrustedIdentity(phoneNumberWithKeyChange, newKey);
+      return store.saveIdentity(phoneNumberWithKeyChange, newKey);
     });
   });
 
@@ -66,7 +66,7 @@ describe('KeyChangeListener', function() {
           done();
         });
       });
-      return store.isTrustedIdentity(phoneNumberWithKeyChange, newKey);
+      return store.saveIdentity(phoneNumberWithKeyChange, newKey);
     });
 
   });

test/storage_test.js

@@ -56,7 +56,7 @@ describe("SignalProtocolStore", function() {
     describe('isTrustedIdentity', function() {
         it('returns true if a key is trusted', function(done) {
             store.saveIdentity(identifier, testKey.pubKey).then(function() {
-                store.isTrustedIdentity(identifier, testKey.pubKey).then(function(trusted) {
+                store.isTrustedIdentity(identifier, testKey.pubKey, store.Direction.RECEIVING).then(function(trusted) {
                     if (trusted) {
                         done();
                     } else {
@@ -68,7 +68,7 @@ describe("SignalProtocolStore", function() {
         it('returns false if a key is untrusted', function(done) {
             var newIdentity = libsignal.crypto.getRandomBytes(33);
             store.saveIdentity(identifier, testKey.pubKey).then(function() {
-                store.isTrustedIdentity(identifier, newIdentity).then(function(trusted) {
+                store.isTrustedIdentity(identifier, newIdentity, store.Direction.SENDING).then(function(trusted) {
                     if (trusted) {
                         done(new Error('Allowed to overwrite identity key'));
                     } else {