summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/mochitest/test_ext_dnr_other_extensions.html
blob: d3074b3dec6e215ba2662cca78ea93b2000d48ca (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
<!DOCTYPE HTML>
<html>
<head>
  <meta charset="utf-8">
  <title>DNR and tabs.create from other extension</title>
  <script type="text/javascript" src="/tests/SimpleTest/SimpleTest.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>
"use strict";


// While most DNR tests are xpcshell tests, this one is a mochitest because the
// tabs.create API does not work in a xpcshell test.

add_setup(async () => {
  await SpecialPowers.pushPrefEnv({
    set: [
      ["extensions.manifestV3.enabled", true],
      ["extensions.dnr.enabled", true],
    ],
  });
});


add_task(async function tabs_create_can_be_redirected_by_other_dnr_extension() {
  let dnrExtension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["declarativeNetRequestWithHostAccess"],
      // redirect action requires host permissions:
      host_permissions: ["*://example.com/*"],
    },
    async background() {
      await browser.declarativeNetRequest.updateSessionRules({
        addRules: [
          {
            id: 1,
            condition: {
              resourceTypes: ["main_frame"],
              urlFilter: "?dnr_redir_me_pls",
            },
            action: {
              type: "redirect",
              redirect: {
                transform: {
                  query: "?dnr_redir_target"
                },
              },
            },
          },
        ],
      });
      browser.test.sendMessage("dnr_registered");
    },
  });
  await dnrExtension.startup();
  await dnrExtension.awaitMessage("dnr_registered");

  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["webNavigation"],
    },
    async background() {
      async function createTabAndGetFinalUrl(url) {
        let navigationDonePromise = new Promise(resolve => {
          browser.webNavigation.onDOMContentLoaded.addListener(
            function listener(details) {
              browser.webNavigation.onDOMContentLoaded.removeListener(listener);
              resolve(details);
            },
            // All input URLs and redirection targets match this URL filter:
            { url: [{ queryPrefix: "dnr_redir_" }] }
          );
        });
        const tab = await browser.tabs.create({ url });
        browser.test.log(`Waiting for navigation done, starting from ${url}`);
        const result = await navigationDonePromise;
        browser.test.assertEq(
          tab.id,
          result.tabId,
          `Observed load completion for navigation tab with initial URL ${url}`
        );
        await browser.tabs.remove(tab.id);
        return result.url;
      }

      browser.test.assertEq(
        "https://example.com/?dnr_redir_target",
        await createTabAndGetFinalUrl("https://example.com/?dnr_redir_me_pls"),
        "DNR rule from other extension should have redirected the navigation"
      );

      browser.test.assertEq(
        "https://example.org/?dnr_redir_me_pls",
        await createTabAndGetFinalUrl("https://example.org/?dnr_redir_me_pls"),
        "DNR redirect ignored for URLs without host permission"
      );
      browser.test.sendMessage("done");
    }
  });
  await extension.startup();
  await extension.awaitMessage("done");

  await dnrExtension.unload();
  await extension.unload();
});

</script>
</body>
</html>