summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/mochitest/test_ext_webrequest_background_events.html
blob: 5ac3dccaa7c7cb08bfa27faf55351448d3ae2dd4 (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
114
115
116
117
118
119
120
<!DOCTYPE HTML>
<html>
<head>
  <title>Test for simple WebExtension</title>
  <script src="/tests/SimpleTest/SimpleTest.js"></script>
  <script 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";

add_task(async function test_webRequest_serviceworker_events() {
  await SpecialPowers.pushPrefEnv({
    set: [["dom.serviceWorkers.testing.enabled", true]],
  });

  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: [
        "webRequest",
        "<all_urls>",
      ],
    },
    background() {
      let eventNames = new Set([
        "onBeforeRequest",
        "onBeforeSendHeaders",
        "onSendHeaders",
        "onHeadersReceived",
        "onResponseStarted",
        "onCompleted",
        "onErrorOccurred",
      ]);

      function listener(name) {
        browser.test.assertTrue(eventNames.has(name), `received ${name}`);
        eventNames.delete(name);
        if (name == "onCompleted") {
          eventNames.delete("onErrorOccurred");
        } else if (name == "onErrorOccurred") {
          eventNames.delete("onCompleted");
        }
        if (eventNames.size == 0) {
          browser.test.sendMessage("done");
        }
      }

      for (let name of eventNames) {
        browser.webRequest[name].addListener(
          listener.bind(null, name),
          {urls: ["https://example.com/*"]}
        );
      }
    },
  });

  await extension.startup();
  let registration = await navigator.serviceWorker.register("webrequest_worker.js", {scope: "."});
  await waitForState(registration.installing, "activated");
  await extension.awaitMessage("done");
  await registration.unregister();
  await extension.unload();
});

add_task(async function test_webRequest_background_events() {
  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: [
        "webRequest",
        "<all_urls>",
      ],
    },
    background() {
      let eventNames = new Set([
        "onBeforeRequest",
        "onBeforeSendHeaders",
        "onSendHeaders",
        "onHeadersReceived",
        "onResponseStarted",
        "onCompleted",
      ]);

      function listener(name, details) {
        browser.test.assertTrue(eventNames.has(name), `received ${name}`);
        eventNames.delete(name);

        if (eventNames.size === 0) {
          browser.test.assertEq("xmlhttprequest", details.type, "correct type for fetch [see bug 1366710]");
          browser.test.assertEq(0, eventNames.size, "messages received");
          browser.test.sendMessage("done");
        }
      }

      for (let name of eventNames) {
        browser.webRequest[name].addListener(
          listener.bind(null, name),
          {urls: ["https://example.com/*"]}
        );
      }

      fetch("https://example.com/example.txt").then(() => {
        browser.test.succeed("Fetch succeeded.");
      }, () => {
        browser.test.fail("fetch received");
        browser.test.sendMessage("done");
      });
    },
  });

  await extension.startup();
  await extension.awaitMessage("done");
  await extension.unload();
});
</script>

</body>
</html>