summaryrefslogtreecommitdiffstats
path: root/browser/components/extensions/test/browser/browser_ext_port_disconnect_on_crash.js
blob: 82ece1da3f68a2f8e0db0a69eec5f05d3b3afb00 (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

add_task(async function connect_from_tab_to_bg_and_crash_tab() {
  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      content_scripts: [
        {
          js: ["contentscript.js"],
          matches: ["http://example.com/?crashme"],
        },
      ],
    },

    background() {
      browser.runtime.onConnect.addListener(port => {
        browser.test.assertEq("tab_to_bg", port.name, "expected port");
        browser.test.assertEq(port.sender.frameId, 0, "correct frameId");

        port.onDisconnect.addListener(() => {
          browser.test.assertEq(
            null,
            port.error,
            "port should be disconnected without errors"
          );
          browser.test.sendMessage("port_disconnected");
        });
        browser.test.sendMessage("bg_runtime_onConnect");
      });
    },

    files: {
      "contentscript.js": function () {
        let port = browser.runtime.connect({ name: "tab_to_bg" });
        port.onDisconnect.addListener(() => {
          browser.test.fail("Unexpected onDisconnect event in content script");
        });
      },
    },
  });

  await extension.startup();

  let tab = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "http://example.com/?crashme"
  );
  await extension.awaitMessage("bg_runtime_onConnect");
  // Force the message manager to disconnect without giving the content a
  // chance to send an "Extension:Port:Disconnect" message.
  await BrowserTestUtils.crashFrame(tab.linkedBrowser);
  await extension.awaitMessage("port_disconnected");
  BrowserTestUtils.removeTab(tab);
  await extension.unload();
});

add_task(async function connect_from_bg_to_tab_and_crash_tab() {
  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      content_scripts: [
        {
          js: ["contentscript.js"],
          matches: ["http://example.com/?crashme"],
        },
      ],
    },

    background() {
      browser.runtime.onMessage.addListener((msg, sender) => {
        browser.test.assertEq("contentscript_ready", msg, "expected message");
        let port = browser.tabs.connect(sender.tab.id, { name: "bg_to_tab" });
        port.onDisconnect.addListener(() => {
          browser.test.assertEq(
            null,
            port.error,
            "port should be disconnected without errors"
          );
          browser.test.sendMessage("port_disconnected");
        });
      });
    },

    files: {
      "contentscript.js": function () {
        browser.runtime.onConnect.addListener(port => {
          browser.test.assertEq("bg_to_tab", port.name, "expected port");
          port.onDisconnect.addListener(() => {
            browser.test.fail(
              "Unexpected onDisconnect event in content script"
            );
          });
          browser.test.sendMessage("tab_runtime_onConnect");
        });
        browser.runtime.sendMessage("contentscript_ready");
      },
    },
  });

  await extension.startup();

  let tab = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "http://example.com/?crashme"
  );
  await extension.awaitMessage("tab_runtime_onConnect");
  // Force the message manager to disconnect without giving the content a
  // chance to send an "Extension:Port:Disconnect" message.
  await BrowserTestUtils.crashFrame(tab.linkedBrowser);
  await extension.awaitMessage("port_disconnected");
  BrowserTestUtils.removeTab(tab);
  await extension.unload();
});