Merge pull request #41 from matrix-org/rav/js_tests

Add some tests for the Javascript wrappers

2 files changed, 162 insertions, 0 deletions

diff --git a/javascript/test/megolm.spec.js b/javascript/test/megolm.spec.js
new file mode 100644
index 0000000..8f9d24a
--- /dev/null
+++ b/javascript/test/megolm.spec.js
@@ -0,0 +1,68 @@
+/*
+Copyright 2016 OpenMarket Ltd
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+"use strict";
+
+var Olm = require('../olm');
+
+describe("megolm", function() {
+    var aliceSession, bobSession;
+
+    beforeEach(function() {
+        aliceSession = new Olm.OutboundGroupSession();
+        bobSession = new Olm.InboundGroupSession();
+    });
+
+    afterEach(function() {
+        if (aliceSession !== undefined) {
+            aliceSession.free();
+            aliceSession = undefined;
+        }
+
+        if (bobSession !== undefined) {
+            bobSession.free();
+            bobSession = undefined;
+        }
+    });
+
+    it("should encrypt and decrypt", function() {
+        aliceSession.create();
+        expect(aliceSession.message_index()).toEqual(0);
+        bobSession.create(aliceSession.session_key());
+
+        var TEST_TEXT='têst1';
+        var encrypted = aliceSession.encrypt(TEST_TEXT);
+        var decrypted = bobSession.decrypt(encrypted);
+        console.log(TEST_TEXT, "->", decrypted);
+        expect(decrypted.plaintext).toEqual(TEST_TEXT);
+        expect(decrypted.message_index).toEqual(0);
+
+        TEST_TEXT='hot beverage: ☕';
+        encrypted = aliceSession.encrypt(TEST_TEXT);
+        decrypted = bobSession.decrypt(encrypted);
+        console.log(TEST_TEXT, "->", decrypted);
+        expect(decrypted.plaintext).toEqual(TEST_TEXT);
+        expect(decrypted.message_index).toEqual(1);
+
+        // shorter text, to spot buffer overruns
+        TEST_TEXT='☕';
+        encrypted = aliceSession.encrypt(TEST_TEXT);
+        decrypted = bobSession.decrypt(encrypted);
+        console.log(TEST_TEXT, "->", decrypted);
+        expect(decrypted.plaintext).toEqual(TEST_TEXT);
+        expect(decrypted.message_index).toEqual(2);
+    });
+});
diff --git a/javascript/test/olm.spec.js b/javascript/test/olm.spec.js
new file mode 100644
index 0000000..b7cc3ae
--- /dev/null
+++ b/javascript/test/olm.spec.js
@@ -0,0 +1,94 @@
+/*
+Copyright 2016 OpenMarket Ltd
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+"use strict";
+
+var Olm = require('../olm');
+
+if (!Object.keys) {
+    Object.keys = function(o) {
+        var k=[], p;
+        for (p in o) if (Object.prototype.hasOwnProperty.call(o,p)) k.push(p);
+        return k;
+    }
+}
+
+describe("olm", function() {
+    var aliceAccount, bobAccount;
+    var aliceSession, bobSession;
+
+    beforeEach(function() {
+        aliceAccount = new Olm.Account();
+        bobAccount = new Olm.Account();
+        aliceSession = new Olm.Session();
+        bobSession = new Olm.Session();
+    });
+
+    afterEach(function() {
+        if (aliceAccount !== undefined) {
+            aliceAccount.free();
+            aliceAccount = undefined;
+        }
+
+        if (bobAccount !== undefined) {
+            bobAccount.free();
+            bobAccount = undefined;
+        }
+
+        if (aliceSession !== undefined) {
+            aliceSession.free();
+            aliceSession = undefined;
+        }
+
+        if (bobSession !== undefined) {
+            bobSession.free();
+            bobSession = undefined;
+        }
+    });
+
+    it('should encrypt and decrypt', function() {
+        aliceAccount.create();
+        bobAccount.create();
+
+        bobAccount.generate_one_time_keys(1);
+        var bobOneTimeKeys = JSON.parse(bobAccount.one_time_keys()).curve25519;
+        bobAccount.mark_keys_as_published();
+
+        var bobIdKey = JSON.parse(bobAccount.identity_keys()).curve25519;
+
+        var otk_id = Object.keys(bobOneTimeKeys)[0];
+
+        aliceSession.create_outbound(
+            aliceAccount, bobIdKey, bobOneTimeKeys[otk_id]
+        );
+
+        var TEST_TEXT='têst1';
+        var encrypted = aliceSession.encrypt(TEST_TEXT);
+        expect(encrypted.type).toEqual(0);
+        bobSession.create_inbound(bobAccount, encrypted.body);
+        bobAccount.remove_one_time_keys(bobSession);
+        var decrypted = bobSession.decrypt(encrypted.type, encrypted.body);
+        console.log(TEST_TEXT, "->", decrypted);
+        expect(decrypted).toEqual(TEST_TEXT);
+
+        TEST_TEXT='hot beverage: ☕';
+        encrypted = bobSession.encrypt(TEST_TEXT);
+        expect(encrypted.type).toEqual(1);
+        decrypted = aliceSession.decrypt(encrypted.type, encrypted.body);
+        console.log(TEST_TEXT, "->", decrypted);
+        expect(decrypted).toEqual(TEST_TEXT);
+    });
+});