1 files changed, 49 insertions, 0 deletions
diff --git a/dom/browser-element/mochitest/browserElement_PrivateBrowsing.js b/dom/browser-element/mochitest/browserElement_PrivateBrowsing.js
new file mode 100644
index 000000000..c32edcc37
--- /dev/null
+++ b/dom/browser-element/mochitest/browserElement_PrivateBrowsing.js
@@ -0,0 +1,49 @@
+/* Any copyright is dedicated to the public domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+// Test that the mozprivatebrowsing attribute works.
+"use strict";
+
+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);