Bug 722861 - Tests for imgLoader privacy-respecting changes. r=joe

image/test/unit/test_private_channel.js

@@ -0,0 +1,115 @@
+Components.utils.import("resource://testing-common/httpd.js");
+const Cc = Components.classes;
+const Ci = Components.interfaces;
+
+var server = new HttpServer();
+server.registerPathHandler('/image.png', imageHandler);
+server.start(8088);
+
+load('image_load_helpers.js');
+
+var gHits = 0;
+
+var gIoService = Cc["@mozilla.org/network/io-service;1"].getService(Ci.nsIIOService);  
+var gLoader = Cc["@mozilla.org/image/loader;1"].getService(Ci.imgILoader);
+
+function imageHandler(metadata, response) {
+  gHits++;
+  response.setHeader("Cache-Control", "max-age=10000", false);
+  response.setStatusLine(metadata.httpVersion, 200, "OK");
+  response.setHeader("Content-Type", "image/png", false);
+  var body = "iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAIAAADZSiLoAAAAEUlEQVQImWP4z8AAQTAamQkAhpcI+DeMzFcAAAAASUVORK5CYII=";
+  response.bodyOutputStream.write(body, body.length);
+}
+
+var requests = [];
+var listeners = [];
+
+function NotificationCallbacks(isPrivate) {
+  this.usePrivateBrowsing = isPrivate;
+}
+
+NotificationCallbacks.prototype = {
+  QueryInterface: function (iid) {
+    if (iid.equals(Ci.nsISupports) ||
+        iid.equals(Ci.nsILoadContext))
+      return this;
+    throw Cr.NS_ERROR_NO_INTERFACE;
+  },
+  getInterface: function(iid) {
+    if (iid.equals(Ci.nsILoadContext))
+      return this;
+    throw Cr.NS_ERROR_NO_INTERFACE;
+  }
+};
+
+var gImgPath = 'http://localhost:8088/image.png';
+
+function setup_chan(path, isPrivate, callback) {
+  var uri = gIoService.newURI(gImgPath, null, null);
+  var chan = gIoService.newChannelFromURI(uri);
+  chan.notificationCallbacks = new NotificationCallbacks(isPrivate);
+  var channelListener = new ChannelListener();
+  chan.asyncOpen(channelListener, null);
+  
+  var listener = new ImageListener(null, callback);
+  listeners.push(listener);
+  var outlistener = {};
+  requests.push(gLoader.loadImageWithChannel(chan, listener, null, outlistener));
+  channelListener.outputListener = outlistener.value;
+  listener.synchronous = false;
+}
+
+function loadImage(isPrivate, callback) {
+  var listener = new ImageListener(null, callback);
+  var uri = gIoService.newURI(gImgPath, null, null);
+  var loadGroup = Cc["@mozilla.org/network/load-group;1"].createInstance(Ci.nsILoadGroup);
+  loadGroup.notificationCallbacks = new NotificationCallbacks(isPrivate);
+  requests.push(gLoader.loadImage(uri, null, null, null, loadGroup, listener, null, 0, null, null, null));
+  listener.synchronous = false;  
+}
+
+function run_loadImage_tests() {
+  clearAllImageCaches();
+  let cs = Cc["@mozilla.org/network/cache-service;1"].getService(Ci.nsICacheService);
+  cs.evictEntries(Ci.nsICache.STORE_ANYWHERE);
+
+  gHits = 0;
+  loadImage(false, function() {
+    loadImage(false, function() {
+      loadImage(true, function() {
+        loadImage(true, function() {
+          do_check_eq(gHits, 2);
+          do_test_finished();
+        });
+      });
+    });
+  });
+}
+
+function clearAllImageCaches() {
+  gLoader.QueryInterface(Ci.imgICache);
+  gLoader.clearCache(false, false);
+  gLoader.clearCache(false, true);
+}
+
+function run_test() {
+  clearAllImageCaches();
+  do_test_pending();
+ 
+  // We create a public channel that loads an image, then an identical
+  // one that should cause a cache read. We then create a private channel
+  // and load the same image, and do that a second time to ensure a cache
+  // read. In total, we should cause two separate http responses to occur,
+  // since the private channels shouldn't be able to use the public cache.
+  setup_chan('/image.png', false, function() {
+    setup_chan('/image.png', false, function() {
+      setup_chan('/image.png', true, function() {
+        setup_chan('/image.png', true, function() {
+          do_check_eq(gHits, 2);
+          run_loadImage_tests();
+        });
+      });
+    });
+  });
+}

image/test/unit/xpcshell.ini

@@ -11,3 +11,4 @@ tail =
 # Bug 676968: test fails consistently on Android
 fail-if = os == "android"
 [test_moz_icon_uri.js]
+[test_private_channel.js]