michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: "use strict";
michael@0: 
michael@0: const kCustomClass = "acustomclassnoonewilluse";
michael@0: let tempElement = null;
michael@0: 
michael@0: function insertClassNameToMenuChildren(parentMenu) {
michael@0:   let el = parentMenu.querySelector("menuitem:first-of-type");
michael@0:   el.classList.add(kCustomClass);
michael@0:   tempElement = el;
michael@0: }
michael@0: 
michael@0: function checkSubviewButtonClass(menuId, buttonId, subviewId) {
michael@0:   return function() {
michael@0:     info("Checking for items without the subviewbutton class in " + buttonId + " widget");
michael@0:     let menu = document.getElementById(menuId);
michael@0:     insertClassNameToMenuChildren(menu);
michael@0: 
michael@0:     let placement = CustomizableUI.getPlacementOfWidget(buttonId);
michael@0:     let changedPlacement = false;
michael@0:     if (!placement || placement.area != CustomizableUI.AREA_PANEL) {
michael@0:       CustomizableUI.addWidgetToArea(buttonId, CustomizableUI.AREA_PANEL);
michael@0:       changedPlacement = true;
michael@0:     }
michael@0:     yield PanelUI.show();
michael@0: 
michael@0:     let button = document.getElementById(buttonId);
michael@0:     button.click();
michael@0: 
michael@0:     yield waitForCondition(() => !PanelUI.multiView.hasAttribute("transitioning"));
michael@0:     let subview = document.getElementById(subviewId);
michael@0:     ok(subview.firstChild, "Subview should have a kid");
michael@0:     let subviewchildren = subview.querySelectorAll("toolbarbutton");
michael@0:     for (let i = 0; i < subviewchildren.length; i++) {
michael@0:       let item = subviewchildren[i];
michael@0:       let itemReadable = "Item '" + item.label + "' (classes: " + item.className + ")";
michael@0:       ok(item.classList.contains("subviewbutton"), itemReadable + " should have the subviewbutton class.");
michael@0:       if (i == 0) {
michael@0:         ok(item.classList.contains(kCustomClass), itemReadable + " should still have its own class, too.");
michael@0:       }
michael@0:     }
michael@0: 
michael@0:     let panelHiddenPromise = promisePanelHidden(window);
michael@0:     PanelUI.hide();
michael@0:     yield panelHiddenPromise;
michael@0: 
michael@0:     if (changedPlacement) {
michael@0:       CustomizableUI.reset();
michael@0:     }
michael@0:   };
michael@0: }
michael@0: 
michael@0: add_task(checkSubviewButtonClass("menuWebDeveloperPopup", "developer-button", "PanelUI-developerItems"));
michael@0: add_task(checkSubviewButtonClass("viewSidebarMenu", "sidebar-button", "PanelUI-sidebarItems"));
michael@0: 
michael@0: registerCleanupFunction(function() {
michael@0:   tempElement.classList.remove(kCustomClass)
michael@0:   tempElement = null;
michael@0: });