1 files changed, 177 insertions, 0 deletions
diff --git a/browser/components/preferences/tests/browser_security-2.js b/browser/components/preferences/tests/browser_security-2.js
new file mode 100644
index 0000000000..07e571cd2e
--- /dev/null
+++ b/browser/components/preferences/tests/browser_security-2.js
@@ -0,0 +1,177 @@
+const PREFS = [
+  "browser.safebrowsing.phishing.enabled",
+  "browser.safebrowsing.malware.enabled",
+
+  "browser.safebrowsing.downloads.enabled",
+
+  "browser.safebrowsing.downloads.remote.block_potentially_unwanted",
+  "browser.safebrowsing.downloads.remote.block_uncommon",
+];
+
+let originals = PREFS.map(pref => [pref, Services.prefs.getBoolPref(pref)]);
+let originalMalwareTable = Services.prefs.getCharPref(
+  "urlclassifier.malwareTable"
+);
+registerCleanupFunction(function() {
+  originals.forEach(([pref, val]) => Services.prefs.setBoolPref(pref, val));
+  Services.prefs.setCharPref(
+    "urlclassifier.malwareTable",
+    originalMalwareTable
+  );
+});
+
+// This test only opens the Preferences once, and then reloads the page
+// each time that it wants to test various preference combinations. We
+// only use one tab (instead of opening/closing for each test) for all
+// to help improve test times on debug builds.
+add_setup(async function() {
+  await openPreferencesViaOpenPreferencesAPI("privacy", { leaveOpen: true });
+  registerCleanupFunction(async function() {
+    BrowserTestUtils.removeTab(gBrowser.selectedTab);
+  });
+});
+
+// test the download protection preference
+add_task(async function() {
+  async function checkPrefSwitch(val) {
+    Services.prefs.setBoolPref("browser.safebrowsing.downloads.enabled", val);
+
+    gBrowser.reload();
+    await BrowserTestUtils.browserLoaded(gBrowser.selectedBrowser);
+
+    let doc = gBrowser.selectedBrowser.contentDocument;
+    let checkbox = doc.getElementById("blockDownloads");
+
+    let blockUncommon = doc.getElementById("blockUncommonUnwanted");
+    let checked = checkbox.checked;
+    is(checked, val, "downloads preference is initialized correctly");
+    // should be disabled when val is false (= pref is turned off)
+    is(
+      blockUncommon.hasAttribute("disabled"),
+      !val,
+      "block uncommon checkbox is set correctly"
+    );
+
+    // scroll the checkbox into view, otherwise the synthesizeMouseAtCenter will be ignored, and click it
+    checkbox.scrollIntoView();
+    EventUtils.synthesizeMouseAtCenter(
+      checkbox,
+      {},
+      gBrowser.selectedBrowser.contentWindow
+    );
+
+    // check that setting is now turned on or off
+    is(
+      Services.prefs.getBoolPref("browser.safebrowsing.downloads.enabled"),
+      !checked,
+      "safebrowsing.downloads preference is set correctly"
+    );
+
+    // check if the uncommon warning checkbox has updated
+    is(
+      blockUncommon.hasAttribute("disabled"),
+      val,
+      "block uncommon checkbox is set correctly"
+    );
+  }
+
+  await checkPrefSwitch(true);
+  await checkPrefSwitch(false);
+});
+
+requestLongerTimeout(2);
+// test the unwanted/uncommon software warning preference
+add_task(async function() {
+  async function checkPrefSwitch(val1, val2, isV2) {
+    Services.prefs.setBoolPref(
+      "browser.safebrowsing.downloads.remote.block_potentially_unwanted",
+      val1
+    );
+    Services.prefs.setBoolPref(
+      "browser.safebrowsing.downloads.remote.block_uncommon",
+      val2
+    );
+    let testMalwareTable = "goog-malware-" + (isV2 ? "shavar" : "proto");
+    testMalwareTable += ",test-malware-simple";
+    if (val1 && val2) {
+      testMalwareTable += ",goog-unwanted-" + (isV2 ? "shavar" : "proto");
+      testMalwareTable += ",moztest-unwanted-simple";
+    }
+    Services.prefs.setCharPref("urlclassifier.malwareTable", testMalwareTable);
+
+    gBrowser.reload();
+    await BrowserTestUtils.browserLoaded(gBrowser.selectedBrowser);
+
+    let doc = gBrowser.selectedBrowser.contentDocument;
+    let checkbox = doc.getElementById("blockUncommonUnwanted");
+    let checked = checkbox.checked;
+    is(
+      checked,
+      val1 && val2,
+      "unwanted/uncommon preference is initialized correctly"
+    );
+
+    // scroll the checkbox into view, otherwise the synthesizeMouseAtCenter will be ignored, and click it
+    checkbox.scrollIntoView();
+    EventUtils.synthesizeMouseAtCenter(
+      checkbox,
+      {},
+      gBrowser.selectedBrowser.contentWindow
+    );
+
+    // check that both settings are now turned on or off
+    is(
+      Services.prefs.getBoolPref(
+        "browser.safebrowsing.downloads.remote.block_potentially_unwanted"
+      ),
+      !checked,
+      "block_potentially_unwanted is set correctly"
+    );
+    is(
+      Services.prefs.getBoolPref(
+        "browser.safebrowsing.downloads.remote.block_uncommon"
+      ),
+      !checked,
+      "block_uncommon is set correctly"
+    );
+
+    // when the preference is on, the malware table should include these ids
+    let malwareTable = Services.prefs
+      .getCharPref("urlclassifier.malwareTable")
+      .split(",");
+    if (isV2) {
+      is(
+        malwareTable.includes("goog-unwanted-shavar"),
+        !checked,
+        "malware table doesn't include goog-unwanted-shavar"
+      );
+    } else {
+      is(
+        malwareTable.includes("goog-unwanted-proto"),
+        !checked,
+        "malware table doesn't include goog-unwanted-proto"
+      );
+    }
+    is(
+      malwareTable.includes("moztest-unwanted-simple"),
+      !checked,
+      "malware table doesn't include moztest-unwanted-simple"
+    );
+    let sortedMalware = malwareTable.slice(0);
+    sortedMalware.sort();
+    Assert.deepEqual(
+      malwareTable,
+      sortedMalware,
+      "malware table has been sorted"
+    );
+  }
+
+  await checkPrefSwitch(true, true, false);
+  await checkPrefSwitch(false, true, false);
+  await checkPrefSwitch(true, false, false);
+  await checkPrefSwitch(false, false, false);
+  await checkPrefSwitch(true, true, true);
+  await checkPrefSwitch(false, true, true);
+  await checkPrefSwitch(true, false, true);
+  await checkPrefSwitch(false, false, true);
+});