gecko/netwerk/test/unit/test_bug826063.js

104 lines
2.4 KiB
JavaScript

/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
/**
* Test that nsIPrivateBrowsingChannel.isChannelPrivate yields the correct
* result for various combinations of .setPrivate() and nsILoadContexts
*/
const {classes: Cc, interfaces: Ci, utils: Cu} = Components;
Cu.import("resource://gre/modules/XPCOMUtils.jsm");
Cu.import("resource://gre/modules/Services.jsm");
var URIs = [
"http://example.org",
"https://example.org",
"ftp://example.org"
];
function LoadContext(usePrivateBrowsing) {
this.usePrivateBrowsing = usePrivateBrowsing;
}
LoadContext.prototype = {
QueryInterface: XPCOMUtils.generateQI([Ci.nsILoadContext, Ci.nsIInterfaceRequestor]),
getInterface: XPCOMUtils.generateQI([Ci.nsILoadContext])
};
function getChannels() {
for (let u of URIs) {
yield Services.io.newChannel(u, null, null);
}
}
function checkPrivate(channel, shouldBePrivate) {
do_check_eq(channel.QueryInterface(Ci.nsIPrivateBrowsingChannel).isChannelPrivate,
shouldBePrivate);
}
/**
* Default configuration
* Default is non-private
*/
add_test(function test_plain() {
for (let c of getChannels()) {
checkPrivate(c, false);
}
run_next_test();
});
/**
* Explicitly setPrivate(true), no load context
*/
add_test(function test_setPrivate_private() {
for (let c of getChannels()) {
c.QueryInterface(Ci.nsIPrivateBrowsingChannel).setPrivate(true);
checkPrivate(c, true);
}
run_next_test();
});
/**
* Explicitly setPrivate(false), no load context
*/
add_test(function test_setPrivate_regular() {
for (let c of getChannels()) {
c.QueryInterface(Ci.nsIPrivateBrowsingChannel).setPrivate(false);
checkPrivate(c, false);
}
run_next_test();
});
/**
* Load context mandates private mode
*/
add_test(function test_LoadContextPrivate() {
let ctx = new LoadContext(true);
for (let c of getChannels()) {
c.notificationCallbacks = ctx;
checkPrivate(c, true);
}
run_next_test();
});
/**
* Load context mandates regular mode
*/
add_test(function test_LoadContextRegular() {
let ctx = new LoadContext(false);
for (let c of getChannels()) {
c.notificationCallbacks = ctx;
checkPrivate(c, false);
}
run_next_test();
});
// Do not test simultanous uses of .setPrivate and load context.
// There is little merit in doing so, and combining both will assert in
// Debug builds anyway.
function run_test() {
run_next_test();
}