The Tor Browser: toolkit/components/url-classifier/tests/unit/test_streamupdater.js@6474c204b198 (annotated)

toolkit/components/url-classifier/tests/unit/test_streamupdater.js@6474c204b198 (annotated)

toolkit/components/url-classifier/tests/unit/test_streamupdater.js

Wed, 31 Dec 2014 06:09:35 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Wed, 31 Dec 2014 06:09:35 +0100
changeset 0: 6474c204b198
permissions: -rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

 function doTest(updates, assertions, expectError)
 {
   if (expectError) {
     doUpdateTest(updates, assertions, updateError, runNextTest);
   } else {
     doUpdateTest(updates, assertions, runNextTest, updateError);
   }
 }
 function testFillDb() {
   var add1Urls = [ "zaz.com/a", "yxz.com/c" ];
   var update = "n:1000\n";
   update += "i:test-phish-simple\n";
   var update1 = buildBareUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:data:," + encodeURIComponent(update1) + "\n";
   var assertions = {
     "tableData" : "test-phish-simple;a:1",
     "urlsExist" : add1Urls
   };
   doTest([update], assertions, false);
 }
 function testSimpleForward() {
   var add1Urls = [ "foo.com/a", "bar.com/c" ];
   var add2Urls = [ "foo.com/b" ];
   var add3Urls = [ "bar.com/d" ];
   var update = "n:1000\n";
   update += "i:test-phish-simple\n";
   var update1 = buildBareUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:data:," + encodeURIComponent(update1) + "\n";
   var update2 = buildBareUpdate(
     [{ "chunkNum" : 2,
        "urls" : add2Urls }]);
   update += "u:data:," + encodeURIComponent(update2) + "\n";
   var update3 = buildBareUpdate(
     [{ "chunkNum" : 3,
        "urls" : add3Urls }]);
   update += "u:data:," + encodeURIComponent(update3) + "\n";
   var assertions = {
     "tableData" : "test-phish-simple;a:1-3",
     "urlsExist" : add1Urls.concat(add2Urls).concat(add3Urls)
   };
   doTest([update], assertions, false);
 }
 // Make sure that a nested forward (a forward within a forward) causes
 // the update to fail.
 function testNestedForward() {
   var add1Urls = [ "foo.com/a", "bar.com/c" ];
   var add2Urls = [ "foo.com/b" ];
   var update = "n:1000\n";
   update += "i:test-phish-simple\n";
   var update1 = buildBareUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:data:," + encodeURIComponent(update1) + "\n";
   var update2 = buildBareUpdate(
     [{ "chunkNum" : 2 }]);
   var update3 = buildBareUpdate(
     [{ "chunkNum" : 3,
        "urls" : add1Urls }]);
   update2 += "u:data:," + encodeURIComponent(update3) + "\n";
   update += "u:data:," + encodeURIComponent(update2) + "\n";
   var assertions = {
     "tableData" : "",
     "urlsDontExist" : add1Urls.concat(add2Urls)
   };
   doTest([update], assertions, true);
 }
 // An invalid URL forward causes the update to fail.
 function testInvalidUrlForward() {
   var add1Urls = [ "foo.com/a", "bar.com/c" ];
   var update = buildPhishingUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:asdf://blah/blah\n";  // invalid URL scheme
   // The first part of the update should have succeeded.
   var assertions = {
     "tableData" : "test-phish-simple;a:1",
     "urlsExist" : add1Urls
   };
   doTest([update], assertions, false);
 }
 // A failed network request causes the update to fail.
 function testErrorUrlForward() {
   var add1Urls = [ "foo.com/a", "bar.com/c" ];
   var update = buildPhishingUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:http://test.invalid/asdf/asdf\n";  // invalid URL scheme
   // The first part of the update should have succeeded
   var assertions = {
     "tableData" : "test-phish-simple;a:1",
     "urlsExist" : add1Urls
   };
   doTest([update], assertions, false);
 }
 function testMultipleTables() {
   var add1Urls = [ "foo.com/a", "bar.com/c" ];
   var add2Urls = [ "foo.com/b" ];
   var add3Urls = [ "bar.com/d" ];
   var update = "n:1000\n";
   update += "i:test-phish-simple\n";
   var update1 = buildBareUpdate(
     [{ "chunkNum" : 1,
        "urls" : add1Urls }]);
   update += "u:data:," + encodeURIComponent(update1) + "\n";
   var update2 = buildBareUpdate(
     [{ "chunkNum" : 2,
        "urls" : add2Urls }]);
   update += "u:data:," + encodeURIComponent(update2) + "\n";
   update += "i:test-malware-simple\n";
   var update3 = buildBareUpdate(
     [{ "chunkNum" : 3,
        "urls" : add3Urls }]);
   update += "u:data:," + encodeURIComponent(update3) + "\n";
   var assertions = {
     "tableData" : "test-malware-simple;a:3\ntest-phish-simple;a:1-2",
     "urlsExist" : add1Urls.concat(add2Urls),
     "malwareUrlsExist" : add3Urls
   };
   doTest([update], assertions, false);
 }
 function Observer(callback) {
   this.observe = callback;
 }
 Observer.prototype =
 {
 QueryInterface: function(iid)
 {
   if (!iid.equals(Ci.nsISupports) &&
       !iid.equals(Ci.nsIObserver)) {
     throw Cr.NS_ERROR_NO_INTERFACE;
   }
   return this;
 }
 };
 // Tests a database reset request.
 function testReset() {
   var addUrls1 = [ "foo.com/a", "foo.com/b" ];
   var update1 = buildPhishingUpdate(
     [
       { "chunkNum" : 1,
         "urls" : addUrls1
       }]);
   var update2 = "n:1000\nr:pleasereset\n";
   var addUrls3 = [ "bar.com/a", "bar.com/b" ];
   var update3 = buildPhishingUpdate(
     [
       { "chunkNum" : 3,
         "urls" : addUrls3
       }]);
   var assertions = {
     "tableData" : "test-phish-simple;a:3",
     "urlsExist" : addUrls3,
     "urlsDontExist" : addUrls1
   };
   doTest([update1, update2, update3], assertions, false);
 }
 function run_test()
 {
   runTests([
     testSimpleForward,
     testNestedForward,
     testInvalidUrlForward,
     testErrorUrlForward,
     testMultipleTables,
     testReset
   ]);
 }
 do_test_pending();