summaryrefslogtreecommitdiffstats
path: root/browser/components/extensions/test/browser/browser_ext_pageAction_activeTab.js
blob: 4ce6b247bf799456e6797560e28cc2dd25853f03 (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
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

add_task(async function test_middle_click_with_activeTab() {
  let tab = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "https://example.com/"
  );

  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      page_action: {},
      permissions: ["activeTab"],
    },

    async background() {
      browser.pageAction.onClicked.addListener(async (tab, info) => {
        browser.test.assertEq(1, info.button, "Expected button value");
        browser.test.assertEq(
          "https://example.com/",
          tab.url,
          "tab.url has the expected url"
        );
        await browser.tabs.insertCSS(tab.id, {
          code: "body { border: 20px solid red; }",
        });
        browser.test.sendMessage("onClick");
      });

      let [tab] = await browser.tabs.query({
        active: true,
        currentWindow: true,
      });
      await browser.pageAction.show(tab.id);
      browser.test.sendMessage("ready");
    },
  });

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

  let ext = WebExtensionPolicy.getByID(extension.id).extension;
  is(
    ext.tabManager.hasActiveTabPermission(tab),
    false,
    "Active tab was not granted permission"
  );

  await clickPageAction(extension, window, { button: 1 });
  await extension.awaitMessage("onClick");

  is(
    ext.tabManager.hasActiveTabPermission(tab),
    true,
    "Active tab was granted permission"
  );
  await extension.unload();

  BrowserTestUtils.removeTab(tab);
});

add_task(async function test_middle_click_without_activeTab() {
  let tab = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "https://example.com/"
  );

  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      page_action: {},
    },

    async background() {
      browser.pageAction.onClicked.addListener(async (tab, info) => {
        browser.test.assertEq(1, info.button, "Expected button value");
        browser.test.assertEq(tab.url, undefined, "tab.url is undefined");
        await browser.test.assertRejects(
          browser.tabs.insertCSS(tab.id, {
            code: "body { border: 20px solid red; }",
          }),
          "Missing host permission for the tab",
          "expected failure of tabs.insertCSS without permission"
        );
        browser.test.sendMessage("onClick");
      });

      let [tab] = await browser.tabs.query({
        active: true,
        currentWindow: true,
      });
      await browser.pageAction.show(tab.id);
      browser.test.sendMessage("ready");
    },
  });

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

  await clickPageAction(extension, window, { button: 1 });
  await extension.awaitMessage("onClick");

  await extension.unload();

  BrowserTestUtils.removeTab(tab);
});