Issue 1561 - Move unit tests out of the browser`s content area

chrome/content/tests/policy.js

 (function()
 {
   let server = null;
   let frame = null;
   let requestNotifier = null;
+  let httpProtocol = null;
 
   module("Content policy", {
     setup: function()
     {
       prepareFilterComponents.call(this);
       preparePrefs.call(this);
 
       server = new nsHttpServer();
       server.start(1234);
 
       frame = document.createElementNS("http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul", "iframe");
       frame.setAttribute("type", "content");
       frame.style.visibility = "collapse";
       document.body.appendChild(frame);
 
       requestNotifier = new RequestNotifier(frame.contentWindow, onPolicyHit);
+
+      httpProtocol = Utils.httpProtocol;
+      Utils.httpProtocol = {userAgent: "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:30.0) Gecko/20100101 Firefox/30.0"};
     },
     teardown: function()
     {
       restoreFilterComponents.call(this);
       restorePrefs.call(this);
 
       stop();
       server.stop(function()
       {
         frame.parentElement.removeChild(frame);
         requestNotifier.shutdown();
 
         server = null;
         frame = null;
         requestNotifier = null;
 
         start();
       });
+
+      Utils.httpProtocol = httpProtocol;
     }
   });
+
+  /*
+  -----BEGIN RSA PRIVATE KEY-----
+  MIIBOQIBAAJBALZc50pEXnz9TSRozwM04rryuaXl/wgUFqV9FHq8HDlkdKvRU0hX
+  hb/AKrSpCJ0NCxHtal1l/kHYlHG9e7Ev6+MCAwEAAQJBALRxYs5irhgAz2b6afOj
+  TcFr0PRtipckwW/IPw5euZKyvswEJt/tWDv4OdmDnRe8FSy6FG2Got3zxvaxYdc3
+  AXkCIQDfFGcytIVq3sbdF3lmhzcXf29R4Hrxg/eoByAKabxknwIhANFGSNMOGPt6
+  JRajfB9XmsltQJzbkr2sfHgjMN2FLM49AiAH6tt2yz1o+5snQawHXYkxBk7XIxZ5
+  9+sURZx3giUzlQIfXF+pxX9zh41i0ZtYLn181WxkGNjS7OY2CtF9wEoIfQIgcHuf
+  shh1qrvuKiXnD9b72PF676laKdzxzX5rX6cZZLA=
+  -----END RSA PRIVATE KEY-----
+  */
+  let publickey = "MFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBALZc50pEXnz9TSRozwM04rryuaXl/wgUFqV9FHq8HDlkdKvRU0hXhb/AKrSpCJ0NCxHtal1l/kHYlHG9e7Ev6+MCAwEAAQ";
+
+  /**
+   * Content:
+   * /test\0127.0.0.1:1234\0Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:30.0) Gecko/20100101 Firefox/30.0
+   */
+  let adblockkey = "MFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBALZc50pEXnz9TSRozwM04rryuaXl/wgUFqV9FHq8HDlkdKvRU0hXhb/AKrSpCJ0NCxHtal1l/kHYlHG9e7Ev6+MCAwEAAQ==_gM4C/j8KkD2byPeP+THXk1GbLTUm5y+5jbdhcMtnzPMgImIfge0dGCtfU9cxLpe8BnqnEGNhTxpuu4pZxjOHYQ==";
 
   let tests = [
     [
       "HTML image with relative URL",
       '<img src="test.gif">',
       "http://127.0.0.1:1234/test.gif", "image", false, false
     ],
     [
       "HTML image with absolute URL",
       '<img src="http://localhost:1234/test.gif">',
@@ skipping 18 matching lines @@
       "Dynamically inserted image button",
       '<div id="insert"></div>' +
       '<script>' +
         'window.addEventListener("DOMContentLoaded", function()' +
         '{' +
           'var div = document.getElementById("insert");' +
           'div.innerHTML = \'<input type="image" id="image" src="test.gif">\';' +
           'var image = document.getElementById("image");' +
           'image.onload = image.onerror = function ()' +
           '{' +
-            'document.dispatchEvent(new CustomEvent("frameready", {bubbles: true}));' +
+            'document.dispatchEvent(new CustomEvent("abp:frameready", {bubbles: true}));' +
           '};' +
         '}, false);' +
       '</script>',
       "http://127.0.0.1:1234/test.gif", "image", false, true
     ],
     [
       "CSS background-image",
       '<div style="background-image: url(test.gif)"></div>',
       "http://127.0.0.1:1234/test.gif", "image", false, false
     ],
@@ skipping 94 matching lines @@
           'xmlDoc.async = false;' +
           'xmlDoc.load("test.xml");' +
         '}' +
         'catch(e){}' +
       '</script>',
       "http://127.0.0.1:1234/test.xml", "xmlhttprequest", false, false
     ],
     [
       "Web worker",
       '<script>' +
-        'var e = new CustomEvent(\'frameready\', {bubbles: true});' +
+        'var e = new CustomEvent(\'abp:frameready\', {bubbles: true});' +
         'try' +
         '{' +
           'var worker = new Worker("test.js");' +
           'worker.onerror = function(event)' +
           '{' +
             'event.preventDefault();' +
             'document.dispatchEvent(e);' +
           '};' +
         '}' +
         'catch (x)' +
@@ skipping 48 matching lines @@
     {
       return;
     }
     if (item.filter instanceof WhitelistFilter)
       return;
 
     if (policyHits.length > 0)
     {
       // Ignore duplicate policy calls (possible due to prefetching)
       let [prevWnd, prevNode, prevItem] = policyHits[policyHits.length - 1];
-      if (prevWnd == wnd && prevItem.location == item.location && prevItem.type == item.type &&  prevItem.docDomain == item.docDomain)
+      if (prevWnd == wnd && prevItem.location == item.location && prevItem.type == item.type && prevItem.docDomain == item.docDomain)
         policyHits.pop();
     }
     policyHits.push([wnd, node, item]);
   }
 
   function runTest([name, body, expectedURL, expectedType, expectedThirdParty, explicitEvent], stage)
   {
     defaultMatcher.clear();
 
     if (stage > 1)
       defaultMatcher.add(Filter.fromText(expectedURL));
     if (stage == 3)
       defaultMatcher.add(Filter.fromText("@@||127.0.0.1:1234/test|$document"));
     if (stage == 4)
       defaultMatcher.add(Filter.fromText("@@||127.0.0.1:1234/test|$~document"));
+    if (stage == 5)
+      defaultMatcher.add(Filter.fromText("@@||127.0.0.1:1234/test|$document,sitekey=" + publickey));
 
     if (!explicitEvent)
     {
       if (body.indexOf("2000/svg") >= 0)
       {
         // SVG image: add an onload attribute to the document element
-        body = body.replace(/(<svg\b)/, '$1 onload="this.dispatchEvent(new CustomEvent(\'frameready\', {bubbles: true}));"');
+        body = body.replace(/(<svg\b)/, '$1 onload="this.dispatchEvent(new CustomEvent(\'abp:frameready\', {bubbles: true}));"');
       }
       else
       {
         // HTML data: wrap it into a <body> tag
-        body = '<body onload="this.dispatchEvent(new CustomEvent(\'frameready\', {bubbles: true}));">' + body + '</body>';
+        body = '<body onload="this.dispatchEvent(new CustomEvent(\'abp:frameready\', {bubbles: true}));">' + body + '</body>';
       }
     }
 
     let serverHit = false;
     server.registerPathHandler("/test", function(metadata, response)
     {
       response.setStatusLine("1.1", "200", "OK");
 
       let contentType = "text/html";
       if (body.indexOf("2000/svg") >= 0)
+      {
         contentType = "image/svg+xml";
+        body = body.replace(/^<svg/, "<svg data-adblockkey='" + adblockkey + "'");
+      }
+      else
+        body = "<html data-adblockkey='" + adblockkey + "'>" + body + "</html>";
       response.setHeader("Content-Type", contentType + "; charset=utf-8");
 
       response.bodyOutputStream.write(body, body.length);
     });
     server.registerPathHandler("/redirect.gif", function(metadata, response)
     {
       response.setStatusLine("1.1", "302", "Moved Temporarily");
       response.setHeader("Location", "http://127.0.0.1:1234/test.gif");
     });
     server.registerPathHandler("/redirect2.gif", function(metadata, response)
@@ skipping 11 matching lines @@
       var error = "<b>Not found...<b>";
       response.bodyOutputStream.write(error, error.length);
     });
 
     policyHits = [];
     var callback = function()
     {
       let expectedStatus = "allowed";
       if (stage == 3)
         equal(policyHits.length, 0, "Number of policy hits");
-      else
+      // We cannot rely on the correctness of policy hits for sitekey filters due to blocking
+      // filter hits being counted even if the resource doesn't end up getting blocked
+      else if (stage != 5)
       {
         equal(policyHits.length, 1, "Number of policy hits");
         if (policyHits.length == 1)
         {
           let [wnd, node, item] = policyHits[0];
 
           equal(item.location, expectedURL, "Request URL");
 
           expectedStatus = (stage == 1 ? "allowed" : "blocked");
           let actualStatus = (item.filter ? "blocked" : "allowed");
 
           equal(actualStatus, expectedStatus, "Request blocked");
           equal(item.typeDescr.toLowerCase(), expectedType, "Request type");
           equal(item.thirdParty, expectedThirdParty, "Third-party flag");
           equal(item.docDomain, "127.0.0.1", "Document domain");
         }
       }
       server.registerPathHandler(expectedURL.replace(/http:\/\/[^\/]+/, ""), null);
       equal(serverHit, expectedStatus == "allowed", "Request received by server");
 
-      frame.removeEventListener("frameready", callback, false);
+      frame.removeEventListener("abp:frameready", callback, false);
 
       start();
     };
-    frame.addEventListener("frameready", callback, false, true);
+    frame.addEventListener("abp:frameready", callback, false, true);
     frame.setAttribute("src", "http://127.0.0.1:1234/test");
   }
 
   let stageDescriptions = {
     1: "running without filters",
     2: "running with filter %S",
     3: "running with filter %S and site exception",
     4: "running with filter %S and exception not applicable to sites",
+    5: "running with filter %S and sitekey exception"
   };
 
   for (let test = 0; test < tests.length; test++)
   {
     let [name, body, expectedURL, expectedType, expectedDomain, expectedThirdParty] = tests[test];
     for (let stage = 1; stage in stageDescriptions; stage++)
     {
       let stageDescription = stageDescriptions[stage];
       if (stageDescription.indexOf("%S") >= 0)
         stageDescription = stageDescription.replace("%S", expectedURL);
 
       asyncTest(name + " (" + stageDescription + ")", runTest.bind(null, tests[test], stage));
     }
   }
 })();