1 files changed, 68 insertions, 0 deletions
diff --git a/browser/components/customizableui/test/browser_947987_removable_default.js b/browser/components/customizableui/test/browser_947987_removable_default.js
new file mode 100644
index 000000000..98325ec2a
--- /dev/null
+++ b/browser/components/customizableui/test/browser_947987_removable_default.js
@@ -0,0 +1,68 @@
+/* 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";
+
+var kWidgetId = "test-removable-widget-default";
+const kNavBar = CustomizableUI.AREA_NAVBAR;
+var widgetCounter = 0;
+
+registerCleanupFunction(removeCustomToolbars);
+
+// Sanity checks
+add_task(function() {
+  let brokenSpec = {id: kWidgetId + (widgetCounter++), removable: false};
+  SimpleTest.doesThrow(() => CustomizableUI.createWidget(brokenSpec),
+                       "Creating non-removable widget without defaultArea should throw.");
+
+  // Widget without removable set should be removable:
+  let wrapper = CustomizableUI.createWidget({id: kWidgetId + (widgetCounter++)});
+  ok(CustomizableUI.isWidgetRemovable(wrapper.id), "Should be removable by default.");
+  CustomizableUI.destroyWidget(wrapper.id);
+});
+
+// Test non-removable widget with defaultArea
+add_task(function*() {
+  // Non-removable widget with defaultArea should work:
+  let spec = {id: kWidgetId + (widgetCounter++), removable: false,
+              defaultArea: kNavBar};
+  let widgetWrapper;
+  try {
+    widgetWrapper = CustomizableUI.createWidget(spec);
+  } catch (ex) {
+    ok(false, "Creating a non-removable widget with a default area should not throw.");
+    return;
+  }
+
+  let placement = CustomizableUI.getPlacementOfWidget(spec.id);
+  ok(placement, "Widget should be placed.");
+  is(placement.area, kNavBar, "Widget should be in navbar");
+  let singleWrapper = widgetWrapper.forWindow(window);
+  ok(singleWrapper, "Widget should exist in window.");
+  ok(singleWrapper.node, "Widget node should exist in window.");
+  let expectedParent = CustomizableUI.getCustomizeTargetForArea(kNavBar, window);
+  is(singleWrapper.node.parentNode, expectedParent, "Widget should be in navbar.");
+
+  let otherWin = yield openAndLoadWindow(true);
+  placement = CustomizableUI.getPlacementOfWidget(spec.id);
+  ok(placement, "Widget should be placed.");
+  is(placement && placement.area, kNavBar, "Widget should be in navbar");
+
+  singleWrapper = widgetWrapper.forWindow(otherWin);
+  ok(singleWrapper, "Widget should exist in other window.");
+  if (singleWrapper) {
+    ok(singleWrapper.node, "Widget node should exist in other window.");
+    if (singleWrapper.node) {
+      let expectedParent = CustomizableUI.getCustomizeTargetForArea(kNavBar, otherWin);
+      is(singleWrapper.node.parentNode, expectedParent,
+         "Widget should be in navbar in other window.");
+    }
+  }
+  CustomizableUI.destroyWidget(spec.id);
+  yield promiseWindowClosed(otherWin);
+});
+
+add_task(function* asyncCleanup() {
+  yield resetCustomization();
+});