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
116
117
118
119
120
121
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
"use strict";
var EXPORTED_SYMBOLS = ["MockRegistry"];
const { MockRegistrar } = ChromeUtils.import(
"resource://testing-common/MockRegistrar.jsm"
);
class MockRegistry {
constructor() {
// Three level structure of Maps pointing to Maps pointing to Maps
// this.roots is the top of the structure and has ROOT_KEY_* values
// as keys. Maps at the second level are the values of the first
// level Map, they have registry keys (also called paths) as keys.
// Third level maps are the values in second level maps, they have
// map registry names to corresponding values (which in this implementation
// are always strings).
this.roots = new Map([
[Ci.nsIWindowsRegKey.ROOT_KEY_LOCAL_MACHINE, new Map()],
[Ci.nsIWindowsRegKey.ROOT_KEY_CURRENT_USER, new Map()],
[Ci.nsIWindowsRegKey.ROOT_KEY_CLASSES_ROOT, new Map()],
]);
let registry = this;
/**
* This is a mock nsIWindowsRegistry implementation. It only implements a
* subset of the interface used in tests. In particular, only values
* of type string are supported.
*/
function MockWindowsRegKey() {}
MockWindowsRegKey.prototype = {
values: null,
// --- Overridden nsISupports interface functions ---
QueryInterface: ChromeUtils.generateQI(["nsIWindowsRegKey"]),
// --- Overridden nsIWindowsRegKey interface functions ---
open(root, path, mode) {
let rootKey = registry.getRoot(root);
if (!rootKey.has(path)) {
rootKey.set(path, new Map());
}
this.values = rootKey.get(path);
},
close() {
this.values = null;
},
get valueCount() {
if (!this.values) {
throw Components.Exception("", Cr.NS_ERROR_FAILURE);
}
return this.values.size;
},
hasValue(name) {
if (!this.values) {
return false;
}
return this.values.has(name);
},
getValueType(name) {
return Ci.nsIWindowsRegKey.TYPE_STRING;
},
getValueName(index) {
if (!this.values || index >= this.values.size) {
throw Components.Exception("", Cr.NS_ERROR_FAILURE);
}
let names = Array.from(this.values.keys());
return names[index];
},
readStringValue(name) {
if (!this.values) {
throw new Error("invalid registry path");
}
return this.values.get(name);
},
};
this.cid = MockRegistrar.register(
"@mozilla.org/windows-registry-key;1",
MockWindowsRegKey
);
}
shutdown() {
MockRegistrar.unregister(this.cid);
this.cid = null;
}
getRoot(root) {
if (!this.roots.has(root)) {
throw new Error(`No such root ${root}`);
}
return this.roots.get(root);
}
setValue(root, path, name, value) {
let rootKey = this.getRoot(root);
if (!rootKey.has(path)) {
rootKey.set(path, new Map());
}
let pathmap = rootKey.get(path);
if (value == null) {
pathmap.delete(name);
} else {
pathmap.set(name, value);
}
}
}
|
