gecko-dev/browser/components/pocket/test/browser_pocket_ui_check.js

"use strict";

add_task(async function test_setup() {
  let clearValue = Services.prefs.prefHasUserValue("extensions.pocket.enabled");
  let enabledOnStartup = Services.prefs.getBoolPref(
    "extensions.pocket.enabled"
  );
  registerCleanupFunction(() => {
    if (clearValue) {
      Services.prefs.clearUserPref("extensions.pocket.enabled");
    } else {
      Services.prefs.setBoolPref("extensions.pocket.enabled", enabledOnStartup);
    }
  });
});

add_task(async function() {
  await promisePocketEnabled();

  checkElements(true, ["pocket-button", "appMenu-library-pocket-button"]);
  let button = document.getElementById("pocket-button");
  is(button.hidden, false, "Button should not have been hidden");

  // check context menu exists
  info("checking content context menu");
  let tab = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "https://example.com/browser/browser/components/pocket/test/test.html"
  );

  let contextMenu = document.getElementById("contentAreaContextMenu");
  let popupShown = BrowserTestUtils.waitForEvent(contextMenu, "popupshown");
  let popupHidden = BrowserTestUtils.waitForEvent(contextMenu, "popuphidden");
  await BrowserTestUtils.synthesizeMouseAtCenter(
    "body",
    {
      type: "contextmenu",
      button: 2,
    },
    tab.linkedBrowser
  );
  await popupShown;

  checkElements(true, ["context-pocket"]);

  contextMenu.hidePopup();
  await popupHidden;
  popupShown = BrowserTestUtils.waitForEvent(contextMenu, "popupshown");
  popupHidden = BrowserTestUtils.waitForEvent(contextMenu, "popuphidden");
  await BrowserTestUtils.synthesizeMouseAtCenter(
    "a",
    {
      type: "contextmenu",
      button: 2,
    },
    tab.linkedBrowser
  );
  await popupShown;

  checkElements(true, ["context-savelinktopocket"]);

  contextMenu.hidePopup();
  await popupHidden;
  BrowserTestUtils.removeTab(tab);

  await promisePocketDisabled();

  checkElements(false, [
    "appMenu-library-pocket-button",
    "context-pocket",
    "context-savelinktopocket",
  ]);
  button = document.getElementById("pocket-button");
  is(button.hidden, true, "Button should have been hidden");

  let newWin = await BrowserTestUtils.openNewBrowserWindow();
  checkElements(
    false,
    [
      "appMenu-library-pocket-button",
      "context-pocket",
      "context-savelinktopocket",
    ],
    newWin
  );
  button = newWin.document.getElementById("pocket-button");
  is(button.hidden, true, "Button should have been hidden");

  await BrowserTestUtils.closeWindow(newWin);

  await promisePocketReset();
});