1 files changed, 30 insertions, 0 deletions
diff --git a/remote/shared/messagehandler/test/browser/resources/modules/ModuleRegistry.sys.mjs b/remote/shared/messagehandler/test/browser/resources/modules/ModuleRegistry.sys.mjs
new file mode 100644
index 0000000000..e5d9c38719
--- /dev/null
+++ b/remote/shared/messagehandler/test/browser/resources/modules/ModuleRegistry.sys.mjs
@@ -0,0 +1,30 @@
+/* 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/. */
+
+/**
+ * Retrieve the WebDriver BiDi module class matching the provided module name
+ * and folder.
+ *
+ * @param {String} moduleName
+ *     The name of the module to get the class for.
+ * @param {String} moduleFolder
+ *     A valid folder name for modules.
+ * @return {Class=}
+ *     The class corresponding to the module name and folder, null if no match
+ *     was found.
+ * @throws {Error}
+ *     If the provided module folder is unexpected.
+ **/
+export const getModuleClass = function(moduleName, moduleFolder) {
+  const root = `chrome://mochitests/content/browser/remote/shared/messagehandler/test/`;
+  const path = `${root}browser/resources/modules/${moduleFolder}/${moduleName}.sys.mjs`;
+  try {
+    return ChromeUtils.importESModule(path)[moduleName];
+  } catch (e) {
+    if (e.result == Cr.NS_ERROR_FILE_NOT_FOUND) {
+      return null;
+    }
+    throw e;
+  }
+};