summaryrefslogtreecommitdiffstats
path: root/browser/components/customizableui/test/browser_create_button_widget.js
blob: 66a7855ddf89d8f37d84dc2f080c12cf664fd75c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/* 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 kButton = "test_dynamically_created_button";
var initialLocation = gBrowser.currentURI.spec;

add_task(async function() {
  info("Check dynamically created button functionality");

  // Let's create a simple button that will open about:addons.
  let widgetSpec = {
    id: kButton,
    type: "button",
    onClick() {
      gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser, "about:addons");
    },
  };
  CustomizableUI.createWidget(widgetSpec);
  CustomizableUI.addWidgetToArea(kButton, CustomizableUI.AREA_NAVBAR);
  ok(
    !CustomizableUI.isWebExtensionWidget(kButton),
    "This button should not be considered an extension widget."
  );

  // check the button's functionality in navigation bar
  let button = document.getElementById(kButton);
  let navBar = document.getElementById("nav-bar");
  ok(button, "Dynamically created button exists");
  ok(navBar.contains(button), "Dynamically created button is in the navbar");
  await checkButtonFunctionality(button);

  resetTabs();

  // move the add-on button in the Panel Menu
  CustomizableUI.addWidgetToArea(
    kButton,
    CustomizableUI.AREA_FIXED_OVERFLOW_PANEL
  );
  ok(
    !navBar.contains(button),
    "Dynamically created button was removed from the browser bar"
  );

  await waitForOverflowButtonShown();

  // check the button's functionality in the Overflow Panel.
  await document.getElementById("nav-bar").overflowable.show();
  var panelMenu = document.getElementById("widget-overflow-mainView");
  let buttonInPanel = panelMenu.getElementsByAttribute("id", kButton);
  ok(
    panelMenu.contains(button),
    "Dynamically created button was added to the Panel Menu"
  );
  await checkButtonFunctionality(buttonInPanel[0]);
});

add_task(async function asyncCleanup() {
  resetTabs();

  // reset the UI to the default state
  await resetCustomization();
  ok(CustomizableUI.inDefaultState, "The UI is in default state again.");

  // destroy the widget
  CustomizableUI.destroyWidget(kButton);
});

function resetTabs() {
  // close all opened tabs
  while (gBrowser.tabs.length > 1) {
    gBrowser.removeTab(gBrowser.selectedTab);
  }

  // restore the initial tab
  BrowserTestUtils.addTab(gBrowser, initialLocation);
  gBrowser.removeTab(gBrowser.selectedTab);
}

async function checkButtonFunctionality(aButton) {
  aButton.click();
  await TestUtils.waitForCondition(
    () => gBrowser.currentURI && gBrowser.currentURI.spec == "about:addons"
  );
}