summaryrefslogtreecommitdiffstats
path: root/toolkit/components/webextensions/test/mochitest/test_ext_webrequest_background_events.html
blob: 78efeab3500262c098c9619437c42c3ae56ee132 (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
<!DOCTYPE HTML>
<html>
<head>
  <title>Test for simple WebExtension</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";

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

  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), `recieved ${name}`);
        eventNames.delete(name);
        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/*"]}
        );
      }
    },
  });

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

add_task(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), `recieved ${name}`);
        eventNames.delete(name);

        if (eventNames.size === 0) {
          browser.test.assertEq(0, eventNames.size, "messages recieved");
          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.pass("Fetch succeeded.");
      }, () => {
        browser.test.fail("fetch recieved");
        browser.test.sendMessage("done");
      });
    },
  });

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

</body>
</html>