mozilla-central/toolkit/components/extensions/test/xpcshell/test_ext_management_policy.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
AddonTestUtils.init(this);
AddonTestUtils.overrideCertDB();
AddonTestUtils.createAppInfo(
"xpcshell@tests.mozilla.org",
"XPCShell",
"42",
"42"
);
const server = AddonTestUtils.createHttpServer();
const { EnterprisePolicyTesting } = ChromeUtils.importESModule(
);
// This initializes the policy engine for xpcshell tests
let policies = Cc["@mozilla.org/enterprisepolicies;1"].getService(
Ci.nsIObserver
);
policies.observe(null, "policies-startup", null);
const TEST_ADDON_ID = "test_management_addon@tests.mozilla.com";
let addon;
add_setup(async () => {
await ExtensionTestUtils.startAddonManager();
addon = ExtensionTestUtils.loadExtension({
manifest: {
name: "Simple addon test",
version: "1.0",
description: "test addon",
browser_specific_settings: {
gecko: {
id: TEST_ADDON_ID,
},
},
},
useAddonManager: "temporary",
});
await addon.startup();
registerCleanupFunction(async () => {
await addon.unload();
});
});
add_task(async function test_enterprise_management_works() {
const TEST_ID = "test_management_policy@tests.mozilla.com";
async function background(TEST_ADDON_ID) {
let addon = await browser.management.get(TEST_ADDON_ID);
browser.test.assertTrue(addon.enabled, "addon is enabled");
await browser.management.setEnabled(addon.id, false);
addon = await browser.management.get(TEST_ADDON_ID);
browser.test.assertFalse(addon.enabled, "addon is disabled");
await browser.management.setEnabled(addon.id, true);
addon = await browser.management.get(TEST_ADDON_ID);
browser.test.assertTrue(addon.enabled, "addon is enabled");
browser.test.sendMessage("done");
}
let originalXpi = AddonTestUtils.createTempWebExtensionFile({
manifest: {
browser_specific_settings: {
gecko: {
id: TEST_ID,
},
},
name: TEST_ID,
permissions: ["management"],
},
background: `(${background})("${TEST_ADDON_ID}")`,
});
let serverHost = `http://localhost:${server.identity.primaryPort}`;
let xpiFilename = `/foo.xpi`;
server.registerFile(xpiFilename, originalXpi);
let extension = ExtensionTestUtils.expectExtension(TEST_ID);
await Promise.all([
AddonTestUtils.promiseInstallEvent("onInstallEnded"),
EnterprisePolicyTesting.setupPolicyEngineWithJson({
policies: {
ExtensionSettings: {
[TEST_ID]: {
installation_mode: "force_installed",
install_url: serverHost + xpiFilename,
},
},
},
}),
]);
await extension.awaitStartup();
await extension.awaitMessage("done");
await extension.unload();
});
add_task(async function test_no_enterprise_management_fails() {
const TEST_ID = "test_management_no_policy@tests.mozilla.com";
async function background(TEST_ADDON_ID) {
let addon = await browser.management.get(TEST_ADDON_ID);
browser.test.assertTrue(addon.enabled, "addon is enabled");
await browser.test.assertRejects(
browser.management.setEnabled(addon.id, false),
/setEnabled can only be used/,
"setEnabled should fail"
);
addon = await browser.management.get(TEST_ADDON_ID);
browser.test.assertTrue(addon.enabled, "addon is still enabled");
browser.test.sendMessage("done");
}
let extension = ExtensionTestUtils.loadExtension({
manifest: {
browser_specific_settings: {
gecko: {
id: TEST_ID,
},
},
permissions: ["management"],
},
background: `(${background})("${TEST_ADDON_ID}")`,
useAddonManager: "temporary",
});
await extension.startup();
await extension.awaitMessage("done");
await extension.unload();
});