summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/xpcshell/test_getAPILevelForWindow.js
blob: 68741a6cca434b1041fd45a19085eff8978541df (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
"use strict";

Cu.import("resource://gre/modules/ExtensionManagement.jsm");
Cu.import("resource://gre/modules/Services.jsm");

function createWindowWithAddonId(addonId) {
  let baseURI = Services.io.newURI("about:blank", null, null);
  let originAttributes = {addonId};
  let principal = Services.scriptSecurityManager
                          .createCodebasePrincipal(baseURI, originAttributes);
  let chromeNav = Services.appShell.createWindowlessBrowser(true);
  let interfaceRequestor = chromeNav.QueryInterface(Ci.nsIInterfaceRequestor);
  let docShell = interfaceRequestor.getInterface(Ci.nsIDocShell);
  docShell.createAboutBlankContentViewer(principal);

  return {chromeNav, window: docShell.contentViewer.DOMDocument.defaultView};
}

add_task(function* test_eventpages() {
  const {getAPILevelForWindow, getAddonIdForWindow} = ExtensionManagement;
  const {NO_PRIVILEGES, FULL_PRIVILEGES} = ExtensionManagement.API_LEVELS;
  const FAKE_ADDON_ID = "fakeAddonId";
  const OTHER_ADDON_ID = "otherFakeAddonId";
  const EMPTY_ADDON_ID = "";

  let fakeAddonId = createWindowWithAddonId(FAKE_ADDON_ID);
  equal(getAddonIdForWindow(fakeAddonId.window), FAKE_ADDON_ID,
        "the window has the expected addonId");

  let apiLevel = getAPILevelForWindow(fakeAddonId.window, FAKE_ADDON_ID);
  equal(apiLevel, FULL_PRIVILEGES,
        "apiLevel for the window with the right addonId should be FULL_PRIVILEGES");

  apiLevel = getAPILevelForWindow(fakeAddonId.window, OTHER_ADDON_ID);
  equal(apiLevel, NO_PRIVILEGES,
        "apiLevel for the window with a different addonId should be NO_PRIVILEGES");

  fakeAddonId.chromeNav.close();

  // NOTE: check that window with an empty addon Id (which are window that are
  // not Extensions pages) always get no WebExtensions APIs.
  let emptyAddonId = createWindowWithAddonId(EMPTY_ADDON_ID);
  equal(getAddonIdForWindow(emptyAddonId.window), EMPTY_ADDON_ID,
        "the window has the expected addonId");

  apiLevel = getAPILevelForWindow(emptyAddonId.window, EMPTY_ADDON_ID);
  equal(apiLevel, NO_PRIVILEGES,
        "apiLevel for empty addonId should be NO_PRIVILEGES");

  apiLevel = getAPILevelForWindow(emptyAddonId.window, OTHER_ADDON_ID);
  equal(apiLevel, NO_PRIVILEGES,
        "apiLevel for an 'empty addonId' window should be always NO_PRIVILEGES");

  emptyAddonId.chromeNav.close();
});