1 files changed, 97 insertions, 0 deletions
diff --git a/comm/mail/test/browser/shared-modules/JunkHelpers.jsm b/comm/mail/test/browser/shared-modules/JunkHelpers.jsm
new file mode 100644
index 0000000000..6d53271ed5
--- /dev/null
+++ b/comm/mail/test/browser/shared-modules/JunkHelpers.jsm
@@ -0,0 +1,97 @@
+/* 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/. */
+
+"use strict";
+
+const EXPORTED_SYMBOLS = [
+  "mark_selected_messages_as_junk",
+  "delete_mail_marked_as_junk",
+];
+
+var EventUtils = ChromeUtils.import(
+  "resource://testing-common/mozmill/EventUtils.jsm"
+);
+var { TestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/TestUtils.sys.mjs"
+);
+
+var {
+  mc,
+  get_about_3pane,
+  plan_to_wait_for_folder_events,
+  wait_for_message_display_completion,
+  wait_for_folder_events,
+} = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+
+/**
+ * Mark the selected messages as junk. This is done by pressing the J key.
+ *
+ * @param aController The controller in whose context to do this, defaults to
+ *     |mc| if omitted.
+ */
+function mark_selected_messages_as_junk(aController) {
+  if (aController === undefined) {
+    aController = mc;
+  }
+  let win = get_about_3pane(aController.window);
+  if (aController == mc) {
+    win.document.getElementById("threadTree").focus();
+  }
+  EventUtils.synthesizeKey("j", {}, win);
+}
+
+/**
+ * Delete all mail marked as junk in the selected folder. This is done by
+ * activating the menu option from the Tools menu.
+ *
+ * @param aNumDeletesExpected The number of deletes expected.
+ * @param aController The controller in whose context to do this, defaults to
+ *     |mc| if omitted.
+ */
+async function delete_mail_marked_as_junk(aNumDeletesExpected, aController) {
+  if (aController === undefined) {
+    aController = mc;
+  }
+  let win = get_about_3pane(aController.window);
+
+  // Monkey patch and wrap around the deleteJunkInFolder function, mainly for
+  // the case where deletes aren't expected.
+  let realDeleteJunkInFolder = win.deleteJunkInFolder;
+  let numMessagesDeleted = null;
+  let fakeDeleteJunkInFolder = function () {
+    numMessagesDeleted = realDeleteJunkInFolder();
+    return numMessagesDeleted;
+  };
+  try {
+    win.deleteJunkInFolder = fakeDeleteJunkInFolder;
+
+    // If something is loading, make sure it finishes loading...
+    wait_for_message_display_completion(aController);
+    if (aNumDeletesExpected != 0) {
+      plan_to_wait_for_folder_events(
+        "DeleteOrMoveMsgCompleted",
+        "DeleteOrMoveMsgFailed"
+      );
+    }
+
+    win.goDoCommand("cmd_deleteJunk");
+
+    if (aNumDeletesExpected != 0) {
+      wait_for_folder_events();
+    }
+
+    // If timeout waiting for numMessagesDeleted to turn non-null,
+    // this either means that deleteJunkInFolder didn't get called or that it
+    // didn't return a value."
+
+    await TestUtils.waitForCondition(
+      () => numMessagesDeleted === aNumDeletesExpected,
+      `Should have got ${aNumDeletesExpected} deletes, not ${numMessagesDeleted}`
+    );
+  } finally {
+    win.deleteJunkInFolder = realDeleteJunkInFolder;
+  }
+}