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
|
"use strict";
this.EXPORTED_SYMBOLS = ["MockRegistry"];
const {classes: Cc, interfaces: Ci, utils: Cu} = Components;
Cu.import("resource://testing-common/MockRegistrar.jsm");
Cu.import("resource://gre/modules/XPCOMUtils.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: XPCOMUtils.generateQI([Ci.nsIWindowsRegKey]),
// --- Overridden nsIWindowsRegKey interface functions ---
open: function(root, path, mode) {
let rootKey = registry.getRoot(root);
if (!rootKey.has(path)) {
rootKey.set(path, new Map());
}
this.values = rootKey.get(path);
},
close: function() {
this.values = null;
},
get valueCount() {
if (!this.values)
throw Components.results.NS_ERROR_FAILURE;
return this.values.size;
},
hasValue: function(name) {
if (!this.values) {
return false;
}
return this.values.has(name);
},
getValueType: function(name) {
return Ci.nsIWindowsRegKey.TYPE_STRING;
},
getValueName: function(index) {
if (!this.values || index >= this.values.size)
throw Components.results.NS_ERROR_FAILURE;
let names = Array.from(this.values.keys());
return names[index];
},
readStringValue: function(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);
}
}
};
|
