1 files changed, 238 insertions, 0 deletions
diff --git a/browser/components/extensions/test/browser/browser_ext_pageAction_popup.js b/browser/components/extensions/test/browser/browser_ext_pageAction_popup.js
new file mode 100644
index 000000000..83defdd68
--- /dev/null
+++ b/browser/components/extensions/test/browser/browser_ext_pageAction_popup.js
@@ -0,0 +1,238 @@
+/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set sts=2 sw=2 et tw=80: */
+"use strict";
+
+add_task(function* testPageActionPopup() {
+  let tab = yield BrowserTestUtils.openNewForegroundTab(gBrowser, "http://example.com/");
+
+  let scriptPage = url => `<html><head><meta charset="utf-8"><script src="${url}"></script></head></html>`;
+
+  let extension = ExtensionTestUtils.loadExtension({
+    manifest: {
+      "background": {
+        "page": "data/background.html",
+      },
+      "page_action": {
+        "default_popup": "popup-a.html",
+      },
+    },
+
+    files: {
+      "popup-a.html": scriptPage("popup-a.js"),
+      "popup-a.js": function() {
+        window.onload = () => {
+          let background = window.getComputedStyle(document.body).backgroundColor;
+          browser.test.assertEq("transparent", background);
+          browser.runtime.sendMessage("from-popup-a");
+        };
+        browser.runtime.onMessage.addListener(msg => {
+          if (msg == "close-popup") {
+            window.close();
+          }
+        });
+      },
+
+      "data/popup-b.html": scriptPage("popup-b.js"),
+      "data/popup-b.js": function() {
+        browser.runtime.sendMessage("from-popup-b");
+      },
+
+      "data/background.html": scriptPage("background.js"),
+
+      "data/background.js": async function() {
+        let tabId;
+
+        let sendClick;
+        let tests = [
+          () => {
+            sendClick({expectEvent: false, expectPopup: "a"});
+          },
+          () => {
+            sendClick({expectEvent: false, expectPopup: "a"});
+          },
+          () => {
+            browser.pageAction.setPopup({tabId, popup: "popup-b.html"});
+            sendClick({expectEvent: false, expectPopup: "b"});
+          },
+          () => {
+            sendClick({expectEvent: false, expectPopup: "b"});
+          },
+          () => {
+            browser.pageAction.setPopup({tabId, popup: ""});
+            sendClick({expectEvent: true, expectPopup: null});
+          },
+          () => {
+            sendClick({expectEvent: true, expectPopup: null});
+          },
+          () => {
+            browser.pageAction.setPopup({tabId, popup: "/popup-a.html"});
+            sendClick({expectEvent: false, expectPopup: "a", runNextTest: true});
+          },
+          () => {
+            browser.test.sendMessage("next-test", {expectClosed: true});
+          },
+          () => {
+            sendClick({expectEvent: false, expectPopup: "a", runNextTest: true});
+          },
+          () => {
+            browser.test.sendMessage("next-test", {closeOnTabSwitch: true});
+          },
+        ];
+
+        let expect = {};
+        sendClick = ({expectEvent, expectPopup, runNextTest}) => {
+          expect = {event: expectEvent, popup: expectPopup, runNextTest};
+          browser.test.sendMessage("send-click");
+        };
+
+        browser.runtime.onMessage.addListener(msg => {
+          if (msg == "close-popup") {
+            return;
+          } else if (expect.popup) {
+            browser.test.assertEq(msg, `from-popup-${expect.popup}`,
+                                  "expected popup opened");
+          } else {
+            browser.test.fail(`unexpected popup: ${msg}`);
+          }
+
+          expect.popup = null;
+          if (expect.runNextTest) {
+            expect.runNextTest = false;
+            tests.shift()();
+          } else {
+            browser.test.sendMessage("next-test");
+          }
+        });
+
+        browser.pageAction.onClicked.addListener(() => {
+          if (expect.event) {
+            browser.test.succeed("expected click event received");
+          } else {
+            browser.test.fail("unexpected click event");
+          }
+
+          expect.event = false;
+          browser.test.sendMessage("next-test");
+        });
+
+        browser.test.onMessage.addListener(msg => {
+          if (msg == "close-popup") {
+            browser.runtime.sendMessage("close-popup");
+            return;
+          }
+
+          if (msg != "next-test") {
+            browser.test.fail("Expecting 'next-test' message");
+          }
+
+          if (tests.length) {
+            let test = tests.shift();
+            test();
+          } else {
+            browser.test.notifyPass("pageaction-tests-done");
+          }
+        });
+
+        let [tab] = await browser.tabs.query({active: true, currentWindow: true});
+        tabId = tab.id;
+
+        await browser.pageAction.show(tabId);
+        browser.test.sendMessage("next-test");
+      },
+    },
+  });
+
+  extension.onMessage("send-click", () => {
+    clickPageAction(extension);
+  });
+
+  let pageActionId, panelId;
+  extension.onMessage("next-test", Task.async(function* (expecting = {}) {
+    pageActionId = `${makeWidgetId(extension.id)}-page-action`;
+    panelId = `${makeWidgetId(extension.id)}-panel`;
+    let panel = document.getElementById(panelId);
+    if (expecting.expectClosed) {
+      ok(panel, "Expect panel to exist");
+      yield promisePopupShown(panel);
+
+      extension.sendMessage("close-popup");
+
+      yield promisePopupHidden(panel);
+      ok(true, `Panel is closed`);
+    } else if (expecting.closeOnTabSwitch) {
+      ok(panel, "Expect panel to exist");
+      yield promisePopupShown(panel);
+
+      let oldTab = gBrowser.selectedTab;
+      ok(oldTab != gBrowser.tabs[0], "Should have an inactive tab to switch to");
+
+      let hiddenPromise = promisePopupHidden(panel);
+
+      gBrowser.selectedTab = gBrowser.tabs[0];
+      yield hiddenPromise;
+      info("Panel closed");
+
+      gBrowser.selectedTab = oldTab;
+    } else if (panel) {
+      yield promisePopupShown(panel);
+      panel.hidePopup();
+    }
+
+    if (panel) {
+      panel = document.getElementById(panelId);
+      is(panel, null, "panel successfully removed from document after hiding");
+    }
+
+    extension.sendMessage("next-test");
+  }));
+
+
+  yield extension.startup();
+  yield extension.awaitFinish("pageaction-tests-done");
+
+  yield extension.unload();
+
+  let node = document.getElementById(pageActionId);
+  is(node, null, "pageAction image removed from document");
+
+  let panel = document.getElementById(panelId);
+  is(panel, null, "pageAction panel removed from document");
+
+  yield BrowserTestUtils.removeTab(tab);
+});
+
+
+add_task(function* testPageActionSecurity() {
+  const URL = "chrome://browser/content/browser.xul";
+
+  let apis = ["browser_action", "page_action"];
+
+  for (let api of apis) {
+    info(`TEST ${api} icon url: ${URL}`);
+
+    let messages = [/Access to restricted URI denied/];
+
+    let waitForConsole = new Promise(resolve => {
+      // Not necessary in browser-chrome tests, but monitorConsole gripes
+      // if we don't call it.
+      SimpleTest.waitForExplicitFinish();
+
+      SimpleTest.monitorConsole(resolve, messages);
+    });
+
+    let extension = ExtensionTestUtils.loadExtension({
+      manifest: {
+        [api]: {"default_popup": URL},
+      },
+    });
+
+    yield Assert.rejects(extension.startup(),
+                         null,
+                         "Manifest rejected");
+
+    SimpleTest.endMonitorConsole();
+    yield waitForConsole;
+  }
+});
+
+add_task(forceGC);