diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 01:47:29 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 01:47:29 +0000 |
commit | 0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d (patch) | |
tree | a31f07c9bcca9d56ce61e9a1ffd30ef350d513aa /testing/modules/MockRegistry.sys.mjs | |
parent | Initial commit. (diff) | |
download | firefox-esr-0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d.tar.xz firefox-esr-0ebf5bdf043a27fd3dfb7f92e0cb63d88954c44d.zip |
Adding upstream version 115.8.0esr.upstream/115.8.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'testing/modules/MockRegistry.sys.mjs')
-rw-r--r-- | testing/modules/MockRegistry.sys.mjs | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/testing/modules/MockRegistry.sys.mjs b/testing/modules/MockRegistry.sys.mjs new file mode 100644 index 0000000000..e9351e576d --- /dev/null +++ b/testing/modules/MockRegistry.sys.mjs @@ -0,0 +1,145 @@ +/* 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/. */ + +import { MockRegistrar } from "resource://testing-common/MockRegistrar.sys.mjs"; + +export 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); + }, + }; + + // See bug 1688838 - nsNotifyAddrListener::CheckAdaptersAddresses might + // attempt to use the registry off the main thread, so we disable that + // feature while the mock registry is active. + this.oldSuffixListPref = Services.prefs.getBoolPref( + "network.notify.dnsSuffixList" + ); + Services.prefs.setBoolPref("network.notify.dnsSuffixList", false); + + this.oldCheckForProxiesPref = Services.prefs.getBoolPref( + "network.notify.checkForProxies" + ); + Services.prefs.setBoolPref("network.notify.checkForProxies", false); + + this.oldCheckForNRPTPref = Services.prefs.getBoolPref( + "network.notify.checkForNRPT" + ); + Services.prefs.setBoolPref("network.notify.checkForNRPT", false); + + this.cid = MockRegistrar.register( + "@mozilla.org/windows-registry-key;1", + MockWindowsRegKey + ); + } + + shutdown() { + MockRegistrar.unregister(this.cid); + Services.prefs.setBoolPref( + "network.notify.dnsSuffixList", + this.oldSuffixListPref + ); + Services.prefs.setBoolPref( + "network.notify.checkForProxies", + this.oldCheckForProxiesPref + ); + Services.prefs.setBoolPref( + "network.notify.checkForNRPT", + this.oldCheckForNRPTPref + ); + 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); + } + } +} |