signal-desktop/libtextsecure/test/generate_keys_test.js

'use strict';

describe('Key generation', function() {
  var count = 10;
  this.timeout(count * 2000);

  function validateStoredKeyPair(keyPair) {
    /* Ensure the keypair matches the format used internally by libsignal-protocol */
    assert.isObject(keyPair, 'Stored keyPair is not an object');
    assert.instanceOf(keyPair.pubKey, ArrayBuffer);
    assert.instanceOf(keyPair.privKey, ArrayBuffer);
    assert.strictEqual(keyPair.pubKey.byteLength, 33);
    assert.strictEqual(new Uint8Array(keyPair.pubKey)[0], 5);
    assert.strictEqual(keyPair.privKey.byteLength, 32);
  }
  function itStoresPreKey(keyId) {
    it('prekey ' + keyId + ' is valid', function(done) {
      return textsecure.storage.protocol
        .loadPreKey(keyId)
        .then(function(keyPair) {
          validateStoredKeyPair(keyPair);
        })
        .then(done, done);
    });
  }
  function itStoresSignedPreKey(keyId) {
    it('signed prekey ' + keyId + ' is valid', function(done) {
      return textsecure.storage.protocol
        .loadSignedPreKey(keyId)
        .then(function(keyPair) {
          validateStoredKeyPair(keyPair);
        })
        .then(done, done);
    });
  }
  function validateResultKey(resultKey) {
    return textsecure.storage.protocol
      .loadPreKey(resultKey.keyId)
      .then(function(keyPair) {
        assertEqualArrayBuffers(resultKey.publicKey, keyPair.pubKey);
      });
  }
  function validateResultSignedKey(resultSignedKey) {
    return textsecure.storage.protocol
      .loadSignedPreKey(resultSignedKey.keyId)
      .then(function(keyPair) {
        assertEqualArrayBuffers(resultSignedKey.publicKey, keyPair.pubKey);
      });
  }

  before(function(done) {
    localStorage.clear();
    libsignal.KeyHelper.generateIdentityKeyPair()
      .then(function(keyPair) {
        return textsecure.storage.protocol.put('identityKey', keyPair);
      })
      .then(done, done);
  });

  describe('the first time', function() {
    var result;
    /* result should have this format
         * {
         *   preKeys: [ { keyId, publicKey }, ... ],
         *   signedPreKey: { keyId, publicKey, signature },
         *   identityKey: <ArrayBuffer>
         * }
         */
    before(function(done) {
      var accountManager = new textsecure.AccountManager('');
      accountManager
        .generateKeys(count)
        .then(function(res) {
          result = res;
        })
        .then(done, done);
    });
    for (var i = 1; i <= count; i++) {
      itStoresPreKey(i);
    }
    itStoresSignedPreKey(1);

    it('result contains ' + count + ' preKeys', function() {
      assert.isArray(result.preKeys);
      assert.lengthOf(result.preKeys, count);
      for (var i = 0; i < count; i++) {
        assert.isObject(result.preKeys[i]);
      }
    });
    it('result contains the correct keyIds', function() {
      for (var i = 0; i < count; i++) {
        assert.strictEqual(result.preKeys[i].keyId, i + 1);
      }
    });
    it('result contains the correct public keys', function(done) {
      Promise.all(result.preKeys.map(validateResultKey))
        .then(function() {
          done();
        })
        .catch(done);
    });
    it('returns a signed prekey', function(done) {
      assert.strictEqual(result.signedPreKey.keyId, 1);
      assert.instanceOf(result.signedPreKey.signature, ArrayBuffer);
      validateResultSignedKey(result.signedPreKey).then(done, done);
    });
  });
  describe('the second time', function() {
    var result;
    before(function(done) {
      var accountManager = new textsecure.AccountManager('');
      accountManager
        .generateKeys(count)
        .then(function(res) {
          result = res;
        })
        .then(done, done);
    });
    for (var i = 1; i <= 2 * count; i++) {
      itStoresPreKey(i);
    }
    itStoresSignedPreKey(1);
    itStoresSignedPreKey(2);
    it('result contains ' + count + ' preKeys', function() {
      assert.isArray(result.preKeys);
      assert.lengthOf(result.preKeys, count);
      for (var i = 0; i < count; i++) {
        assert.isObject(result.preKeys[i]);
      }
    });
    it('result contains the correct keyIds', function() {
      for (var i = 1; i <= count; i++) {
        assert.strictEqual(result.preKeys[i - 1].keyId, i + count);
      }
    });
    it('result contains the correct public keys', function(done) {
      Promise.all(result.preKeys.map(validateResultKey))
        .then(function() {
          done();
        })
        .catch(done);
    });
    it('returns a signed prekey', function(done) {
      assert.strictEqual(result.signedPreKey.keyId, 2);
      assert.instanceOf(result.signedPreKey.signature, ArrayBuffer);
      validateResultSignedKey(result.signedPreKey).then(done, done);
    });
  });
  describe('the third time', function() {
    var result;
    before(function(done) {
      var accountManager = new textsecure.AccountManager('');
      accountManager
        .generateKeys(count)
        .then(function(res) {
          result = res;
        })
        .then(done, done);
    });
    for (var i = 1; i <= 3 * count; i++) {
      itStoresPreKey(i);
    }
    itStoresSignedPreKey(2);
    itStoresSignedPreKey(3);
    it('result contains ' + count + ' preKeys', function() {
      assert.isArray(result.preKeys);
      assert.lengthOf(result.preKeys, count);
      for (var i = 0; i < count; i++) {
        assert.isObject(result.preKeys[i]);
      }
    });
    it('result contains the correct keyIds', function() {
      for (var i = 1; i <= count; i++) {
        assert.strictEqual(result.preKeys[i - 1].keyId, i + 2 * count);
      }
    });
    it('result contains the correct public keys', function(done) {
      Promise.all(result.preKeys.map(validateResultKey))
        .then(function() {
          done();
        })
        .catch(done);
    });
    it('result contains a signed prekey', function(done) {
      assert.strictEqual(result.signedPreKey.keyId, 3);
      assert.instanceOf(result.signedPreKey.signature, ArrayBuffer);
      validateResultSignedKey(result.signedPreKey).then(done, done);
    });
  });
});