1 files changed, 99 insertions, 0 deletions
diff --git a/dom/base/test/unit/test_thirdpartyutil.js b/dom/base/test/unit/test_thirdpartyutil.js
new file mode 100644
index 0000000000..777dc8dc62
--- /dev/null
+++ b/dom/base/test/unit/test_thirdpartyutil.js
@@ -0,0 +1,99 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+// Test ThirdPartyUtil methods. See mozIThirdPartyUtil.
+
+var prefs = Services.prefs;
+
+// Since this test creates a TYPE_DOCUMENT channel via javascript, it will
+// end up using the wrong LoadInfo constructor. Setting this pref will disable
+// the ContentPolicyType assertion in the constructor.
+prefs.setBoolPref("network.loadinfo.skip_type_assertion", true);
+
+var NS_ERROR_INVALID_ARG = Cr.NS_ERROR_INVALID_ARG;
+
+function do_check_throws(f, result, stack) {
+  if (!stack) {
+    try {
+      // We might not have a 'Components' object.
+      stack = Components.stack.caller;
+    } catch (e) {}
+  }
+
+  try {
+    f();
+  } catch (exc) {
+    Assert.equal(exc.result, result);
+    return;
+  }
+  do_throw("expected " + result + " exception, none thrown", stack);
+}
+
+function run_test() {
+  let util = Cc["@mozilla.org/thirdpartyutil;1"].getService(
+    Ci.mozIThirdPartyUtil
+  );
+
+  // Create URIs and channels pointing to foo.com and bar.com.
+  // We will use these to put foo.com into first and third party contexts.
+  let spec1 = "http://foo.com/foo.html";
+  let spec2 = "http://bar.com/bar.html";
+  let uri1 = NetUtil.newURI(spec1);
+  let uri2 = NetUtil.newURI(spec2);
+  const contentPolicyType = Ci.nsIContentPolicy.TYPE_DOCUMENT;
+  let channel1 = NetUtil.newChannel({
+    uri: uri1,
+    loadUsingSystemPrincipal: true,
+    contentPolicyType,
+  });
+  NetUtil.newChannel({
+    uri: uri2,
+    loadUsingSystemPrincipal: true,
+    contentPolicyType,
+  });
+
+  // Create some file:// URIs.
+  let filespec1 = "file://foo.txt";
+  let filespec2 = "file://bar.txt";
+  let fileuri1 = NetUtil.newURI(filespec1);
+  let fileuri2 = NetUtil.newURI(filespec2);
+  NetUtil.newChannel({ uri: fileuri1, loadUsingSystemPrincipal: true });
+  NetUtil.newChannel({ uri: fileuri2, loadUsingSystemPrincipal: true });
+
+  // Test isThirdPartyURI.
+  Assert.ok(!util.isThirdPartyURI(uri1, uri1));
+  Assert.ok(util.isThirdPartyURI(uri1, uri2));
+  Assert.ok(util.isThirdPartyURI(uri2, uri1));
+  Assert.ok(!util.isThirdPartyURI(fileuri1, fileuri1));
+  Assert.ok(!util.isThirdPartyURI(fileuri1, fileuri2));
+  Assert.ok(util.isThirdPartyURI(uri1, fileuri1));
+  do_check_throws(function () {
+    util.isThirdPartyURI(uri1, null);
+  }, NS_ERROR_INVALID_ARG);
+  do_check_throws(function () {
+    util.isThirdPartyURI(null, uri1);
+  }, NS_ERROR_INVALID_ARG);
+  do_check_throws(function () {
+    util.isThirdPartyURI(null, null);
+  }, NS_ERROR_INVALID_ARG);
+
+  // We can't test isThirdPartyWindow since we can't really set up a window
+  // hierarchy. We leave that to mochitests.
+
+  // Test isThirdPartyChannel. As above, we can't test the bits that require
+  // a load context or window heirarchy. Because of bug 1259873, we assume
+  // that these are not third-party.
+  do_check_throws(function () {
+    util.isThirdPartyChannel(null);
+  }, NS_ERROR_INVALID_ARG);
+  Assert.ok(!util.isThirdPartyChannel(channel1));
+  Assert.ok(!util.isThirdPartyChannel(channel1, uri1));
+  Assert.ok(util.isThirdPartyChannel(channel1, uri2));
+
+  let httpchannel1 = channel1.QueryInterface(Ci.nsIHttpChannelInternal);
+  httpchannel1.forceAllowThirdPartyCookie = true;
+  Assert.ok(!util.isThirdPartyChannel(channel1));
+  Assert.ok(!util.isThirdPartyChannel(channel1, uri1));
+  Assert.ok(util.isThirdPartyChannel(channel1, uri2));
+}