Bug 1287007 - Fix test_ext_api_permissions.js
"extension.views" will only be available in content, so
use the "proxy-context-load" event to detect new views.
MozReview-Commit-ID: K86Be5IDk42
--- a/toolkit/components/extensions/test/xpcshell/test_ext_api_permissions.js
+++ b/toolkit/components/extensions/test/xpcshell/test_ext_api_permissions.js
@@ -1,41 +1,53 @@
/* -*- 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() {},
manifest: {
permissions: [],
},
});
+ let contextPromise = getNextContext();
yield extension.startup();
- let context = Array.from(extension.extension.views)[0];
+ let context = yield contextPromise;
ok(!("storage" in context._unwrappedAPIs),
"The storage API should not be initialized");
yield extension.unload();
});
add_task(function* test_storage_api_with_permissions() {
let extension = ExtensionTestUtils.loadExtension({
background() {},
manifest: {
permissions: ["storage"],
},
});
+ let contextPromise = getNextContext();
yield extension.startup();
- let context = Array.from(extension.extension.views)[0];
+ let context = yield contextPromise;
equal(typeof context._unwrappedAPIs.storage, "object",
"The storage API should be initialized");
yield extension.unload();
});