1 files changed, 110 insertions, 0 deletions
diff --git a/comm/mailnews/search/test/unit/test_bug366491.js b/comm/mailnews/search/test/unit/test_bug366491.js
new file mode 100644
index 0000000000..774e8932a3
--- /dev/null
+++ b/comm/mailnews/search/test/unit/test_bug366491.js
@@ -0,0 +1,110 @@
+/* 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 return of junk percent from bayesian filter
+
+// main setup
+
+// only needed during debug
+// do_import_script("mailnews/extensions/bayesian-spam-filter/test/resources/trainingfile.js");
+
+var { MailServices } = ChromeUtils.import(
+  "resource:///modules/MailServices.jsm"
+);
+
+// local constants
+var kUnclassified = MailServices.junk.UNCLASSIFIED;
+var kJunk = MailServices.junk.JUNK;
+var kGood = MailServices.junk.GOOD;
+
+/*
+ * This test is not intended to check the spam calculations,
+ * but only that the junk percent is transmitted (particularly
+ * for intermediate values). The test
+ * junkPercent values below were calculated by the plugin,
+ * not indepedently verified.
+ */
+
+var tests = [
+  {
+    fileName: "ham2.eml",
+    junkPercent: 8,
+  },
+  {
+    fileName: "spam2.eml",
+    junkPercent: 81,
+  },
+];
+
+var emails = [
+  {
+    fileName: "ham1.eml",
+    classification: kGood,
+  },
+  {
+    fileName: "spam1.eml",
+    classification: kJunk,
+  },
+];
+
+// main test
+function run_test() {
+  localAccountUtils.loadLocalMailAccount();
+  do_test_pending();
+  doTestingListener.onMessageClassified(null, null, null);
+  return true;
+}
+
+var haveClassification = false;
+var doTestingListener = {
+  onMessageClassified(aMsgURI, aClassification, aJunkPercent) {
+    // Do we have more training emails? If so, train
+    var email = emails.shift();
+    if (email) {
+      MailServices.junk.setMessageClassification(
+        getSpec(email.fileName),
+        kUnclassified,
+        email.classification,
+        null,
+        doTestingListener
+      );
+      return;
+    }
+
+    if (!aMsgURI) {
+      // Ignore end of batch.
+      return;
+    }
+
+    // Have we completed a classification? If so, test
+    if (haveClassification) {
+      let test = tests.shift();
+      Assert.equal(getSpec(test.fileName), aMsgURI);
+      Assert.equal(test.junkPercent, aJunkPercent);
+    }
+
+    // Do we have more classifications to do? Then classify the first one.
+    if (tests.length) {
+      haveClassification = true;
+      MailServices.junk.classifyMessage(
+        getSpec(tests[0].fileName),
+        null,
+        doTestingListener
+      );
+    } else {
+      do_test_finished();
+    }
+  },
+};
+
+// helper functions
+
+function getSpec(aFileName) {
+  var file = do_get_file(
+    "../../../extensions/bayesian-spam-filter/test/unit/resources/" + aFileName
+  );
+  var uri = Services.io.newFileURI(file).QueryInterface(Ci.nsIURL);
+  uri = uri.mutate().setQuery("type=application/x-message-display").finalize();
+  return uri.spec;
+}