1 files changed, 115 insertions, 0 deletions
diff --git a/testing/modules/MockRegistry.jsm b/testing/modules/MockRegistry.jsm
new file mode 100644
index 000000000..f450b5b1c
--- /dev/null
+++ b/testing/modules/MockRegistry.jsm
@@ -0,0 +1,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);
+    }
+  }
+};
+