test_distribution_idonly.js

Enable keyboard shortcuts

/* Any copyright is dedicated to the Public Domain.

   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

const IDONLY_INI = `

[Global]

id=idonlytest

[Preferences]

distribution.test.idonly=true

`;

async function setupDistributionDir(iniContent) {

  Services.prefs.setBoolPref("distribution.testing.loadFromProfile", true);

  let distroDir = gProfD.clone();

  distroDir.leafName = "distribution";

  await IOUtils.makeDirectory(distroDir.path, { ignoreExisting: true });

  let iniFile = distroDir.clone();

  iniFile.append("distribution.ini");

  await IOUtils.writeUTF8(iniFile.path, iniContent);

registerCleanupFunction(async function () {

  let folderPath = PathUtils.join(PathUtils.profileDir, "distribution");

  await IOUtils.remove(folderPath, { ignoreAbsent: true, recursive: true });

  Services.prefs.clearUserPref("distribution.testing.loadFromProfile");

});

add_setup(async function () {

  do_get_profile();

  await setupDistributionDir(IDONLY_INI);

});

add_task(async function test_idonly_distribution() {

  let { DistributionManagement } = ChromeUtils.importESModule(

    "resource:///modules/distribution.sys.mjs"

);

  DistributionManagement.applyCustomizations();

  let defaultBranch = Services.prefs.getDefaultBranch(null);

  Assert.equal(

    defaultBranch.getCharPref("distribution.id"),

    "idonlytest",

    "distribution.id should be set from an id-only distribution.ini"

);

  Assert.ok(

    defaultBranch.getBoolPref("distribution.test.idonly"),

    "customization should continue past the id-only global section"

);

});