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 buttonBox = document.getElementById("pocket-button-box");
  is(buttonBox.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"]);
  buttonBox = document.getElementById("pocket-button-box");
  is(buttonBox.hidden, true, "Button should have been hidden");

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

  await BrowserTestUtils.closeWindow(newWin);

  await promisePocketReset();
});