gecko/dom/tests/browser/browser_ConsoleStoragePBTest.js

/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

function test() {
  try {
    var pb = Cc["@mozilla.org/privatebrowsing;1"].getService(Ci.nsIPrivateBrowsingService);
  } catch (ex) {
    ok(true, "nothing to do here, PB service doesn't exist");
    return;
  }

  waitForExplicitFinish();

  var CSS = {};
  Cu.import("resource://gre/modules/ConsoleAPIStorage.jsm", CSS);

  function checkStorageOccurs(shouldOccur) {
    let win = XPCNativeWrapper.unwrap(browser.contentWindow);
    let innerID = getInnerWindowId(win);

    let beforeEvents = CSS.ConsoleAPIStorage.getEvents(innerID);
    win.console.log("foo bar baz (private: " + !shouldOccur + ")");

    let afterEvents = CSS.ConsoleAPIStorage.getEvents(innerID);

    is(beforeEvents.length == afterEvents.length - 1,
       shouldOccur, "storage should" + (shouldOccur ? "" : "n't") + " occur");
  }

  function pbObserver(aSubject, aTopic, aData) {
    if (aData == "enter") {
      checkStorageOccurs(false);

      executeSoon(function () { pb.privateBrowsingEnabled = false; });
    } else if (aData == "exit") {
      executeSoon(finish);
    }
  }

  const TEST_URI = "http://example.com/browser/dom/tests/browser/test-console-api.html";
  var tab = gBrowser.selectedTab = gBrowser.addTab(TEST_URI);
  var browser = gBrowser.selectedBrowser;

  Services.obs.addObserver(pbObserver, "private-browsing", false);

  const PB_KEEP_SESSION_PREF = "browser.privatebrowsing.keep_current_session";
  Services.prefs.setBoolPref(PB_KEEP_SESSION_PREF, true);

  registerCleanupFunction(function () {
    gBrowser.removeTab(tab);

    Services.obs.removeObserver(pbObserver, "private-browsing");

    if (Services.prefs.prefHasUserValue(PB_KEEP_SESSION_PREF))
      Services.prefs.clearUserPref(PB_KEEP_SESSION_PREF);
  });

  browser.addEventListener("DOMContentLoaded", function onLoad(event) {
    if (browser.currentURI.spec != TEST_URI)
      return;

    browser.removeEventListener("DOMContentLoaded", onLoad, false);

    checkStorageOccurs(true);

    pb.privateBrowsingEnabled = true;
  }, false);
}

function getInnerWindowId(aWindow) {
  return aWindow.QueryInterface(Ci.nsIInterfaceRequestor)
                .getInterface(Ci.nsIDOMWindowUtils)
                .currentInnerWindowID;
}