1 files changed, 237 insertions, 0 deletions
diff --git a/tools/addon-sdk-1.3/packages/addon-kit/tests/test-private-browsing.js b/tools/addon-sdk-1.3/packages/addon-kit/tests/test-private-browsing.js
new file mode 100644
index 0000000..268e933
--- /dev/null
+++ b/tools/addon-sdk-1.3/packages/addon-kit/tests/test-private-browsing.js
@@ -0,0 +1,237 @@
+/* ***** BEGIN LICENSE BLOCK *****
+ * Version: MPL 1.1/GPL 2.0/LGPL 2.1
+ *
+ * The contents of this file are subject to the Mozilla Public License Version
+ * 1.1 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ * http://www.mozilla.org/MPL/
+ *
+ * Software distributed under the License is distributed on an "AS IS" basis,
+ * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
+ * for the specific language governing rights and limitations under the
+ * License.
+ *
+ * The Original Code is Jetpack.
+ *
+ * The Initial Developer of the Original Code is
+ * Mozilla Foundation.
+ * Portions created by the Initial Developer are Copyright (C) 2010
+ * the Initial Developer. All Rights Reserved.
+ *
+ * Contributor(s):
+ *  Paul O’Shannessy <paul@oshannessy.com>
+ *  Irakli Gozalishvili <gozala@mozilla.com>
+ *
+ * Alternatively, the contents of this file may be used under the terms of
+ * either the GNU General Public License Version 2 or later (the "GPL"), or
+ * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
+ * in which case the provisions of the GPL or the LGPL are applicable instead
+ * of those above. If you wish to allow use of your version of this file only
+ * under the terms of either the GPL or the LGPL, and not to allow others to
+ * use your version of this file under the terms of the MPL, indicate your
+ * decision by deleting the provisions above and replace them with the notice
+ * and other provisions required by the GPL or the LGPL. If you do not delete
+ * the provisions above, a recipient may use your version of this file under
+ * the terms of any one of the MPL, the GPL or the LGPL.
+ *
+ * ***** END LICENSE BLOCK ***** */
+
+let pb = require("private-browsing");
+let {Cc,Ci} = require("chrome");
+
+let pbService;
+// Currently, only Firefox implements the private browsing service.
+if (require("xul-app").is("Firefox")) {
+  pbService = Cc["@mozilla.org/privatebrowsing;1"].
+              getService(Ci.nsIPrivateBrowsingService);
+}
+
+if (pbService) {
+
+  // tests that isActive has the same value as the private browsing service
+  // expects
+  exports.testGetIsActive = function (test) {
+    test.assertEqual(pb.isActive, false,
+                     "private-browsing.isActive is correct without modifying PB service");
+
+    pbService.privateBrowsingEnabled = true;
+    test.assert(pb.isActive,
+                "private-browsing.isActive is correct after modifying PB service");
+
+    // Switch back to normal mode.
+    pbService.privateBrowsingEnabled = false;
+  };
+
+  // tests that activating does put the browser into private browsing mode
+  exports.testActivateDeactivate = function (test) {
+    test.waitUntilDone();
+    pb.once("start", function onStart() {
+      test.assertEqual(pbService.privateBrowsingEnabled, true,
+                       "private browsing mode was activated");
+      pb.deactivate();
+    });
+    pb.once("stop", function onStop() {
+      test.assertEqual(pbService.privateBrowsingEnabled, false,
+                       "private browsing mode was deactivate");
+      test.done();
+    });
+    pb.activate();
+  };
+
+  exports.testStart = function(test) {
+    test.waitUntilDone();
+    pb.on("start", function onStart() {
+      test.assertEqual(this, pb, "`this` should be private-browsing module");
+      test.assert(pbService.privateBrowsingEnabled,
+                  'private mode is active when "start" event is emitted');
+      test.assert(pb.isActive,
+                  '`isActive` is `true` when "start" event is emitted');
+      pb.removeListener("start", onStart);
+      test.done();
+    });
+    pb.activate();
+  };
+
+  exports.testStop = function(test) {
+    test.waitUntilDone();
+    pb.on("stop", function onStop() {
+      test.assertEqual(this, pb, "`this` should be private-browsing module");
+      test.assertEqual(pbService.privateBrowsingEnabled, false,
+                       "private mode is disabled when stop event is emitted");
+      test.assertEqual(pb.isActive, false,
+                       "`isActive` is `false` when stop event is emitted");
+      pb.removeListener("stop", onStop);
+      test.done();
+    });
+    pb.activate();
+    pb.deactivate();
+  };
+  
+  exports.testAutomaticUnload = function(test) {
+    test.waitUntilDone();
+    // Create another private browsing instance and unload it
+    let loader = test.makeSandboxedLoader();
+    let pb2 = loader.require("private-browsing");
+    let called = false;
+    pb2.on("start", function onStart() {
+      called = true;
+      test.fail("should not be called:x");
+    });
+    loader.unload();
+
+    // Then switch to private mode in order to check that the previous instance
+    // is correctly destroyed
+    pb.activate();
+    pb.once("start", function onStart() {
+      require("timer").setTimeout(function () {
+        test.assert(!called, 
+          "First private browsing instance is destroyed and inactive");
+
+        // Must reset to normal mode, so that next test starts with it.
+        pb.deactivate();
+        test.done();
+      }, 0);
+    });
+  };
+
+  exports.testBothListeners = function(test) {
+    test.waitUntilDone();
+    let stop = false;
+    let start = false;
+
+    function onStop() {
+      test.assertEqual(stop, false,
+                       "stop callback must be called only once");
+      test.assertEqual(pbService.privateBrowsingEnabled, false,
+                       "private mode is disabled when stop event is emitted");
+      test.assertEqual(pb.isActive, false,
+                       "`isActive` is `false` when stop event is emitted");
+
+      pb.on("start", finish);
+      pb.removeListener("start", onStart);
+      pb.removeListener("start", onStart2);
+      pb.activate();
+      stop = true;
+    }
+
+    function onStart() {
+      test.assertEqual(false, start,
+                       "stop callback must be called only once");
+      test.assert(pbService.privateBrowsingEnabled,
+                  "private mode is active when start event is emitted");
+      test.assert(pb.isActive,
+                  "`isActive` is `true` when start event is emitted");
+
+      pb.on("stop", onStop);
+      pb.deactivate();
+      start = true;
+    }
+
+    function onStart2() {
+      test.assert(start, "start listener must be called already");
+      test.assertEqual(false, stop, "stop callback must not be called yet");
+    }
+
+    function finish() {
+      test.assert(pbService.privateBrowsingEnabled, true,
+                  "private mode is active when start event is emitted");
+      test.assert(pb.isActive,
+                  "`isActive` is `true` when start event is emitted");
+
+      pb.removeListener("start", finish);
+      pb.removeListener("stop", onStop);
+
+      pb.deactivate();
+      pb.once("stop", function () {
+        test.assertEqual(pbService.privateBrowsingEnabled, false);
+        test.assertEqual(pb.isActive, false);
+
+        test.done();
+      });
+    }
+
+    pb.on("start", onStart);
+    pb.on("start", onStart2);
+    pbService.privateBrowsingEnabled = true;
+  };
+
+  exports["test activate private mode via handler"] = function(test) {
+    const tabs = require("tabs");
+
+    test.waitUntilDone();
+    function onReady(tab) {
+      if (tab.url == "about:robots")
+        tab.close(function() pb.activate());
+    }
+    function cleanup(tab) {
+      if (tab.url == "about:") {
+        tabs.removeListener("ready", cleanup);
+        tab.close(function onClose() {
+          test.done();
+        });
+      }
+    }
+
+    tabs.on("ready", onReady);
+    pb.once("start", function onStart() {
+      test.pass("private mode was activated");
+      pb.deactivate();
+    });
+    pb.once("stop", function onStop() {
+      test.pass("private mode was deactivated");
+      tabs.removeListener("ready", onReady);
+      tabs.on("ready", cleanup);
+    });
+    tabs.once("open", function onOpen() {
+      tabs.open("about:robots");
+    });
+    tabs.open("about:");
+  };
+}
+else {
+  // tests for the case where private browsing doesn't exist
+  exports.testNoImpl = function (test) {
+    test.assertEqual(pb.isActive, false,
+                     "pb.isActive returns false when private browsing isn't supported");
+  };
+}