summaryrefslogtreecommitdiffstats
path: root/browser/components/extensions/test/browser/browser_ext_browserAction_popup_port.js
blob: 415d69738dff127b9cc62dfa57f78d9ca87a9f2d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

let scriptPage = url =>
  `<html><head><meta charset="utf-8"><script src="${url}"></script></head><body>${url}</body></html>`;

// Tests that message ports still function correctly after a browserAction popup
// <browser> has been reparented.
add_task(async function test_browserActionPort() {
  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      browser_action: {
        default_popup: "popup.html",
        default_area: "navbar",
        browser_style: true,
      },
    },

    background() {
      new Promise(resolve => {
        browser.runtime.onConnect.addListener(port => {
          resolve(
            Promise.all([
              new Promise(r => port.onMessage.addListener(r)),
              new Promise(r => port.onDisconnect.addListener(r)),
            ])
          );
        });
      }).then(([msg]) => {
        browser.test.assertEq("Hallo.", msg, "Got expected message");
        browser.test.notifyPass("browserAction-popup-port");
      });
    },

    files: {
      "popup.html": scriptPage("popup.js"),
      "popup.js"() {
        let port = browser.runtime.connect();
        window.onload = () => {
          setTimeout(() => {
            port.postMessage("Hallo.");
            window.close();
          }, 0);
        };
      },
    },
  });

  await extension.startup();

  await clickBrowserAction(extension);

  await extension.awaitFinish("browserAction-popup-port");
  await extension.unload();
});