UXP/toolkit/components/webextensions/test/xpcshell/test_ext_api_permissions.js

/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

let {Management} = Cu.import("resource://gre/modules/Extension.jsm", {});
function getNextContext() {
  return new Promise(resolve => {
    Management.on("proxy-context-load", function listener(type, context) {
      Management.off("proxy-context-load", listener);
      resolve(context);
    });
  });
}

add_task(function* test_storage_api_without_permissions() {
  let extension = ExtensionTestUtils.loadExtension({
    background() {
      // Force API initialization.
      void browser.storage;
    },

    manifest: {
      permissions: [],
    },
  });

  let contextPromise = getNextContext();
  yield extension.startup();

  let context = yield contextPromise;

  // Force API initialization.
  void context.apiObj;

  ok(!("storage" in context.apiObj),
     "The storage API should not be initialized");

  yield extension.unload();
});

add_task(function* test_storage_api_with_permissions() {
  let extension = ExtensionTestUtils.loadExtension({
    background() {
      void browser.storage;
    },

    manifest: {
      permissions: ["storage"],
    },
  });

  let contextPromise = getNextContext();
  yield extension.startup();

  let context = yield contextPromise;

  // Force API initialization.
  void context.apiObj;

  equal(typeof context.apiObj.storage, "object",
        "The storage API should be initialized");

  yield extension.unload();
});