The Tor Browser / file revision

 /* Any copyright is dedicated to the Public Domain.

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

  */

 // Tests resetting of preferences in blocklist entry when an add-on is blocked.

 // See bug 802434.

 const {classes: Cc, interfaces: Ci, utils: Cu, results: Cr} = Components;

 const URI_EXTENSION_BLOCKLIST_DIALOG = "chrome://mozapps/content/extensions/blocklist.xul";

 XPCOMUtils.defineLazyGetter(this, "gPref", function bls_gPref() {

   return Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefService).

          QueryInterface(Ci.nsIPrefBranch);

 });

 Cu.import("resource://testing-common/httpd.js");

 var testserver = new HttpServer();

 testserver.start(-1);

 gPort = testserver.identity.primaryPort;

 // register static files with server and interpolate port numbers in them

 mapFile("/data/test_blocklist_prefs_1.xml", testserver);

 const profileDir = gProfD.clone();

 profileDir.append("extensions");

 // A window watcher to handle the blocklist UI.

 // Don't need the full interface, attempts to call other methods will just

 // throw which is just fine

 var WindowWatcher = {

   openWindow: function(parent, url, name, features, arguments) {

     // Should be called to list the newly blocklisted items

     do_check_eq(url, URI_EXTENSION_BLOCKLIST_DIALOG);

     // Simulate auto-disabling any softblocks

     var list = arguments.wrappedJSObject.list;

     list.forEach(function(aItem) {

       if (!aItem.blocked)

         aItem.disable = true;

     });

     //run the code after the blocklist is closed

     Services.obs.notifyObservers(null, "addon-blocklist-closed", null);

   },

   QueryInterface: function(iid) {

     if (iid.equals(Ci.nsIWindowWatcher)

      || iid.equals(Ci.nsISupports))

       return this;

     throw Cr.NS_ERROR_NO_INTERFACE;

   }

 };

 var WindowWatcherFactory = {

   createInstance: function createInstance(outer, iid) {

     if (outer != null)

       throw Cr.NS_ERROR_NO_AGGREGATION;

     return WindowWatcher.QueryInterface(iid);

   }

 };

 var registrar = Components.manager.QueryInterface(Ci.nsIComponentRegistrar);

 registrar.registerFactory(Components.ID("{1dfeb90a-2193-45d5-9cb8-864928b2af55}"),

                           "Fake Window Watcher",

                           "@mozilla.org/embedcomp/window-watcher;1",

                           WindowWatcherFactory);

 function load_blocklist(aFile, aCallback) {

   Services.obs.addObserver(function() {

     Services.obs.removeObserver(arguments.callee, "blocklist-updated");

     do_execute_soon(aCallback);

   }, "blocklist-updated", false);

   Services.prefs.setCharPref("extensions.blocklist.url", "http://localhost:" +

                              gPort + "/data/" + aFile);

   var blocklist = Cc["@mozilla.org/extensions/blocklist;1"].

                   getService(Ci.nsITimerCallback);

   blocklist.notify(null);

 }

 function end_test() {

   testserver.stop(do_test_finished);

 }

 function run_test() {

   do_test_pending();

   createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "1");

   // Add 2 extensions

   writeInstallRDFForExtension({

     id: "block1@tests.mozilla.org",

     version: "1.0",

     name: "Blocked add-on-1 with to-be-reset prefs",

     targetApplications: [{

       id: "xpcshell@tests.mozilla.org",

       minVersion: "1",

       maxVersion: "3"

     }]

   }, profileDir);

   writeInstallRDFForExtension({

     id: "block2@tests.mozilla.org",

     version: "1.0",

     name: "Blocked add-on-2 with to-be-reset prefs",

     targetApplications: [{

       id: "xpcshell@tests.mozilla.org",

       minVersion: "1",

       maxVersion: "3"

     }]

   }, profileDir);

   // Pre-set the preferences that we expect to get reset.

   gPref.setIntPref("test.blocklist.pref1", 15);

   gPref.setIntPref("test.blocklist.pref2", 15);

   gPref.setBoolPref("test.blocklist.pref3", true);

   gPref.setBoolPref("test.blocklist.pref4", true);

   startupManager();

   // Before blocklist is loaded.

   AddonManager.getAddonsByIDs(["block1@tests.mozilla.org",

                                "block2@tests.mozilla.org"], function([a1, a2]) {

     do_check_eq(a1.blocklistState, Ci.nsIBlocklistService.STATE_NOT_BLOCKED);

     do_check_eq(a2.blocklistState, Ci.nsIBlocklistService.STATE_NOT_BLOCKED);

     do_check_eq(gPref.getIntPref("test.blocklist.pref1"), 15);

     do_check_eq(gPref.getIntPref("test.blocklist.pref2"), 15);

     do_check_eq(gPref.getBoolPref("test.blocklist.pref3"), true);

     do_check_eq(gPref.getBoolPref("test.blocklist.pref4"), true);

     run_test_1();

   });

 }

 function run_test_1() {

   load_blocklist("test_blocklist_prefs_1.xml", function() {

     restartManager();

     // Blocklist changes should have applied and the prefs must be reset.

     AddonManager.getAddonsByIDs(["block1@tests.mozilla.org",

                                  "block2@tests.mozilla.org"], function([a1, a2]) {

       do_check_neq(a1, null);

       do_check_eq(a1.blocklistState, Ci.nsIBlocklistService.STATE_SOFTBLOCKED);

       do_check_neq(a2, null);

       do_check_eq(a2.blocklistState, Ci.nsIBlocklistService.STATE_BLOCKED);

       // All these prefs must be reset to defaults.

       do_check_eq(gPref.prefHasUserValue("test.blocklist.pref1"), false);

       do_check_eq(gPref.prefHasUserValue("test.blocklist.pref2"), false);

       do_check_eq(gPref.prefHasUserValue("test.blocklist.pref3"), false);

       do_check_eq(gPref.prefHasUserValue("test.blocklist.pref4"), false);

       end_test();

     });

   });

 }