summaryrefslogtreecommitdiffstats
path: root/toolkit/components/url-classifier/tests/unit/test_threat_type_conversion.js
blob: b1bec40b4ffede5f87fb21d4899acc6f90825c90 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
function run_test() {
  let urlUtils = Cc["@mozilla.org/url-classifier/utils;1"].getService(
    Ci.nsIUrlClassifierUtils
  );

  // Test list name to threat type conversion.

  equal(urlUtils.convertListNameToThreatType("goog-malware-proto"), 1);
  equal(urlUtils.convertListNameToThreatType("googpub-phish-proto"), 2);
  equal(urlUtils.convertListNameToThreatType("goog-unwanted-proto"), 3);
  equal(urlUtils.convertListNameToThreatType("goog-harmful-proto"), 4);
  equal(urlUtils.convertListNameToThreatType("goog-phish-proto"), 5);
  equal(urlUtils.convertListNameToThreatType("goog-badbinurl-proto"), 7);
  equal(urlUtils.convertListNameToThreatType("goog-downloadwhite-proto"), 9);

  try {
    urlUtils.convertListNameToThreatType("bad-list-name");
    ok(false, "Bad list name should lead to exception.");
  } catch (e) {}

  try {
    urlUtils.convertListNameToThreatType("bad-list-name");
    ok(false, "Bad list name should lead to exception.");
  } catch (e) {}

  // Test threat type to list name conversion.
  equal(urlUtils.convertThreatTypeToListNames(1), "goog-malware-proto");
  equal(
    urlUtils.convertThreatTypeToListNames(2),
    "googpub-phish-proto,moztest-phish-proto,test-phish-proto"
  );
  equal(
    urlUtils.convertThreatTypeToListNames(3),
    "goog-unwanted-proto,moztest-unwanted-proto,test-unwanted-proto"
  );
  equal(urlUtils.convertThreatTypeToListNames(4), "goog-harmful-proto");
  equal(urlUtils.convertThreatTypeToListNames(5), "goog-phish-proto");
  equal(urlUtils.convertThreatTypeToListNames(7), "goog-badbinurl-proto");
  equal(urlUtils.convertThreatTypeToListNames(9), "goog-downloadwhite-proto");

  try {
    urlUtils.convertThreatTypeToListNames(0);
    ok(false, "Bad threat type should lead to exception.");
  } catch (e) {}

  try {
    urlUtils.convertThreatTypeToListNames(100);
    ok(false, "Bad threat type should lead to exception.");
  } catch (e) {}
}