116 lines
3.1 KiB
JavaScript
116 lines
3.1 KiB
JavaScript
|
|
"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);
|
|
}
|
|
}
|
|
};
|
|
|