# HG changeset patch
# User Andrew Swan <aswan@mozilla.com>
# Date 1510212203 28800
#      Wed Nov 08 23:23:23 2017 -0800
# Node ID 2c99f80acc977d7f1f65815adad059632660cd10
# Parent  f98c002b37df85133d165a67b92489f7370f8221
Bug 1396399 Test for content scripts and activeTab permission

MozReview-Commit-ID: DyVtSuA9voI

diff --git a/testing/specialpowers/content/SpecialPowersObserver.jsm b/testing/specialpowers/content/SpecialPowersObserver.jsm
--- a/testing/specialpowers/content/SpecialPowersObserver.jsm
+++ b/testing/specialpowers/content/SpecialPowersObserver.jsm
@@ -83,16 +83,17 @@ SpecialPowersObserver.prototype._loadFra
     this._messageManager.addMessageListener("SPImportInMainProcess", this);
     this._messageManager.addMessageListener("SPChromeScriptMessage", this);
     this._messageManager.addMessageListener("SPQuotaManager", this);
     this._messageManager.addMessageListener("SPSetTestPluginEnabledState", this);
     this._messageManager.addMessageListener("SPLoadExtension", this);
     this._messageManager.addMessageListener("SPStartupExtension", this);
     this._messageManager.addMessageListener("SPUnloadExtension", this);
     this._messageManager.addMessageListener("SPExtensionMessage", this);
+    this._messageManager.addMessageListener("SPExtensionGrantActiveTab", this);
     this._messageManager.addMessageListener("SPCleanUpSTSData", this);
     this._messageManager.addMessageListener("SPRequestDumpCoverageCounters", this);
     this._messageManager.addMessageListener("SPRequestResetCoverageCounters", this);
 
     this._messageManager.loadFrameScript(CHILD_LOGGER_SCRIPT, true);
     this._messageManager.loadFrameScript(CHILD_SCRIPT_API, true);
     this._messageManager.loadFrameScript(CHILD_SCRIPT, true);
     this._isFrameScriptLoaded = true;
@@ -154,16 +155,17 @@ SpecialPowersObserver.prototype.uninit =
     this._messageManager.removeMessageListener("SPImportInMainProcess", this);
     this._messageManager.removeMessageListener("SPChromeScriptMessage", this);
     this._messageManager.removeMessageListener("SPQuotaManager", this);
     this._messageManager.removeMessageListener("SPSetTestPluginEnabledState", this);
     this._messageManager.removeMessageListener("SPLoadExtension", this);
     this._messageManager.removeMessageListener("SPStartupExtension", this);
     this._messageManager.removeMessageListener("SPUnloadExtension", this);
     this._messageManager.removeMessageListener("SPExtensionMessage", this);
+    this._messageManager.removeMessageListener("SPExtensionGrantActiveTab", this);
     this._messageManager.removeMessageListener("SPCleanUpSTSData", this);
     this._messageManager.removeMessageListener("SPRequestDumpCoverageCounters", this);
     this._messageManager.removeMessageListener("SPRequestResetCoverageCounters", this);
 
     this._messageManager.removeDelayedFrameScript(CHILD_LOGGER_SCRIPT);
     this._messageManager.removeDelayedFrameScript(CHILD_SCRIPT_API);
     this._messageManager.removeDelayedFrameScript(CHILD_SCRIPT);
     this._isFrameScriptLoaded = false;
diff --git a/testing/specialpowers/content/SpecialPowersObserverAPI.js b/testing/specialpowers/content/SpecialPowersObserverAPI.js
--- a/testing/specialpowers/content/SpecialPowersObserverAPI.js
+++ b/testing/specialpowers/content/SpecialPowersObserverAPI.js
@@ -639,16 +639,23 @@ SpecialPowersObserverAPI.prototype = {
 
       case "SPExtensionMessage": {
         let id = aMessage.data.id;
         let extension = this._extensions.get(id);
         extension.testMessage(...aMessage.data.args);
         return undefined;
       }
 
+      case "SPExtensionGrantActiveTab": {
+        let {id, tabId} = aMessage.data;
+        let {tabManager} = this._extensions.get(id);
+        tabManager.addActiveTabPermission(tabManager.get(tabId).nativeTab);
+        return undefined;
+      }
+
       case "SPUnloadExtension": {
         let id = aMessage.data.id;
         let extension = this._extensions.get(id);
         this._extensions.delete(id);
         let done = () => this._sendReply(aMessage, "SPExtensionMessage", {id, type: "extensionUnloaded", args: []});
         extension.shutdown().then(done, done);
         return undefined;
       }
diff --git a/testing/specialpowers/content/specialpowers.js b/testing/specialpowers/content/specialpowers.js
--- a/testing/specialpowers/content/specialpowers.js
+++ b/testing/specialpowers/content/specialpowers.js
@@ -47,17 +47,18 @@ function SpecialPowers(window) {
                             "SpecialPowers.Quit",
                             "SpecialPowers.CreateFiles",
                             "SpecialPowers.RemoveFiles",
                             "SPPingService",
                             "SPLoadExtension",
                             "SPProcessCrashManagerWait",
                             "SPStartupExtension",
                             "SPUnloadExtension",
-                            "SPExtensionMessage"];
+                            "SPExtensionMessage",
+                            "SPExtensionGrantActiveTab"];
   addMessageListener("SPPingService", this._messageListener);
   addMessageListener("SpecialPowers.FilesCreated", this._messageListener);
   addMessageListener("SpecialPowers.FilesError", this._messageListener);
   let self = this;
   Services.obs.addObserver(function onInnerWindowDestroyed(subject, topic, data) {
     var id = subject.QueryInterface(Components.interfaces.nsISupportsPRUint64).data;
     if (self._windowID === id) {
       Services.obs.removeObserver(onInnerWindowDestroyed, "inner-window-destroyed");
diff --git a/testing/specialpowers/content/specialpowersAPI.js b/testing/specialpowers/content/specialpowersAPI.js
--- a/testing/specialpowers/content/specialpowersAPI.js
+++ b/testing/specialpowers/content/specialpowersAPI.js
@@ -2027,16 +2027,20 @@ SpecialPowersAPI.prototype = {
         state = "unloading";
         sp._sendAsyncMessage("SPUnloadExtension", {id});
         return unloadPromise;
       },
 
       sendMessage(...args) {
         sp._sendAsyncMessage("SPExtensionMessage", {id, args});
       },
+
+      grantActiveTab(tabId) {
+        sp._sendAsyncMessage("SPExtensionGrantActiveTab", {id, tabId});
+      },
     };
 
     this._sendAsyncMessage("SPLoadExtension", {ext, id});
 
     let listener = (msg) => {
       if (msg.data.id == id) {
         if (msg.data.type == "extensionStarted") {
           state = "running";
diff --git a/toolkit/components/extensions/test/mochitest/file_contentscript_activeTab.html b/toolkit/components/extensions/test/mochitest/file_contentscript_activeTab.html
new file mode 100644
--- /dev/null
+++ b/toolkit/components/extensions/test/mochitest/file_contentscript_activeTab.html
@@ -0,0 +1,10 @@
+<!DOCTYPE HTML>
+
+<html>
+<head>
+  <meta charset="utf-8">
+</head>
+<body>
+  <iframe id="frame" src="http://www.example.com/"/>
+</body>
+</html>
diff --git a/toolkit/components/extensions/test/mochitest/mochitest-common.ini b/toolkit/components/extensions/test/mochitest/mochitest-common.ini
--- a/toolkit/components/extensions/test/mochitest/mochitest-common.ini
+++ b/toolkit/components/extensions/test/mochitest/mochitest-common.ini
@@ -1,16 +1,17 @@
 [DEFAULT]
 support-files =
   chrome_cleanup_script.js
   head.js
   head_unlimitedStorage.js
   head_webrequest.js
   hsts.sjs
   file_mixed.html
+  file_contentscript_activeTab.html
   file_csp.html
   file_csp.html^headers^
   file_page_xhr.html
   file_to_drawWindow.html
   file_WebRequest_page3.html
   file_WebRequest_permission_original.html
   file_WebRequest_permission_redirected.html
   file_WebRequest_permission_original.js
@@ -62,16 +63,17 @@ support-files =
 [test_ext_clipboard.html]
 [test_ext_clipboard_image.html]
 skip-if = headless # disabled test case with_permission_allow_copy, see inline comment. Headless: Bug 1405872
 [test_ext_inIncognitoContext_window.html]
 skip-if = os == 'android' # Android does not support multiple windows.
 [test_ext_geturl.html]
 [test_ext_background_canvas.html]
 [test_ext_content_security_policy.html]
+[test_ext_contentscript_activeTab.html]
 [test_ext_contentscript_api_injection.html]
 [test_ext_contentscript_async_loading.html]
 skip-if = os == 'android' && debug # The generated script takes too long to load on Android debug
 [test_ext_contentscript_cache.html]
 skip-if = (os == 'linux' && debug) || (toolkit == 'android' && debug) # bug 1348241
 [test_ext_contentscript_canvas.html]
 [test_ext_contentscript_context.html]
 [test_ext_contentscript_create_iframe.html]
diff --git a/toolkit/components/extensions/test/mochitest/test_ext_contentscript_activeTab.html b/toolkit/components/extensions/test/mochitest/test_ext_contentscript_activeTab.html
new file mode 100644
--- /dev/null
+++ b/toolkit/components/extensions/test/mochitest/test_ext_contentscript_activeTab.html
@@ -0,0 +1,112 @@
+<!DOCTYPE HTML>
+<html>
+<head>
+  <title>Test for content script</title>
+  <script type="text/javascript" src="/tests/SimpleTest/SimpleTest.js"></script>
+  <script type="text/javascript" src="/tests/SimpleTest/SpawnTask.js"></script>
+  <script type="text/javascript" src="/tests/SimpleTest/ExtensionTestUtils.js"></script>
+  <script type="text/javascript" src="head.js"></script>
+  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css"/>
+</head>
+<body>
+
+<script type="text/javascript">
+"use strict";
+
+const {ObjectUtils} = SpecialPowers.Cu.import("resource://gre/modules/ObjectUtils.jsm", {});
+
+add_task(async function test_contentscript() {
+  async function background() {
+    const URL = "http://mochi.test:8888/tests/toolkit/components/extensions/test/mochitest/file_contentscript_activeTab.html";
+    let tab = await browser.tabs.create({url: URL});
+
+    function addFrame() {
+      let frame = document.createElement("iframe");
+      frame.src = "http://test1.example.com/";
+      frame.addEventListener("load", () => {
+        browser.test.sendMessage("frame-added");
+      }, {once: true});
+      document.body.appendChild(frame);
+    }
+
+    function navigateFrame() {
+      let frame = document.getElementById("frame");
+      frame.addEventListener("load", () => {
+        browser.test.sendMessage("frame-navigated");
+      }, {once: true});
+      frame.src = "http://test2.example.com/";
+    }
+
+    browser.test.onMessage.addListener(async msg => {
+      if (msg == "execute") {
+        let result = await browser.tabs.executeScript(tab.id, {
+          allFrames: true,
+          code: "window.location.hostname;",
+        }).catch(err => ({error: err.message}));
+        browser.test.sendMessage("result", result);
+      } else if (msg == "add-frame") {
+        await browser.tabs.executeScript(tab.id, {
+          code: `(${addFrame})();`,
+        });
+      } else if (msg == "navigate-frame") {
+        await browser.tabs.executeScript(tab.id, {
+          code: `(${navigateFrame})();`,
+        });
+      } else if (msg == "close") {
+        await browser.tabs.remove(tab.id);
+        browser.test.sendMessage("closed");
+      }
+    });
+
+    browser.test.sendMessage("ready", tab.id);
+  }
+
+  let extensionData = {
+    manifest: {
+      permissions: ["activeTab"],
+    },
+    background,
+  };
+
+  let extension = ExtensionTestUtils.loadExtension(extensionData);
+  await extension.startup();
+
+  let tabId = await extension.awaitMessage("ready");
+
+  extension.sendMessage("execute");
+  let result = await extension.awaitMessage("result");
+  ok(/^Missing host permission/.test(result.error), "executeScript() without activeTab permission failed");
+
+  extension.grantActiveTab(tabId);
+
+  const BASE_HOST = "mochi.test";
+  const FRAME_HOST = "www.example.com";
+
+  extension.sendMessage("execute");
+  result = await extension.awaitMessage("result");
+  ok(ObjectUtils.deepEqual(result.sort(), [BASE_HOST, FRAME_HOST]),
+     "Script was injected into base page and original frame");
+
+  extension.sendMessage("add-frame");
+  await extension.awaitMessage("frame-added");
+  extension.sendMessage("execute");
+  result = await extension.awaitMessage("result");
+  ok(ObjectUtils.deepEqual(result.sort(), [BASE_HOST, FRAME_HOST]),
+     "Script was injected into base page and original frame");
+
+  extension.sendMessage("navigate-frame");
+  await extension.awaitMessage("frame-navigated");
+  extension.sendMessage("execute");
+  result = await extension.awaitMessage("result");
+  ok(ObjectUtils.deepEqual(result, [BASE_HOST]),
+     "Script was only injected into base page");
+
+  extension.sendMessage("close");
+  await extension.awaitMessage("closed");
+
+  await extension.unload();
+});
+</script>
+
+</body>
+</html>