1 files changed, 145 insertions, 0 deletions
diff --git a/comm/mail/test/browser/tabmail/browser_customize.js b/comm/mail/test/browser/tabmail/browser_customize.js
new file mode 100644
index 0000000000..d9a0fc777d
--- /dev/null
+++ b/comm/mail/test/browser/tabmail/browser_customize.js
@@ -0,0 +1,145 @@
+/* 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 customization features of the tabs toolbar.
+ */
+
+"use strict";
+
+const { close_popup, mc, wait_for_popup_to_open } = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+
+const { drag_n_drop_element } = ChromeUtils.import(
+  "resource://testing-common/mozmill/MouseEventHelpers.jsm"
+);
+
+const { click_through_appmenu } = ChromeUtils.import(
+  "resource://testing-common/mozmill/WindowHelpers.jsm"
+);
+
+const { wait_for_element_visible, wait_for_element_invisible } =
+  ChromeUtils.import("resource://testing-common/mozmill/DOMHelpers.jsm");
+
+add_setup(function () {
+  Services.prefs.setBoolPref("mail.tabs.autoHide", false);
+});
+
+registerCleanupFunction(function () {
+  // Let's reset any and all of our changes to the toolbar
+  Services.prefs.clearUserPref("mail.tabs.autoHide");
+});
+
+/**
+ * Test that we can access the unified toolbar by clicking
+ * customize on the toolbar context menu
+ */
+add_task(async function test_open_unified_by_context() {
+  // First, ensure that the context menu is closed.
+  let contextPopup = mc.window.document.getElementById("toolbar-context-menu");
+  Assert.notEqual(
+    contextPopup.state,
+    "open",
+    "Context menu is currently open!"
+  );
+
+  // Right click on the tab bar.
+  EventUtils.synthesizeMouseAtCenter(
+    mc.window.document.getElementById("tabmail-tabs"),
+    { type: "contextmenu" },
+    window
+  );
+
+  // Ensure that the popup opened.
+  await wait_for_popup_to_open(contextPopup);
+  Assert.equal(contextPopup.state, "open", "Context menu was not opened!");
+
+  const customizeButton = document.getElementById("CustomizeMailToolbar");
+  // Click customize.
+  contextPopup.activateItem(customizeButton);
+
+  // Wait for hidden css attribute on unified toolbar
+  // customization to be removed.
+  await wait_for_element_visible(
+    window,
+    "unifiedToolbarCustomizationContainer"
+  );
+
+  // Ensure messengerWindow (HTML element) has customizingUnifiedToolbar class,
+  // which means unified toolbar customization should be open.
+  Assert.ok(
+    document
+      .getElementById("messengerWindow")
+      .classList.contains("customizingUnifiedToolbar"),
+    "customizingUnifiedToolbar class not found on messengerWindow element"
+  );
+
+  // Click cancel.
+  const cancelButton = document.getElementById(
+    "unifiedToolbarCustomizationCancel"
+  );
+  cancelButton.click();
+
+  // Wait for hidden css attribute on Unified Toolbar
+  // customization to be added.
+  await wait_for_element_invisible(
+    window,
+    "unifiedToolbarCustomizationContainer"
+  );
+
+  await close_popup(mc, contextPopup);
+});
+
+/**
+ * Test that we can access the unified toolbar customization by clicking
+ * the toolbar layout menu option
+ */
+add_task(async function test_open_unified_by_menu() {
+  // First, ensure that the menu is closed.
+  let appMenu = mc.window.document.getElementById("appMenu-popup");
+  Assert.notEqual(
+    appMenu.getAttribute("panelopen"),
+    "true",
+    "appMenu-popup is currently open!"
+  );
+
+  // Click through app menu to view unified toolbar.
+  click_through_appmenu(
+    [{ id: "appmenu_View" }, { id: "appmenu_Toolbars" }],
+    { id: "appmenu_toolbarLayout" },
+    window
+  );
+
+  // Wait for hidden css attribute on unified toolbar
+  // customization to be removed.
+  await wait_for_element_visible(
+    window,
+    "unifiedToolbarCustomizationContainer"
+  );
+
+  // Ensure messengerWindow (HTML element) has customizingUnifiedToolbar class,
+  // which means unified toolbar customization should be open.
+  Assert.ok(
+    document
+      .getElementById("messengerWindow")
+      .classList.contains("customizingUnifiedToolbar"),
+    "customizingUnifiedToolbar class not found on messengerWindow element"
+  );
+
+  // Click cancel.
+  const cancelButton = document.getElementById(
+    "unifiedToolbarCustomizationCancel"
+  );
+  cancelButton.click();
+
+  // Wait for hidden css attribute on unified toolbar
+  // customization to be added.
+  await wait_for_element_invisible(
+    window,
+    "unifiedToolbarCustomizationContainer"
+  );
+
+  await close_popup(mc, appMenu);
+});