gecko-dev/dom/browser-element/mochitest/browserElement_PrivateBrowsing.js

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

// Test that the mozprivatebrowsing attribute works.
"use strict";

/* global browserElementTestHelpers */

SimpleTest.waitForExplicitFinish();
browserElementTestHelpers.setEnabledPref(true);
browserElementTestHelpers.addPermission();

function createFrame(aIsPrivate) {
  var iframe = document.createElement("iframe");
  iframe.setAttribute("mozbrowser", "true");
  if (aIsPrivate) {
    iframe.setAttribute("mozprivatebrowsing", "true");
  }
  return iframe;
}

function createTest(aIsPrivate, aExpected, aClearStorage) {
  info("createTest " + aIsPrivate + " " + aExpected);
  return new Promise(function(resolve, reject) {
    var iframe = createFrame(aIsPrivate);
    document.body.appendChild(iframe);

    iframe.addEventListener("mozbrowsershowmodalprompt", function(e) {
      is(e.detail.message, aExpected, "Checking localstorage");
      resolve();
    });

    var src = "file_browserElement_PrivateBrowsing.html";
    iframe.src = aClearStorage ? src + "?clear=true" : src;
  });
}

function runTest() {
  // We first create a iframe in non private browsing mode, set up some
  // localstorage, reopen it to check that we get the previously set value.
  // Finally, open it in private browsing mode and check that localstorage
  // is clear.
  createTest(false, "CLEAR", true)
    .then(() => {
      return createTest(false, "EMPTY", false);
    })
    .then(() => {
      return createTest(false, "bar", false);
    })
    .then(() => {
      return createTest(true, "EMPTY", false);
    })
    .then(SimpleTest.finish);
}

addEventListener("testready", runTest);