summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/mochitest/test_ext_streamfilter_multiple.html
blob: d1bfbd824b08640de2bc1dca819afc4fb13f42c2 (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
<!DOCTYPE html>
<html>
<head>
  <title>Test for multiple extensions trying to filterResponseData on the same request</title>
  <script src="/tests/SimpleTest/SimpleTest.js"></script>
  <script src="/tests/SimpleTest/EventUtils.js"></script>
  <script src="/tests/SimpleTest/ExtensionTestUtils.js"></script>
  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css"/>
</head>
<body>

<script type="text/javascript">
"use strict";

const TEST_URL =
  "http://example.org/tests/toolkit/components/extensions/test/mochitest/file_streamfilter.txt";

add_task(async () => {
  const firstExtension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["webRequest", "webRequestBlocking", "<all_urls>"],
    },

    background() {
      browser.webRequest.onBeforeRequest.addListener(
        ({ requestId }) => {
          const filter = browser.webRequest.filterResponseData(requestId);
          filter.ondata = event => {
            filter.write(new TextEncoder().encode("Start "));
            filter.write(event.data);
            filter.disconnect();
          };
        },
        {
          urls: [
            "http://example.org/*/file_streamfilter.txt",
          ],
        },
        ["blocking"]
      );
    },
  });

  const secondExtension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["webRequest", "webRequestBlocking", "<all_urls>"],
    },

    background() {
      browser.webRequest.onBeforeRequest.addListener(
        ({ requestId }) => {
          const filter = browser.webRequest.filterResponseData(requestId);
          filter.ondata = event => {
            filter.write(event.data);
          };
          filter.onstop = event => {
            filter.write(new TextEncoder().encode(" End"));
            filter.close();
          };
        },
        {
          urls: [
            "http://example.org/tests/toolkit/components/extensions/test/mochitest/file_streamfilter.txt",
          ],
        },
        ["blocking"]
      );
    },
  });

  await firstExtension.startup();
  await secondExtension.startup();

  let iframe = document.createElement("iframe");
  iframe.src = TEST_URL;
  document.body.appendChild(iframe);
  await new Promise(resolve => iframe.addEventListener("load", () => resolve(), {once: true}));

  let content = await SpecialPowers.spawn(iframe, [], async () => {
    return this.content.document.body.textContent;
  });
  SimpleTest.is(content, "Start Middle\n End", "Correctly intercepted page content");

  await firstExtension.unload();
  await secondExtension.unload();
});

</script>

</body>
</html>