summaryrefslogtreecommitdiffstats
path: root/toolkit/modules/tests/xpcshell/test_readCertPrefs.js
blob: 0a030f9dd73b3660a85224a7422445d076430c32 (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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

const { CertUtils } = ChromeUtils.import(
  "resource://gre/modules/CertUtils.jsm"
);

const PREF_PREFIX = "certutils.certs.";

function resetPrefs() {
  var prefs = Services.prefs.getChildList(PREF_PREFIX);
  prefs.forEach(Services.prefs.clearUserPref);
}

function attributes_match(aCert, aExpected) {
  if (Object.keys(aCert).length != Object.keys(aExpected).length) {
    return false;
  }

  for (var attribute in aCert) {
    if (!(attribute in aExpected)) {
      return false;
    }
    if (aCert[attribute] != aExpected[attribute]) {
      return false;
    }
  }

  return true;
}

function test_results(aCerts, aExpected) {
  Assert.equal(aCerts.length, aExpected.length);

  for (var i = 0; i < aCerts.length; i++) {
    if (!attributes_match(aCerts[i], aExpected[i])) {
      dump(
        "Attributes for certificate " +
          (i + 1) +
          " did not match expected attributes\n"
      );
      dump("Saw: " + aCerts[i].toSource() + "\n");
      dump("Expected: " + aExpected[i].toSource() + "\n");
      Assert.ok(false);
    }
  }
}

add_test(function test_singleCert() {
  Services.prefs.setCharPref(PREF_PREFIX + "1.attribute1", "foo");
  Services.prefs.setCharPref(PREF_PREFIX + "1.attribute2", "bar");

  var certs = CertUtils.readCertPrefs(PREF_PREFIX);
  test_results(certs, [
    {
      attribute1: "foo",
      attribute2: "bar",
    },
  ]);

  resetPrefs();
  run_next_test();
});

add_test(function test_multipleCert() {
  Services.prefs.setCharPref(
    PREF_PREFIX + "1.md5Fingerprint",
    "cf84a9a2a804e021f27cb5128fe151f4"
  );
  Services.prefs.setCharPref(PREF_PREFIX + "1.nickname", "1st cert");
  Services.prefs.setCharPref(
    PREF_PREFIX + "2.md5Fingerprint",
    "9441051b7eb50e5ca2226095af710c1a"
  );
  Services.prefs.setCharPref(PREF_PREFIX + "2.nickname", "2nd cert");

  var certs = CertUtils.readCertPrefs(PREF_PREFIX);
  test_results(certs, [
    {
      md5Fingerprint: "cf84a9a2a804e021f27cb5128fe151f4",
      nickname: "1st cert",
    },
    {
      md5Fingerprint: "9441051b7eb50e5ca2226095af710c1a",
      nickname: "2nd cert",
    },
  ]);

  resetPrefs();
  run_next_test();
});

add_test(function test_skippedCert() {
  Services.prefs.setCharPref(PREF_PREFIX + "1.issuerName", "Mozilla");
  Services.prefs.setCharPref(PREF_PREFIX + "1.nickname", "1st cert");
  Services.prefs.setCharPref(PREF_PREFIX + "2.issuerName", "Top CA");
  Services.prefs.setCharPref(PREF_PREFIX + "2.nickname", "2nd cert");
  Services.prefs.setCharPref(PREF_PREFIX + "4.issuerName", "Unknown CA");
  Services.prefs.setCharPref(PREF_PREFIX + "4.nickname", "Ignored cert");

  var certs = CertUtils.readCertPrefs(PREF_PREFIX);
  test_results(certs, [
    {
      issuerName: "Mozilla",
      nickname: "1st cert",
    },
    {
      issuerName: "Top CA",
      nickname: "2nd cert",
    },
  ]);

  resetPrefs();
  run_next_test();
});