1 files changed, 139 insertions, 0 deletions
diff --git a/browser/components/customizableui/test/browser_panelUINotifications_bannerVisibility.js b/browser/components/customizableui/test/browser_panelUINotifications_bannerVisibility.js
new file mode 100644
index 0000000000..ff4c10ee11
--- /dev/null
+++ b/browser/components/customizableui/test/browser_panelUINotifications_bannerVisibility.js
@@ -0,0 +1,139 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { AppMenuNotifications } = ChromeUtils.importESModule(
+  "resource://gre/modules/AppMenuNotifications.sys.mjs"
+);
+
+/**
+ * The update banner should become visible when the badge-only notification is
+ * shown before opening the menu.
+ */
+add_task(async function testBannerVisibilityBeforeOpen() {
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  AppMenuNotifications.showBadgeOnlyNotification("update-restart");
+
+  let menuButton = newWin.document.getElementById("PanelUI-menu-button");
+  let shown = BrowserTestUtils.waitForEvent(
+    newWin.PanelUI.mainView,
+    "ViewShown"
+  );
+  menuButton.click();
+  await shown;
+
+  let banner = newWin.document.getElementById("appMenu-proton-update-banner");
+
+  let labelPromise = BrowserTestUtils.waitForMutationCondition(
+    banner,
+    { attributes: true, attributeFilter: ["label"] },
+    () => banner.hasAttribute("label")
+  );
+
+  ok(!banner.hidden, "Update banner should be shown");
+
+  await labelPromise;
+
+  ok(banner.getAttribute("label") != "", "Update banner should contain text");
+
+  AppMenuNotifications.removeNotification(/.*/);
+
+  await BrowserTestUtils.closeWindow(newWin);
+});
+
+/**
+ * The update banner should become visible when the badge-only notification is
+ * shown during the menu is opened.
+ */
+add_task(async function testBannerVisibilityDuringOpen() {
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  let menuButton = newWin.document.getElementById("PanelUI-menu-button");
+  let shown = BrowserTestUtils.waitForEvent(
+    newWin.PanelUI.mainView,
+    "ViewShown"
+  );
+  menuButton.click();
+  await shown;
+
+  let banner = newWin.document.getElementById("appMenu-proton-update-banner");
+  ok(
+    !banner.hasAttribute("label"),
+    "Update banner shouldn't contain text before notification"
+  );
+
+  let labelPromise = BrowserTestUtils.waitForMutationCondition(
+    banner,
+    { attributes: true, attributeFilter: ["label"] },
+    () => banner.hasAttribute("label")
+  );
+
+  AppMenuNotifications.showNotification("update-restart");
+
+  ok(!banner.hidden, "Update banner should be shown");
+
+  await labelPromise;
+
+  ok(banner.getAttribute("label") != "", "Update banner should contain text");
+
+  AppMenuNotifications.removeNotification(/.*/);
+
+  await BrowserTestUtils.closeWindow(newWin);
+});
+
+/**
+ * The update banner should become visible when the badge-only notification is
+ * shown after opening/closing the menu, so that the DOM tree is there but
+ * the menu is closed.
+ */
+add_task(async function testBannerVisibilityAfterClose() {
+  let newWin = await BrowserTestUtils.openNewBrowserWindow();
+
+  let menuButton = newWin.document.getElementById("PanelUI-menu-button");
+  let shown = BrowserTestUtils.waitForEvent(
+    newWin.PanelUI.mainView,
+    "ViewShown"
+  );
+  menuButton.click();
+  await shown;
+
+  ok(newWin.PanelUI.mainView.hasAttribute("visible"));
+
+  let banner = newWin.document.getElementById("appMenu-proton-update-banner");
+
+  ok(banner.hidden, "Update banner should be hidden before notification");
+  ok(
+    !banner.hasAttribute("label"),
+    "Update banner shouldn't contain text before notification"
+  );
+
+  let labelPromise = BrowserTestUtils.waitForMutationCondition(
+    banner,
+    { attributes: true, attributeFilter: ["label"] },
+    () => banner.hasAttribute("label")
+  );
+
+  let hidden = BrowserTestUtils.waitForCondition(() => {
+    return !newWin.PanelUI.mainView.hasAttribute("visible");
+  });
+  menuButton.click();
+  await hidden;
+
+  AppMenuNotifications.showBadgeOnlyNotification("update-restart");
+
+  shown = BrowserTestUtils.waitForEvent(newWin.PanelUI.mainView, "ViewShown");
+  menuButton.click();
+  await shown;
+
+  ok(!banner.hidden, "Update banner should be shown");
+
+  await labelPromise;
+
+  ok(banner.getAttribute("label") != "", "Update banner should contain text");
+
+  AppMenuNotifications.removeNotification(/.*/);
+
+  await BrowserTestUtils.closeWindow(newWin);
+});