1 files changed, 204 insertions, 0 deletions
diff --git a/tools/addon-sdk-1.7/packages/addon-kit/tests/test-private-browsing.js b/tools/addon-sdk-1.7/packages/addon-kit/tests/test-private-browsing.js
new file mode 100644
index 0000000..1d60f6b
--- /dev/null
+++ b/tools/addon-sdk-1.7/packages/addon-kit/tests/test-private-browsing.js
@@ -0,0 +1,204 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+let pb = require("private-browsing");
+let {Cc,Ci} = require("chrome");
+const { Loader } = require('./helpers');
+
+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 = Loader(module);
+    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");
+  };
+}