1 files changed, 137 insertions, 0 deletions
diff --git a/comm/mail/extensions/openpgp/test/unit/rnp/test_strip.js b/comm/mail/extensions/openpgp/test/unit/rnp/test_strip.js
new file mode 100644
index 0000000000..4260921ef2
--- /dev/null
+++ b/comm/mail/extensions/openpgp/test/unit/rnp/test_strip.js
@@ -0,0 +1,137 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+/**
+ * Tests stripping keys.
+ */
+
+"use strict";
+
+const { RNP } = ChromeUtils.import("chrome://openpgp/content/modules/RNP.jsm");
+const { EnigmailKeyRing } = ChromeUtils.import(
+  "chrome://openpgp/content/modules/keyRing.jsm"
+);
+const { OpenPGPTestUtils } = ChromeUtils.import(
+  "resource://testing-common/mozmill/OpenPGPTestUtils.jsm"
+);
+const { MailStringUtils } = ChromeUtils.import(
+  "resource:///modules/MailStringUtils.jsm"
+);
+
+const keyDir = "../../../../../test/browser/openpgp/data/keys";
+
+/**
+ * Initialize OpenPGP add testing keys.
+ */
+add_setup(async function () {
+  do_get_profile();
+
+  await OpenPGPTestUtils.initOpenPGP();
+});
+
+add_task(async function testStripSignatures() {
+  await OpenPGPTestUtils.importPublicKey(
+    null,
+    do_get_file(`${keyDir}/heisenberg-signed-by-pinkman.asc`)
+  );
+
+  let heisenbergFpr = "8E3D32E652A254F05BEA9F66CF3EB4AFCAC29340";
+  let foundKeys = await RNP.getKeys(["0x" + heisenbergFpr]);
+
+  Assert.equal(foundKeys.length, 1);
+
+  let sigs = RNP.getKeyObjSignatures(foundKeys[0]);
+
+  // Signatures for one user ID
+  Assert.equal(sigs.length, 1);
+
+  // The key in the file has two signatures: one self signature,
+  // plus one foreign certification signature.
+  Assert.equal(sigs[0].sigList.length, 2);
+
+  let reducedKey = RNP.getMultiplePublicKeys([], ["0x" + heisenbergFpr], []);
+
+  // Delete the key we have previously imported
+  await RNP.deleteKey(heisenbergFpr);
+  foundKeys = await RNP.getKeys(["0x" + heisenbergFpr]);
+  Assert.equal(foundKeys.length, 0);
+
+  // Import the reduced key
+  let errorObj = {};
+  let fingerPrintObj = {};
+  let result = await EnigmailKeyRing.importKeyAsync(
+    null,
+    false,
+    reducedKey,
+    false,
+    null,
+    errorObj,
+    fingerPrintObj,
+    false,
+    [],
+    false
+  );
+  Assert.equal(result, 0);
+
+  foundKeys = await RNP.getKeys(["0x" + heisenbergFpr]);
+  Assert.equal(foundKeys.length, 1);
+
+  sigs = RNP.getKeyObjSignatures(foundKeys[0]);
+
+  // The imported stripped key should have only the self signature.
+  Assert.equal(sigs[0].sigList.length, 1);
+});
+
+add_task(async function testKeyWithUnicodeComment() {
+  let keyFile = do_get_file(`${keyDir}/key-with-utf8-comment.asc`);
+  let keyBlock = await IOUtils.readUTF8(keyFile.path);
+
+  let errorObj = {};
+  let fingerPrintObj = {};
+  let result = await EnigmailKeyRing.importKeyAsync(
+    null,
+    false,
+    keyBlock,
+    false,
+    null,
+    errorObj,
+    fingerPrintObj,
+    false,
+    [],
+    false
+  );
+  Assert.equal(result, 0);
+
+  let fpr = "72514F43D0060FC588E80238852C55E6D2AFD7EF";
+  let foundKeys = await RNP.getKeys(["0x" + fpr]);
+
+  Assert.equal(foundKeys.length, 1);
+});
+
+add_task(async function testBinaryKey() {
+  let keyFile = do_get_file(`${keyDir}/key-binary.gpg`);
+  let keyData = await IOUtils.read(keyFile.path);
+  let keyBlock = MailStringUtils.uint8ArrayToByteString(keyData);
+
+  let errorObj = {};
+  let fingerPrintObj = {};
+  let result = await EnigmailKeyRing.importKeyAsync(
+    null,
+    false,
+    keyBlock,
+    true,
+    null,
+    errorObj,
+    fingerPrintObj,
+    false,
+    [],
+    false
+  );
+  Assert.equal(result, 0);
+
+  let fpr = "683F775BA2E5F0ADEBB29697A2D1B914F722004E";
+  let foundKeys = await RNP.getKeys(["0x" + fpr]);
+
+  Assert.equal(foundKeys.length, 1);
+});