Commit 1ba65bb6 authored by DimiL's avatar DimiL
Browse files

Bug 1272239 - P2. Testcase, only tables with provider could register gethash... 

Bug 1272239 - P2. Testcase, only tables with provider could register gethash url in listmanager. r=francois

--HG--
extra : rebase_source : 9bb3c392f9443412ad60a01467ab9c77645f81ef
parent 8044ad85

toolkit/components/url-classifier/tests/mochitest/chrome.ini

+1 −0
Original line number Diff line number Diff line
@@ -17,4 +17,5 @@ tags = trackingprotection 
tags = trackingprotection

[test_trackingprotection_whitelist.html]

tags = trackingprotection

[test_safebrowsing_bug1272239.html]

[test_donottrack.html]

toolkit/components/url-classifier/tests/mochitest/test_safebrowsing_bug1272239.html

0 → 100644
+88 −0
Original line number Diff line number Diff line 
<!DOCTYPE HTML>

<html>

<head>

  <title>Bug 1272239 - Only tables with provider could register gethash url in listmanager.</title>

  <script type="text/javascript" src="chrome://mochikit/content/tests/SimpleTest/SimpleTest.js"></script>

  <link rel="stylesheet" type="text/css" href="chrome://mochikit/content/tests/SimpleTest/test.css">

</head>



<body>

<p id="display"></p>

<div id="content" style="display: none">

</div>

<pre id="test">



<script class="testbody" type="text/javascript">



var Cc = SpecialPowers.Cc;

var Ci = SpecialPowers.Ci;



// List all the tables

const prefs = [

  "urlclassifier.phishTable",

  "urlclassifier.malwareTable",

  "urlclassifier.downloadBlockTable",

  "urlclassifier.downloadAllowTable",

  "urlclassifier.trackingTable",

  "urlclassifier.trackingWhitelistTable",

  "urlclassifier.forbiddenTable",

  "urlclassifier.blockedTable"

];



var prefService = Cc["@mozilla.org/preferences-service;1"]

                  .getService(Ci.nsIPrefService);



// Get providers

var providers = {};



var branch = prefService.getBranch("browser.safebrowsing.provider.");

var children = branch.getChildList("", {});



for (var child of children) {

  var prefComponents =  child.split(".");

  var providerName = prefComponents[0];

  providers[providerName] = {};

}



// Get lists from |browser.safebrowsing.provider.PROVIDER_NAME.lists| preference.

var listsWithProvider = [];

var listsToProvider = [];

for (var provider in providers) {

  var pref = "browser.safebrowsing.provider." + provider + ".lists";

  var list = SpecialPowers.getCharPref(pref).split(",");



  listsToProvider = listsToProvider.concat(list.map( () => { return provider; }));

  listsWithProvider = listsWithProvider.concat(list);

}



// Get all the lists

var lists = [];

for (var pref of prefs) {

  lists = lists.concat(SpecialPowers.getCharPref(pref).split(","));

}



var listmanager = Cc["@mozilla.org/url-classifier/listmanager;1"].

                  getService(Ci.nsIUrlListManager);



for (var list of lists) {

  if (!list)

    continue;



  // For lists having a provider, it should have a correct gethash url

  // For lists without a provider, for example, test-malware-simple, it should not

  // have a gethash url.

  var url = listmanager.getGethashUrl(list);

  var index = listsWithProvider.indexOf(list);

  if (index >= 0) {

    var provider = listsToProvider[index];

    var pref = "browser.safebrowsing.provider." + provider + ".gethashURL";

    is(url, SpecialPowers.getCharPref(pref), list + " matches its gethash url");

  } else {

    is(url, "", list + " should not have a gethash url");

  }

}



</script>

</pre>

</body>

</html>