gecko/dom/tests/mochitest/general/browserFrameHelpers.js

"use strict";

// Helpers for managing the browser frame preferences.

const browserFrameHelpers = {
  'getEnabledPref': function() {
    try {
      return SpecialPowers.getBoolPref('dom.mozBrowserFramesEnabled');
    }
    catch(e) {
      return undefined;
    }
  },

  'getWhitelistPref': function() {
    try {
      return SpecialPowers.getCharPref('dom.mozBrowserFramesWhitelist');
    }
    catch(e) {
      return undefined;
    }
  },

  'setEnabledPref': function(enabled) {
    if (enabled !== undefined) {
      SpecialPowers.setBoolPref('dom.mozBrowserFramesEnabled', enabled);
    }
    else {
      SpecialPowers.clearUserPref('dom.mozBrowserFramesEnabled');
    }
  },

  'setWhitelistPref': function(whitelist) {
    if (whitelist !== undefined) {
      SpecialPowers.setCharPref('dom.mozBrowserFramesWhitelist', whitelist);
    }
    else {
      SpecialPowers.clearUserPref('dom.mozBrowserFramesWhitelist');
    }
  },

  'addToWhitelist': function() {
    var whitelist = browserFrameHelpers.getWhitelistPref();
    whitelist += ',  http://' + window.location.host + ',  ';
    browserFrameHelpers.setWhitelistPref(whitelist);
  },

  'restoreOriginalPrefs': function() {
    browserFrameHelpers.setEnabledPref(browserFrameHelpers.origEnabledPref);
    browserFrameHelpers.setWhitelistPref(browserFrameHelpers.origWhitelistPref);
  },

  'origEnabledPref': null,
  'origWhitelistPref': null
};

browserFrameHelpers.origEnabledPref = browserFrameHelpers.getEnabledPref();
browserFrameHelpers.origWhitelistPref = browserFrameHelpers.getWhitelistPref();

addEventListener('unload', function() {
  browserFrameHelpers.restoreOriginalPrefs();
});