1 files changed, 94 insertions, 0 deletions
diff --git a/browser/components/extensions/test/browser/browser_ext_contentscript_connect.js b/browser/components/extensions/test/browser/browser_ext_contentscript_connect.js
new file mode 100644
index 0000000000..11dbe88240
--- /dev/null
+++ b/browser/components/extensions/test/browser/browser_ext_contentscript_connect.js
@@ -0,0 +1,94 @@
+/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set sts=2 sw=2 et tw=80: */
+"use strict";
+
+add_task(async function () {
+  let tab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    "http://mochi.test:8888/"
+  );
+
+  let extension = ExtensionTestUtils.loadExtension({
+    manifest: {
+      permissions: ["http://mochi.test/"],
+    },
+
+    background: function () {
+      let ports_received = 0;
+      let port_messages_received = 0;
+
+      browser.runtime.onConnect.addListener(port => {
+        browser.test.assertTrue(!!port, "port1 received");
+
+        ports_received++;
+        browser.test.assertEq(1, ports_received, "1 port received");
+
+        port.onMessage.addListener((msg, msgPort) => {
+          browser.test.assertEq(
+            "port message",
+            msg,
+            "listener1 port message received"
+          );
+          browser.test.assertEq(
+            port,
+            msgPort,
+            "onMessage should receive port as second argument"
+          );
+
+          port_messages_received++;
+          browser.test.assertEq(
+            1,
+            port_messages_received,
+            "1 port message received"
+          );
+        });
+      });
+      browser.runtime.onConnect.addListener(port => {
+        browser.test.assertTrue(!!port, "port2 received");
+
+        ports_received++;
+        browser.test.assertEq(2, ports_received, "2 ports received");
+
+        port.onMessage.addListener((msg, msgPort) => {
+          browser.test.assertEq(
+            "port message",
+            msg,
+            "listener2 port message received"
+          );
+          browser.test.assertEq(
+            port,
+            msgPort,
+            "onMessage should receive port as second argument"
+          );
+
+          port_messages_received++;
+          browser.test.assertEq(
+            2,
+            port_messages_received,
+            "2 port messages received"
+          );
+
+          browser.test.notifyPass("contentscript_connect.pass");
+        });
+      });
+
+      browser.tabs.executeScript({ file: "script.js" }).catch(e => {
+        browser.test.fail(`Error: ${e} :: ${e.stack}`);
+        browser.test.notifyFail("contentscript_connect.pass");
+      });
+    },
+
+    files: {
+      "script.js": function () {
+        let port = browser.runtime.connect();
+        port.postMessage("port message");
+      },
+    },
+  });
+
+  await extension.startup();
+  await extension.awaitFinish("contentscript_connect.pass");
+  await extension.unload();
+
+  BrowserTestUtils.removeTab(tab);
+});