1 files changed, 82 insertions, 0 deletions
diff --git a/comm/mail/test/browser/shared-modules/SubscribeWindowHelpers.jsm b/comm/mail/test/browser/shared-modules/SubscribeWindowHelpers.jsm
new file mode 100644
index 0000000000..2f096479f0
--- /dev/null
+++ b/comm/mail/test/browser/shared-modules/SubscribeWindowHelpers.jsm
@@ -0,0 +1,82 @@
+/* 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 = [
+  "open_subscribe_window_from_context_menu",
+  "enter_text_in_search_box",
+  "check_newsgroup_displayed",
+];
+
+var utils = ChromeUtils.import("resource://testing-common/mozmill/utils.jsm");
+var { get_about_3pane, right_click_on_folder } = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+var { input_value, delete_all_existing } = ChromeUtils.import(
+  "resource://testing-common/mozmill/KeyboardHelpers.jsm"
+);
+var { click_menus_in_sequence, plan_for_modal_dialog, wait_for_modal_dialog } =
+  ChromeUtils.import("resource://testing-common/mozmill/WindowHelpers.jsm");
+
+/**
+ * Open a subscribe dialog from the context menu.
+ *
+ * @param aFolder the folder to open the subscribe dialog for
+ * @param aFunction Callback that will be invoked with a controller
+ *        for the subscribe dialogue as parameter
+ */
+async function open_subscribe_window_from_context_menu(aFolder, aFunction) {
+  let win = get_about_3pane();
+
+  await right_click_on_folder(aFolder);
+  let callback = function (controller) {
+    // When the "stop button" is disabled, the panel is populated.
+    utils.waitFor(
+      () => controller.window.document.getElementById("stopButton").disabled
+    );
+    aFunction(controller);
+  };
+  plan_for_modal_dialog("mailnews:subscribe", callback);
+  await click_menus_in_sequence(
+    win.document.getElementById("folderPaneContext"),
+    [{ id: "folderPaneContext-subscribe" }]
+  );
+  wait_for_modal_dialog("mailnews:subscribe");
+}
+
+/**
+ * Enter a string in the text box for the search value.
+ *
+ * @param swc A controller for a subscribe dialog
+ * @param text The text to enter
+ */
+function enter_text_in_search_box(swc, text) {
+  let textbox = swc.window.document.getElementById("namefield");
+  delete_all_existing(swc, textbox);
+  input_value(swc, text, textbox);
+}
+
+/**
+ * Check whether the given newsgroup is in the searchview.
+ *
+ * @param swc A controller for the subscribe window
+ * @param name Name of the newsgroup
+ * @returns {boolean} Result of the check
+ */
+function check_newsgroup_displayed(swc, name) {
+  let tree = swc.window.document.getElementById("searchTree");
+  if (!tree.columns) {
+    // Maybe not yet available.
+    return false;
+  }
+  let treeview = tree.view;
+  let nameCol = tree.columns.getNamedColumn("nameColumn2");
+  for (let i = 0; i < treeview.rowCount; i++) {
+    if (treeview.getCellText(i, nameCol) == name) {
+      return true;
+    }
+  }
+  return false;
+}