1 files changed, 182 insertions, 0 deletions
diff --git a/browser/components/extensions/test/browser/browser_ext_webNavigation_onCreatedNavigationTarget_contextmenu.js b/browser/components/extensions/test/browser/browser_ext_webNavigation_onCreatedNavigationTarget_contextmenu.js
new file mode 100644
index 0000000000..8fd94af4f1
--- /dev/null
+++ b/browser/components/extensions/test/browser/browser_ext_webNavigation_onCreatedNavigationTarget_contextmenu.js
@@ -0,0 +1,182 @@
+/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set sts=2 sw=2 et tw=80: */
+"use strict";
+
+loadTestSubscript("head_webNavigation.js");
+
+SpecialPowers.pushPrefEnv({
+  set: [["security.allow_eval_with_system_principal", true]],
+});
+
+async function clickContextMenuItem({
+  pageElementSelector,
+  contextMenuItemLabel,
+  frameIndex,
+}) {
+  let contentAreaContextMenu;
+  if (frameIndex == null) {
+    contentAreaContextMenu = await openContextMenu(pageElementSelector);
+  } else {
+    contentAreaContextMenu = await openContextMenuInFrame(
+      pageElementSelector,
+      frameIndex
+    );
+  }
+  const item = contentAreaContextMenu.getElementsByAttribute(
+    "label",
+    contextMenuItemLabel
+  );
+  is(item.length, 1, `found contextMenu item for "${contextMenuItemLabel}"`);
+  const closed = promiseContextMenuClosed(contentAreaContextMenu);
+  contentAreaContextMenu.activateItem(item[0]);
+  await closed;
+}
+
+async function background() {
+  const tabs = await browser.tabs.query({ active: true, currentWindow: true });
+  const sourceTabId = tabs[0].id;
+
+  const sourceTabFrames = await browser.webNavigation.getAllFrames({
+    tabId: sourceTabId,
+  });
+
+  browser.webNavigation.onCreatedNavigationTarget.addListener(msg => {
+    browser.test.sendMessage("webNavOnCreated", msg);
+  });
+
+  browser.webNavigation.onCompleted.addListener(async msg => {
+    // NOTE: checking the url is currently necessary because of Bug 1252129
+    // ( Filter out webNavigation events related to new window initialization phase).
+    if (msg.tabId !== sourceTabId && msg.url !== "about:blank") {
+      await browser.tabs.remove(msg.tabId);
+      browser.test.sendMessage("webNavOnCompleted", msg);
+    }
+  });
+
+  browser.tabs.onCreated.addListener(tab => {
+    browser.test.sendMessage("tabsOnCreated", tab.id);
+  });
+
+  browser.test.sendMessage("expectedSourceTab", {
+    sourceTabId,
+    sourceTabFrames,
+  });
+}
+
+add_task(async function test_on_created_navigation_target_from_context_menu() {
+  const tab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    SOURCE_PAGE
+  );
+
+  const extension = ExtensionTestUtils.loadExtension({
+    background,
+    manifest: {
+      permissions: ["webNavigation"],
+    },
+  });
+
+  await extension.startup();
+
+  const expectedSourceTab = await extension.awaitMessage("expectedSourceTab");
+
+  info("Open link in a new tab from the context menu");
+
+  await runCreatedNavigationTargetTest({
+    extension,
+    async openNavTarget() {
+      await clickContextMenuItem({
+        pageElementSelector: "#test-create-new-tab-from-context-menu",
+        contextMenuItemLabel: "Open Link in New Tab",
+      });
+    },
+    expectedWebNavProps: {
+      sourceTabId: expectedSourceTab.sourceTabId,
+      sourceFrameId: 0,
+      url: `${OPENED_PAGE}#new-tab-from-context-menu`,
+    },
+  });
+
+  info("Open link in a new window from the context menu");
+
+  await runCreatedNavigationTargetTest({
+    extension,
+    async openNavTarget() {
+      await clickContextMenuItem({
+        pageElementSelector: "#test-create-new-window-from-context-menu",
+        contextMenuItemLabel: "Open Link in New Window",
+      });
+    },
+    expectedWebNavProps: {
+      sourceTabId: expectedSourceTab.sourceTabId,
+      sourceFrameId: 0,
+      url: `${OPENED_PAGE}#new-window-from-context-menu`,
+    },
+  });
+
+  BrowserTestUtils.removeTab(tab);
+
+  await extension.unload();
+});
+
+add_task(
+  async function test_on_created_navigation_target_from_context_menu_subframe() {
+    const tab = await BrowserTestUtils.openNewForegroundTab(
+      gBrowser,
+      SOURCE_PAGE
+    );
+
+    const extension = ExtensionTestUtils.loadExtension({
+      background,
+      manifest: {
+        permissions: ["webNavigation"],
+      },
+    });
+
+    await extension.startup();
+
+    const expectedSourceTab = await extension.awaitMessage("expectedSourceTab");
+
+    info("Open a subframe link in a new tab from the context menu");
+
+    await runCreatedNavigationTargetTest({
+      extension,
+      async openNavTarget() {
+        await clickContextMenuItem({
+          pageElementSelector:
+            "#test-create-new-tab-from-context-menu-subframe",
+          contextMenuItemLabel: "Open Link in New Tab",
+          frameIndex: 0,
+        });
+      },
+      expectedWebNavProps: {
+        sourceTabId: expectedSourceTab.sourceTabId,
+        sourceFrameId: expectedSourceTab.sourceTabFrames[1].frameId,
+        url: `${OPENED_PAGE}#new-tab-from-context-menu-subframe`,
+      },
+    });
+
+    info("Open a subframe link in a new window from the context menu");
+
+    await runCreatedNavigationTargetTest({
+      extension,
+      async openNavTarget() {
+        await clickContextMenuItem({
+          pageElementSelector:
+            "#test-create-new-window-from-context-menu-subframe",
+          contextMenuItemLabel: "Open Link in New Window",
+          frameIndex: 0,
+        });
+      },
+      expectedWebNavProps: {
+        sourceTabId: expectedSourceTab.sourceTabId,
+        sourceFrameId: expectedSourceTab.sourceTabFrames[1].frameId,
+        url: `${OPENED_PAGE}#new-window-from-context-menu-subframe`,
+      },
+    });
+
+    BrowserTestUtils.removeTab(tab);
+
+    await extension.unload();
+  }
+);