fune/browser/modules/test/unit/test_Sanitizer_interrupted.js

71 lines
3.7 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");
Services.prefs.setBoolPref(Sanitizer.PREF_NEWTAB_SEGREGATION, false);
registerCleanupFunction(() => {
Services.prefs.clearUserPref(Sanitizer.PREF_SANITIZE_ON_SHUTDOWN);
Services.prefs.clearUserPref(Sanitizer.PREF_SHUTDOWN_BRANCH + "formdata");
Services.prefs.clearUserPref(Sanitizer.PREF_NEWTAB_SEGREGATION);
});
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");
});