summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/mochitest/test_ext_optional_permissions.html
blob: 659a55f5c96ec74353d3f419cbd30197c66d8728 (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
<!DOCTYPE HTML>
<html>
<head>
  <meta charset="utf-8">
  <title>optional permissions and preloaded processes</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" href="/tests/SimpleTest/test.css"/>
</head>
<body>

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

add_task(async function setup() {
  await SpecialPowers.pushPrefEnv({
    set: [["extensions.webextOptionalPermissionPrompts", false]],
  });
});

// This test case verifies that newly granted optional permissions are
// propagated to all processes, especially preloaded processes.
add_task(async function test_optional_permissions_should_be_propagated() {
  let anOptionalPermission = "*://example.org/*";

  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: [
        "scripting",
        "*://example.com/*",
      ],
      optional_permissions: [anOptionalPermission],
    },
    async background() {
      browser.test.onMessage.addListener(async (msg, value) => {
        browser.test.assertEq("grant-permission", msg, "expected message");

        let granted = await new Promise(resolve => {
          browser.test.withHandlingUserInput(() => {
            resolve(browser.permissions.request(value));
          });
        });
        browser.test.assertTrue(granted, "permission request succeeded");
        browser.test.sendMessage("permission-granted");
      });

      await browser.scripting.registerContentScripts([
        {
          id: "script",
          js: ["script.js"],
          matches: ["*://example.com/*", "*://example.org/*"],
          persistAcrossSessions: false,
        },
      ]);

      browser.test.sendMessage("background-ready");
    },
    files: {
      "script.js": () => {
        browser.test.sendMessage("script-ran", window.location.host);
      },
    },
  });

  await extension.startup();
  await extension.awaitMessage("background-ready");

  let tab = await AppTestDelegate.openNewForegroundTab(
    window,
    "http://example.com/",
    true
  );
  let host = await extension.awaitMessage("script-ran");
  is(host, "example.com", "expected host: example.com");
  await AppTestDelegate.removeTab(window, tab);

  extension.sendMessage("grant-permission", {
    origins: ["*://example.org/*"],
  });
  await extension.awaitMessage("permission-granted");

  tab = await AppTestDelegate.openNewForegroundTab(
    window,
    "https://example.org/",
    true
  );
  host = await extension.awaitMessage("script-ran");
  is(host, "example.org", "expected host: example.org");
  await AppTestDelegate.removeTab(window, tab);

  await extension.unload();
});

</script>

</body>
</html>