fune/browser/modules/test/unit/test_Sanitizer_interrupted.js
Marco Bonardo fa2de7886f Bug 1434414 - Locking the 'sanitize on shutdown' pref causes sanitization to happen at every startup. r=johannh
MozReview-Commit-ID: 6PvRFmaZsBC

--HG--
extra : rebase_source : 9b9a670abf6037d239b3b8efd025b9341f45d820
2018-02-15 00:42:58 +01:00

67 lines
3.5 KiB
JavaScript

/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
ChromeUtils.import("resource://gre/modules/Services.jsm");
do_get_profile();
// Test that interrupted sanitizations are properly tracked.
add_task(async function() {
ChromeUtils.import("resource:///modules/Sanitizer.jsm");
registerCleanupFunction(() => {
Services.prefs.clearUserPref(Sanitizer.PREF_SANITIZE_ON_SHUTDOWN);
Services.prefs.clearUserPref(Sanitizer.PREF_SHUTDOWN_BRANCH + "formdata");
});
Services.prefs.setBoolPref(Sanitizer.PREF_SANITIZE_ON_SHUTDOWN, true);
Services.prefs.setBoolPref(Sanitizer.PREF_SHUTDOWN_BRANCH + "formdata", true);
await Sanitizer.onStartup();
Assert.ok(Sanitizer.shouldSanitizeOnShutdown, "Should sanitize on shutdown");
let pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 1, "Should have 1 pending sanitization");
Assert.equal(pendingSanitizations[0].id, "shutdown", "Should be the shutdown sanitization");
Assert.ok(pendingSanitizations[0].itemsToClear.includes("formdata"), "Pref has been setup");
Assert.ok(!pendingSanitizations[0].options.isShutdown, "Shutdown option is not present");
// Check the preference listeners.
Services.prefs.setBoolPref(Sanitizer.PREF_SANITIZE_ON_SHUTDOWN, false);
pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 0, "Should not have pending sanitizations");
Assert.ok(!Sanitizer.shouldSanitizeOnShutdown, "Should not sanitize on shutdown");
Services.prefs.setBoolPref(Sanitizer.PREF_SANITIZE_ON_SHUTDOWN, true);
pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 1, "Should have 1 pending sanitization");
Assert.equal(pendingSanitizations[0].id, "shutdown", "Should be the shutdown sanitization");
Assert.ok(pendingSanitizations[0].itemsToClear.includes("formdata"),
"Pending sanitizations should include formdata");
Services.prefs.setBoolPref(Sanitizer.PREF_SHUTDOWN_BRANCH + "formdata", false);
pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 1, "Should have 1 pending sanitization");
Assert.ok(!pendingSanitizations[0].itemsToClear.includes("formdata"),
"Pending sanitizations should have been updated");
// Check a sanitization properly rebuilds the pref.
await Sanitizer.sanitize(["formdata"]);
pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 1, "Should have 1 pending sanitization");
Assert.equal(pendingSanitizations[0].id, "shutdown", "Should be the shutdown sanitization");
// Startup should run the pending one and setup a new shutdown sanitization.
Services.prefs.setBoolPref(Sanitizer.PREF_SHUTDOWN_BRANCH + "formdata", false);
await Sanitizer.onStartup();
pendingSanitizations = JSON.parse(
Services.prefs.getStringPref(Sanitizer.PREF_PENDING_SANITIZATIONS, "[]"));
Assert.equal(pendingSanitizations.length, 1, "Should have 1 pending sanitization");
Assert.equal(pendingSanitizations[0].id, "shutdown", "Should be the shutdown sanitization");
Assert.ok(!pendingSanitizations[0].itemsToClear.includes("formdata"), "Pref has been setup");
});