summaryrefslogtreecommitdiffstats
path: root/toolkit/mozapps/extensions/test/xpcshell/test_provider_unsafe_access_startup.js
blob: 8e066973f23f6de7b0540488ee4873083e5630c7 (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
createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "1.9.2");

var startupOrder = [];

function mockAddonProvider(name) {
  let mockProvider = {
    hasStarted: false,
    unsafeAccess: false,

    startupCallback: null,

    startup() {
      this.hasStarted = true;
      startupOrder.push(this.name);
      if (this.startupCallback) {
        this.startupCallback();
      }
    },
    getAddonByID(id, callback) {
      if (!this.hasStarted) {
        this.unsafeAccess = true;
      }
      callback(null);
    },

    get name() {
      return name;
    },
  };

  return mockProvider;
}

add_task(async function unsafeProviderStartup() {
  let secondProvider = null;

  await new Promise(resolve => {
    let firstProvider = mockAddonProvider("Mock1");
    firstProvider.startupCallback = function () {
      resolve(AddonManager.getAddonByID("does-not-exist"));
    };
    AddonManagerPrivate.registerProvider(firstProvider);

    secondProvider = mockAddonProvider("Mock2");
    AddonManagerPrivate.registerProvider(secondProvider);

    promiseStartupManager();
  });

  equal(
    startupOrder.join(","),
    ["Mock1", "Mock2"].join(","),
    "Mock providers should have hasStarted in expected order"
  );
  ok(
    !secondProvider.unsafeAccess,
    "Second registered mock provider should not have been accessed unsafely"
  );
});